A Theoretical Analysis of Why Masked Diffusion Models Mitigate the Reversal Curse
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Autoregressive language models (ARMs) suffer from the reversal curse: after learning ''A is B,'' they often fail on the reverse query ''B is A.'' Masked diffusion language models (MDMs) exhibit this failure in a much weaker form, but the underlying reason has remained unclear. A common explanation attributes this mitigation to their any-order masked training objective. However, observing ''[M] is B'' during training teaches recovery of A from B in one positional configuration, and does not by itself explain why the learned evidence should transfer to the reverse prompt ''B is [M].'' We provide a theoretical analysis showing that this transfer arises from a parameter-level coupling between forward and reverse positional conditionals: shared Transformer parameters store token-pair evidence, while relative positional encodings route attention through queries and keys without changing the value-side evidence being retrieved. In a one-layer MDM, we prove that forward masked training strengthens evidence that is reusable in reverse queries, induces correlated forward--reverse attention routes, and yields a positively aligned shared-storage gradient component that decreases the reverse loss to first order. Controlled one-layer experiments and large-scale LLaDA/Dream experiments verify these signatures and show that they translate into improved reverse prediction.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 31 of its 35 references that are in the map. First posted on arXiv 2 Feb 2026.
Builds on
31 of 35 references.
Closest ancestors · 7
Same direction, newest first
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025Large Language Diffusion Models14 Feb
- 2024Scaling up Masked Diffusion Models on Text24 Oct
- 2024Simple and Effective Masked Diffusion Language Models11 Jun
- 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data6 Jun
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
Other references · 24
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2017Convolutional Sequence to Sequence Learningcited by 550
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 437
12 more
- 2023The Reversal Curse: LLMs trained on "A is B" fail to learn "B is A"cited by 399
- 2023Scan and Snap: Understanding Training Dynamics and Token Composition in 1-layer Transformercited by 101
- 2023How Do Transformers Learn Topic Structure: Towards a Mechanistic Understandingcited by 85
- 2024Reverse Training to Nurse the Reversal Cursecited by 48
- 2024Towards a Theoretical Understanding of the 'Reversal Curse' via Training Dynamicscited by 27
- 2024The Factorization Curse: Which Tokens You Predict Underlie the Reversal Curse and Morecited by 25
- 2024One-Layer Transformer Provably Learns One-Nearest Neighbor In Contextcited by 14
- 2025Is the Reversal Curse a Binding Problem? Uncovering Limitations of Transformers from a Basic Generalization Failurecited by 13
- 2024Delving into the Reversal Curse: How Far Can Large Language Models Generalize?cited by 11
- 2025How Transformers Learn Regular Language Recognition: A Theoretical Study on Training Dynamics and Implicit Biascited by 10
- 2025Generalization or Hallucination? Understanding Out-of-Context Reasoning in Transformerscited by 10
- 2026DiffER: Diffusion Entity-Relation Modeling for Reversal Curse in Diffusion Large Language Modelscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Induction in Both Directions: A Mechanistic Analysis of In-Context Learning in Masked Diffusion Language Models
- 2025Activation Steering for Masked Diffusion Language Models
- 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions
- 2025Diffusion Beats Autoregressive in Data-Constrained Settings
- 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models
- 2026The Confidence Shortcut: A Reasoning Failure Mode of Masked Diffusion Models
- 2026Understanding and Accelerating the Training of Masked Diffusion Language Models
- 2026CoDR: Training-Free Confidence-Drift Remasking for Diffusion Language Models