The Illusion of Latent Generalization: Bi-directionality and the Reversal Curse
Organizations: TUM, Helmholtz Munich · Google DeepMind
Abstract
The reversal curse describes a failure of autoregressive language models to retrieve a fact in reverse order (e.g., training on $A > B$'' but failing on ''). Recent work shows that objectives with bidirectional supervision (e.g., bidirectional attention or masking-based reconstruction for decoder-only models) can mitigate the reversal curse. We extend this evaluation to include a vanilla masked language modeling (MLM) objective and compare it to decoder-only masking-based training across four reversal benchmarks and then provide a minimal mechanistic study of how these objectives succeed. We show that reversal accuracy requires training signal that explicitly makes the source entity a prediction target, and we find little evidence that success corresponds to a single direction-agnostic representation of a fact. Instead, representation distances and linear probes are consistent with storing forward and reverse directions as distinct entries, with different indexing geometry for MLM versus decoder-only masking-based training. Our results caution that objective-level ``fixes'' can improve reversal behavior without necessarily inducing the kind of latent generalization one might expect from a unified concept.
Figures & tables
| Dataset | MLM | NTP+Mask | Abl. I |
|---|---|---|---|
| Simple Reversal | 100% | 99.5% | 0% |
| Nonsense | 60.2% | 16.7% | 0% |
| Celebrity | 86.4% | 45.0% | 0% |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Objective | Example training sequences for fact |
|---|---|
| NTP | A > B |
| MLM | [MASK] > B , A [MASK] B , A > [MASK] |
| NTP + Masking | A > B , [MASK] > B [SEP] A > B , A [MASK] B [SEP] A > B , A > [MASK] [SEP] A > B |
| Dataset | Training fact example | Probe (false frame; relation unchanged) | Acc. |
|---|---|---|---|
| Simple Reversal | (would wrongly suggest ) | 0% | |
| Nonsense | “femp are more dangerous than glon” | “glon are more dangerous than [MASK] ” (would wrongly suggest femp ) | 0% |
Decoder-Only Language Models
Decoder-only language models use causal attention to generate text and other sequences one token at a time.
See all 98 papers