Unifying Learning Dynamics and Generalization in Transformers Scaling Law
Not on the map. Why?
Abstract (arXiv)
The scaling law, a cornerstone of Large Language Model (LLM) development, predicts improvements in model performance with increasing computational resources. Yet, while empirically validated, its theoretical underpinnings remain poorly understood. This work formalizes the learning dynamics of transformer-based language models as an ordinary differential equation (ODE) system, then approximates this process to kernel behaviors. Departing from prior toy-model analyses, we rigorously analyze stochastic gradient descent (SGD) training for multi-layer transformers on sequence-to-sequence data with arbitrary data distribution, closely mirroring real-world conditions. Our analysis characterizes the convergence of generalization error to the irreducible risk as computational resources scale with data, especially during the optimization process. We establish matching upper and lower bounds on the excess risk, characterized by a distinct phase transition. In the initial optimization phase, the excess risk decays exponentially relative to the computational cost sf C. However, once a specific resource allocation threshold is crossed, the system enters a statistical phase, where the generalization error follows a power-law decay of Theta(mathsfC-1/7). These rates are certified by complementary lower bounds -- statistical, via an information-theoretic two-point reduction, and optimization-side, via a first-order oracle argument -- rendering the two-stage law tight up to constants, logarithmic factors, and a condition-number gap. Beyond this unified framework, our theory derives isolated scaling laws for model size, training time, and dataset size, elucidating how each variable independently governs the bounds of generalization.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 69 of its 84 references that are in the map. First posted on arXiv 26 Dec 2025.
Builds on
69 of 84 references.
References · 69
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2023Sparks of Artificial General Intelligence: Early experiments with GPT-4cited by 2,592
38 more
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2024DeepSeek-Coder: When the Large Language Model Meets Programming -- The Rise of Code Intelligencecited by 1,181
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,135
- 2023Video-ChatGPT: Towards Detailed Video Understanding via Large Vision and Language Modelscited by 1,062
- 2023StarCoder: may the source be with you!cited by 799
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
- 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 735
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2024DeepSeek LLM: Scaling Open-Source Language Models with Longtermismcited by 642
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 582
- 2019On Exact Computation with an Infinitely Wide Neural Netcited by 573
- 2017Nonparametric regression using deep neural networks with ReLU activation functioncited by 433
- 2021Explaining Neural Scaling Lawscited by 363
- 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 327
- 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 270
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 268
- 2018Learning Overparameterized Neural Networks via Stochastic Gradient Descent on Structured Datacited by 261
- 2019A Constructive Prediction of the Generalization Error Across Scalescited by 210
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
- 2019On the Inductive Bias of Neural Tangent Kernelscited by 177
- 2018Adaptivity of deep ReLU network for learning in Besov and mixed smooth Besov spaces: optimal rate and curse of dimensionalitycited by 162
- 2020Finite Versus Infinite Neural Networks: an Empirical Studycited by 157
- 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 144
- 2023Physics of Language Models: Part 3.2, Knowledge Manipulationcited by 143
- 2018Regularization Matters: Generalization and Optimization of Neural Nets v.s. their Induced Kernelcited by 131
- 2020Infinite attention: NNGP and NTK for deep attention networkscited by 121
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2024A Dynamical Model of Neural Scaling Lawscited by 111
- 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
- 2023Fast Attention Requires Bounded Entriescited by 102
- 2022More Than a Toy: Random Matrix Models Predict How Real-World Neural Representations Generalizecited by 91
- 2023HyperAttention: Long-context Attention in Near-Linear Timecited by 85
- 2024A Primer on the Inner Workings of Transformer-based Language Modelscited by 85
- 2023White-Box Transformers via Sparse Rate Reductioncited by 80
- 20244+3 Phases of Compute-Optimal Neural Scaling Lawscited by 68
- 2020The Recurrent Neural Tangent Kernelcited by 67
- 2019Quadratic Suffices for Over-parametrization via Matrix Chernoff Boundcited by 59
50 most cited of 69 All 69 (JSON)
Built on by
Citing papers per year: 2026: 2.
2 papers.