English

A Theoretical Analysis of Why Masked Diffusion Models Mitigate the Reversal Curse

Artificial Intelligence 2026-05-13 v2 Computation and Language

Abstract

Autoregressive language models (ARMs) suffer from the reversal curse: after learning ''AA is BB,'' they often fail on the reverse query ''BB is AA.'' 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][\mathbf{M}] is BB'' during training teaches recovery of AA from BB in one positional configuration, and does not by itself explain why the learned evidence should transfer to the reverse prompt ''BB is [M][\mathbf{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.

Keywords

Cite

@article{arxiv.2602.02133,
  title  = {A Theoretical Analysis of Why Masked Diffusion Models Mitigate the Reversal Curse},
  author = {Moongyu Jeon and Sangwoo Shin and BumJun Kim and Kyelim Lee and Albert No},
  journal= {arXiv preprint arXiv:2602.02133},
  year   = {2026}
}
R2 v1 2026-07-01T09:31:54.994Z