Emergent One-Third Scaling Law as Attention Tries to Concentrate
cs.LG, cs.AI, stat.ML
Submitted: 2026-09-26
Updated: 2026-09-26
Comments: 32 pages, 16 figures
License: http://creativecommons.org/licenses/by/4.0/
The gist: The neural scaling law relating longer training to better performance through a power law is central to today's large language models (LLMs), yet its origin remains debated.
Terminology
Abstract
The neural scaling law relating longer training to better performance through a power law is central to today's large language models (LLMs), yet its origin remains debated. One recent proposal is that power laws can emerge from the strong non-linearity of a single softmax head learning peaked distributions. What happens with multiple softmax functions, as in LLMs, is unclear. Here, we show through toy models that any softmax learning peaked distributions, regardless of its position in the model, can develop logit magnitudes that grow in a power law with exponent 1/3, becoming a training bottleneck whose loss contribution decays as a power law with the same exponent 1/3. The overall loss therefore obeys 1/3 scaling whenever at least one softmax learns peaked distributions. We confirm that many softmax functions in LLMs learn peaked distributions and that LLM loss scaling matches this 1/3 prediction. Moreover, logit growth dynamics reveal that attention heads, rather than the language modeling head, are the bottleneck likely driving the 1/3 loss scaling in LLMs. Attention trying to concentrate on specific information, which is the heart of Transformers, may therefore also be the heart of the neural scaling law of training.
Sources
- GPT-4 Technical Report
- A Theory for Emergence of Complex Skills in Language Models
- Chinchilla Scaling: A replication attempt
- Learning Curves for SGD on Structured Features
- A Dynamical Model of Neural Scaling Laws
- Theory of Scaling Laws for In-Context Regression: Depth, Width, Context and Time
- Neural Scaling Laws Rooted in the Data Distribution
- Deriving Neural Scaling Laws from the statistics of natural language
- Scaling Laws and Spectra of Shallow Neural Networks in the Feature Learning Regime
- The Pile: An 800GB Dataset of Diverse Text for Language Modeling
- Scaling Laws for Autoregressive Generative Modeling
- Deep Learning Scaling is Predictable, Empirically
- Training Compute-Optimal Large Language Models
- Learning Curve Theory
- Scaling Laws for Neural Language Models
- A Boundary-Layer Mechanism for One-Third Scaling in Online Softmax Classification
- Neural Scaling Universality: If Exponents Are Fixed, Time to Understand Coefficients
- Inverse Depth Scaling From Most Layers Being Similar
- Universal One-third Time Scaling in Learning Peaked Distributions
- Physics of Skill Learning
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks