Quantifying the Memorization-to-Generalization Transition: Scaling Laws and Phase Structure in Grokking
Not on the map. Why?
Abstract (arXiv)
Neural networks trained past memorization frequently undergo a delayed transition to generalization, a phenomenon known as grokking. Despite theoretical progress on why this transition occurs, the quantitative structure of when it occurs in hyperparameter space remains uncharacterized. We map the memorization-to-generalization boundary across 384 configurations of two-hidden-layer MLPs on modular arithmetic, fitting a power-law scaling relation for generalization onset time: Tgrok propto H-0.27, D-2.04, η-0.50, λ-0.64 (R2 = 0.732; 0.821 with interactions). The exponent hierarchy reveals that data complexity (D-2.04) is the dominant driver of regime transition, not model capacity (H-0.27): doubling data accelerates generalization by ~4×, while doubling width yields only ~1.2×. A sharp phase boundary at weight decay λ gtrsim 1.0 separates grokking from non-grokking configurations, and weight norm trajectories show monotonic compression during the transition, consistent with implicit regularization selecting low-complexity solutions. These results provide a quantitative foundation for predicting and controlling regime transitions in overparameterized networks.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 15 of its 16 references that are in the map. First posted on arXiv 9 Sep 2026.
Builds on
15 of 16 references.
References · 15
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2023Extracting Training Data from Diffusion Modelscited by 732
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2022Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Modelscited by 424
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 306
- 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 187
3 more
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2024A Dynamical Model of Neural Scaling Lawscited by 111
- 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 83
Built on by
Citing papers per year: 2026: 1.
1 paper.