Generalization Dynamics of LM Pre-training
Not on the map. Why?
Abstract (arXiv)
People typically assume that LMs stably mature from pattern-matching parrots to generalizable intelligence during pre-training. We build a toy eval suite and show this mental model is wrong: throughout pre-training, LMs frequently and suddenly hop between parrot-like and intelligence-like computations. We call this mode-hopping. Across our suite, LMs suddenly latch onto memorized or in-context patterns instead of in-context learning, use System 1 instead of System 2 thinking, pick up what sounds true instead of what is true, fail at multi-hop persona QA, out-of-context reasoning, and emergent misalignment -- then just as suddenly revert and generalize. Mode-hopping is not explained by standard optimization dynamics: it is locally stable and cannot be fixed by checkpoint averaging. We instead think of it as a capacity allocation problem: in a capacity-bounded model, generalizable circuits must compete with the shallow ones learned early in training, and the data in each pre-training window may decide which circuits win. Our suite provides a new efficient lens on generalization. We demonstrate two concrete applications: (i) select intermediate pre-training checkpoints that strongly generalize reasoning and alignment, better than the final pre- or mid-training checkpoints, and (ii) select pre-training data that controls and stabilizes generalization dynamics.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 31 of its 44 references that are in the map. First posted on arXiv 27 Sep 2026.
Builds on
31 of 44 references.
References · 31
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2022Toy Models of Superpositioncited by 971
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2023Are Emergent Abilities of Large Language Models a Mirage?cited by 523
- 2024Safety Alignment Should Be Made More Than Just a Few Tokens Deepcited by 472
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2023Larger language models do in-context learning differentlycited by 394
19 more
- 2023Function Vectors in Large Language Modelscited by 308
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 240
- 2019SGD on Neural Networks Learns Functions of Increasing Complexitycited by 204
- 2025GDPval: Evaluating AI Model Performance on Real-World Economically Valuable Taskscited by 158
- 2024Fine-Tuning Enhances Existing Mechanisms: A Case Study on Entity Trackingcited by 149
- 2023Taken out of context: On measuring situational awareness in LLMscited by 143
- 2024Language Models Learn to Mislead Humans via RLHFcited by 124
- 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 81
- 2023Successor Heads: Recurring, Interpretable Attention Heads In The Wildcited by 77
- 2025STAR-1: Safer Alignment of Reasoning LLMs with 1K Datacited by 68
- 2025Which Attention Heads Matter for In-Context Learning?cited by 61
- 2025Weird Generalization and Inductive Backdoors: New Ways to Corrupt LLMscited by 55
- 2024Competition of Mechanisms: Tracing How Language Models Handle Facts and Counterfactualscited by 40
- 2024Sharpness-Aware Minimization Efficiently Selects Flatter Minima Late in Trainingcited by 24
- 2025Data Mixing Can Induce Phase Transitions in Knowledge Acquisitioncited by 16
- 2026Rethinking Generalization in Reasoning SFT: A Conditional Analysis on Optimization, Data, and Model Capabilitycited by 14
- 2025Simple Mechanistic Explanations for Out-Of-Context Reasoningcited by 10
- 2025Compute-Optimal LLMs Provably Generalize Better With Scalecited by 5
- 2026Nexus: Same Pretraining Loss, Better Downstream Generalization via Common Minimacited by 4
Built on by
Not cited yet.