mlsanityData to 14 Oct 2026

Small Initialization Matters for Large Language Models

arXiv 2606.17945 ↗First posted 16 Jun 2026cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Large language models provide a tractable system for asking how intelligence itself emerges, rather than only how LLMs can be engineered. Although progress is usually attributed to scale, data and architecture, we show that parameter initialization is a gene-like determinant of training and, in particular, of model capacity. Reducing the initialization scale consistently improves pretraining, with the largest gains on reasoning-demanding tasks. We identify two widely used empirical settings that restrain the advantage of small initialization, and show how relaxing them restores favorable scaling. We further uncover a critical initialization that balances the reasoning and training. Mechanistically, small initialization drives a distinct developmental trajectory: parameters first condense into low-complexity structures and later expand into richer representations, giving concrete form to the idea that compression is intelligence. Token-level analyses show that the gains concentrate on non-trivial, context-constrained predictions rather than all tokens uniformly. These results motivate a simple γ-initialization rule: expose initialization rage as an explicit knob and use small initialization by default, an almost cost-free intervention that improves pretraining and strengthens reasoning across model scales.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 37 of its 43 references that are in the map. First posted on arXiv 16 Jun 2026.

Builds on

37 of 43 references.

References · 37

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2020Language Models are Few-Shot Learnerscited by 32,419
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  5. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  6. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  7. 2020Scaling Laws for Neural Language Modelscited by 6,586
  8. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  9. 2016Layer Normalizationcited by 5,669
  10. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  11. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  12. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
25 more
  1. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  2. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  3. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  4. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  5. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
  6. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
  7. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  8. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
  9. 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,807
  10. 2020Aligning AI With Shared Human Valuescited by 723
  11. 2018On Lazy Training in Differentiable Programmingcited by 675
  12. 2020Kernel and Rich Regimes in Overparametrized Modelscited by 306
  13. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
  14. 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 265
  15. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 208
  16. 2025Why do LLMs attend to the first token?cited by 127
  17. 2024Unveiling and Harnessing Hidden Attention Sinks: Enhancing Large Language Models without Training through Attention Calibrationcited by 74
  18. 2020Phase diagram for two-layer ReLU neural networks at infinite-width limitcited by 67
  19. 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
  20. 2021Towards Understanding the Condensation of Neural Networks at Initial Trainingcited by 32
  21. 2025Complexity Control Facilitates Reasoning-Based Compositional Generalization in Transformerscited by 20
  22. 2024Initialization is Critical to Whether Transformers Fit Composite Functions by Reasoning or Memorizingcited by 17
  23. 2025An overview of condensation phenomenon in deep learningcited by 17
  24. 2025An Analysis for Reasoning Bias of Language Models with Small Initializationcited by 13
  25. 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamicscited by 10

Built on by

Not cited yet.