The Orthogonalized Read Is a Removable Training Scaffold for Recurrent Memory
Not on the map. Why?
Abstract (arXiv)
Orthogonalizing the mLSTM memory matrix at read time with five differentiable Newton-Schulz iterations improves noisy associative recall. We replicate this effect and investigate its mechanism. Training on MAD noisy recall exhibits a long chance-level plateau followed by a sharp increase in accuracy. The orthogonalized read improves conditioning during this plateau and can be removed after escape. Ablations support three findings. First, the benefit requires a self-consistent read and gradient: an exact recursive least-squares read (the Mesa layer) yields a similar benefit, while straight-through variants, delta-rule writes, frozen random keys, and Frobenius normalization show no improvement over baseline. Second, across a learning-rate x task-difficulty grid, orthogonalization multiplies escape hazard roughly six-fold, with no detectable dependence on difficulty, and widens the range of learning rates that produce successful runs. Third, adding orthogonalization at inference leaves chance-level failures unresolved, while removing it gradually after escape yields standard mLSTMs at near-perfect accuracy. Schedule changes alone recover much of the reported gain. A batch-size x learning-rate analysis separates the effects of per-step learning rate and gradient noise on escape hazard (elasticities +3.0 and -1.65, respectively), linking the original vocab-96 result to its large-batch training regime. Direct decoding of the memory state recovers roughly half of the associations in behaviorally failed models, indicating a readout-learning limitation despite substantial stored information. These results show that fixed-budget recall benchmarks are sensitive to trainability and provide a tractable setting for investigating abrupt behavioral transitions through measurements of internal representations.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 31 references that are in the map. First posted on arXiv 2 Jul 2026.
Builds on
21 of 31 references.
References · 21
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,593
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2023Are Emergent Abilities of Large Language Models a Mirage?cited by 523
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 502
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2024xLSTM: Extended Long Short-Term Memorycited by 327
- 2024Simple linear attention language models balance the recall-throughput tradeoffcited by 208
- 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
9 more
- 2024Old Optimizer, New Norm: An Anthologycited by 193
- 2023Uncovering mesa-optimization algorithms in Transformerscited by 100
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 74
- 2024Mechanistic Design and Scaling of Hybrid Architecturescited by 59
- 2025MesaNet: Sequence Modeling by Locally Optimal Test-Time Trainingcited by 47
- 2025Muon Outperforms Adam in Tail-End Associative Memory Learningcited by 46
- 2026Noise-Driven Escape from Metastable Phases explains Grokking in Deep Neural Networkscited by 5
- 2026An Embarrassingly Simple Way to Optimize Orthogonal Matrices at Scalecited by 3
- 2026Variational Linear Attention: Stable Associative Memory for Long-Context Transformerscited by 2
Built on by
Not cited yet.