mlsanityData to 7 Oct 2026

The Long Delay to Arithmetic Generalization: When Learned Representations Outrun Behavior

arXiv 2604.13082 ↗First posted 30 Mar 2026cs.LG cs.AIcited by 3

Not on the map. Why?

Abstract (arXiv)

Grokking in transformers trained on algorithmic tasks is characterized by a long delay between training-set fit and abrupt generalization, but the source of that delay remains poorly understood. In encoder-decoder arithmetic models, we argue that this delay reflects limited access to already learned structure rather than failure to acquire that structure in the first place. We study one-step Collatz prediction and find that the encoder organizes parity and residue structure within the first few thousand training steps, while output accuracy remains near chance for tens of thousands more. Causal interventions support the decoder bottleneck hypothesis. Transplanting a trained encoder into a fresh model accelerates grokking by 2.75 times, while transplanting a trained decoder actively hurts. Freezing a converged encoder and retraining only the decoder eliminates the plateau entirely and yields 97.6% accuracy, compared to 86.1% for joint training. What makes the decoder's job harder or easier depends on numeral representation. Across 15 bases, those whose factorization aligns with the Collatz map's arithmetic (e.g., base 24) reach 99.8% accuracy, while binary fails completely because its representations collapse and never recover. The choice of base acts as an inductive bias that controls how much local digit structure the decoder can exploit, producing large differences in learnability from the same underlying task.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 28 of its 34 references that are in the map. First posted on arXiv 30 Mar 2026.

Builds on

28 of 34 references.

References · 28

  1. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  2. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  3. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  4. 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
  5. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  6. 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 733
  7. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
  8. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
  9. 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 374
  10. 2019Deep Learning for Symbolic Mathematicscited by 266
  11. 2020Do Wide and Deep Networks Learn the Same Things? Uncovering How Neural Network Representations Vary with Width and Depthcited by 241
  12. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 229
16 more
  1. 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 184
  2. 2023A Toy Model of Universality: Reverse Engineering How Networks Learn Group Operationscited by 141
  3. 2023The Quantization Model of Neural Scalingcited by 137
  4. 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 136
  5. 2023Teaching Arithmetic to Small Transformerscited by 121
  6. 2023Explaining grokking through circuit efficiencycited by 105
  7. 2023Grokking as the Transition from Lazy to Rich Training Dynamicscited by 102
  8. 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 78
  9. 2024Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalizationcited by 78
  10. 2023Length Generalization in Arithmetic Transformerscited by 64
  11. 2023Understanding Addition in Transformerscited by 34
  12. 2023Learning the greatest common divisor: explaining transformer predictionscited by 33
  13. 2024Emergence in non-neural models: grokking modular arithmetic via average gradient outer productcited by 26
  14. 2023DecoderLens: Layerwise Interpretation of Encoder-Decoder Transformerscited by 22
  15. 2025Probing for Arithmetic Errors in Language Modelscited by 17
  16. 2025Transformers know more than they can tell -- Learning the Collatz sequencecited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.