mlsanityData to 14 Oct 2026

Too Big to Think: Capacity, Memorization, and Generalization in Pre-Trained Transformers

arXiv 2506.09099 ↗First posted 10 Jun 2025cs.LG cs.AI cs.CLcited by 4

On the map: LLM reasoning → Reasoning benchmarks and evaluation

Abstract (arXiv)

The relationship between memorization and generalization in large language models (LLMs) remains an open area of research, with growing evidence that the two are deeply intertwined. In this work, we investigate this relationship by pre-training a series of capacity-limited Transformer models from scratch on two synthetic character-level tasks designed to separately probe generalization (via arithmetic extrapolation) and memorization (via factual recall). We observe a consistent trade-off: small models extrapolate to unseen arithmetic cases but fail to memorize facts, while larger models memorize but fail to extrapolate. An intermediate-capacity model exhibits a similar shift toward memorization. When trained on both tasks jointly, no model (regardless of size) succeeds at extrapolation. These findings suggest that pre-training may intrinsically favor one learning mode over the other. By isolating these dynamics in a controlled setting, our study offers insight into how model capacity shapes learning behavior and offers broader implications for the design and deployment of small language models.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Reasoning benchmarks and evaluation, one of 17 directions in LLM reasoning. Builds on 15 of its 20 references that are in the map. First posted on arXiv 10 Jun 2025.

Builds on

15 of 20 references.

Closest ancestors · 1

Same direction, newest first

  1. 2024Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalization23 May

Other references · 14

  1. 2023GPT-4 Technical Reportcited by 18,396
  2. 2022Training language models to follow instructions with human feedbackcited by 15,971
  3. 2024GPT-4o System Cardcited by 4,588
  4. 2023Gemini: A Family of Highly Capable Multimodal Modelscited by 872
  5. 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
  6. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  7. 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 396
  8. 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 295
  9. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 230
  10. 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 220
  11. 2024Generalization v.s. Memorization: Tracing Language Models' Capabilities Back to Pretraining Datacited by 85
  12. 2024Alice in Wonderland: Simple Tasks Showing Complete Reasoning Breakdown in State-Of-the-Art Large Language Modelscited by 68
2 more
  1. 2023Exploring Memorization in Fine-tuned Language Modelscited by 45
  2. 2024Unveiling the Secret Recipe: A Guide For Supervised Fine-Tuning Small LLMscited by 40

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 3.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Analyzing Memorization in Large Language Models through the Lens of Model Attribution
  2. 2025How much do language models memorize?
  3. 2025An Analysis for Reasoning Bias of Language Models with Small Initialization
  4. 2026How Learning Governs Unlearning across the Memorization-Generalization Spectrum
  5. 2021How BPE Affects Memorization in Transformers
  6. 2024What Do Learning Dynamics Reveal About Generalization in LLM Reasoning?
  7. 2024Unforgettable Generalization in Language Models
  8. 2025The Landscape of Memorization in LLMs: Mechanisms, Measurement, and Mitigation