mlsanityData to 14 Oct 2026

Shattered Compositionality: Counterintuitive Learning Dynamics of Transformers for Arithmetic

arXiv 2601.22510 ↗First posted 30 Jan 2026cs.LG cs.AIcited by 3

Not on the map. Why?

Abstract (arXiv)

Large language models (LLMs) often achieve strong benchmark accuracy yet remain brittle under small distribution shifts. While recent mechanistic studies reveal the discrepancy between LLMs and humans in skill compositions, the learning dynamics of skill acquisition and the role of data distributions remain elusive. In this study, we train transformers on synthetic arithmetic tasks with black-box model-agnostic metrics for analyzing non-human skill compositions. We discover that transformers often acquire skills for arithmetic in reverse order or in parallel instead of human-like sequential rules--a phenomenon we refer to as shattered compositionality. To explain these behaviors, we provide evidence that correlational matching to the training data, rather than causal or procedural composition, shapes learning dynamics. As a consequence, this non-human acquisition creates competition between partially learned skills, producing characteristic mixing errors and weaker robustness under controlled distribution shifts. We further show that the same qualitative behavior persists in modern LLMs and is not mitigated by pure model scaling or scratchpad supervision. Our results highlight a mismatch between training-time skill acquisition and the human-like hierarchical compositions, with implications for reasoning reliability and out-of-distribution robustness.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 35 of its 43 references that are in the map. First posted on arXiv 30 Jan 2026.

Builds on

35 of 43 references.

References · 35

  1. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  2. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  3. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  4. 2022In-context Learning and Induction Headscited by 964
  5. 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
  6. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  7. 2023Quantifying Language Models' Sensitivity to Spurious Features in Prompt Design or: How I learned to start worrying about prompt formattingcited by 817
  8. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  9. 2024GSM-Symbolic: Understanding the Limitations of Mathematical Reasoning in Large Language Modelscited by 550
  10. 2023Faith and Fate: Limits of Transformers on Compositionalitycited by 503
  11. 2023The Reversal Curse: LLMs trained on "A is B" fail to learn "B is A"cited by 407
  12. 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 327
23 more
  1. 2023In-Context Learning Creates Task Vectorscited by 312
  2. 2025The Illusion of Thinking: Understanding the Strengths and Limitations of Reasoning Models via the Lens of Problem Complexitycited by 302
  3. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 230
  4. 2022Exploring Length Generalization in Large Language Modelscited by 206
  5. 2025Natural Emergent Misalignment from Reward Hacking in Production RLcited by 136
  6. 2023Teaching Arithmetic to Small Transformerscited by 123
  7. 2023A Theory for Emergence of Complex Skills in Language Modelscited by 113
  8. 2024Arithmetic Without Algorithms: Language Models Solve Math With a Bag of Heuristicscited by 97
  9. 2023A Mechanistic Interpretation of Arithmetic Reasoning in Language Models using Causal Mediation Analysiscited by 89
  10. 2024LLMs know their vulnerabilities: Uncover Safety Gaps through Natural Distribution Shiftscited by 86
  11. 2024LLM Circuit Analyses Are Consistent Across Training and Scalecited by 63
  12. 2024Interpreting and Improving Large Language Models in Arithmetic Calculationcited by 45
  13. 2024Pre-trained Large Language Models Use Fourier Features to Compute Additioncited by 39
  14. 2024How Far Can Transformers Reason? The Globality Barrier and Inductive Scratchpadcited by 36
  15. 2025Learning Compositional Functions with Transformers from Easy-to-Hard Datacited by 36
  16. 2024Answer, Assemble, Ace: Understanding How LMs Answer Multiple Choice Questionscited by 35
  17. 2025Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challengescited by 31
  18. 2024Emergence in non-neural models: grokking modular arithmetic via average gradient outer productcited by 28
  19. 2024Out-of-distribution generalization via composition: a lens through induction heads in Transformerscited by 28
  20. 2024Progressive distillation induces an implicit curriculumcited by 15
  21. 2025Task Generalization With AutoRegressive Compositional Structure: Can Learning From D Tasks Generalize to DT Tasks?cited by 11
  22. 2025Exploring LLM Reasoning Through Controlled Prompt Variationscited by 10
  23. 2025Understanding In-context Learning of Addition via Activation Subspacescited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.