The Norm-Separation Delay Law of Grokking: A First-Principles Theory of Delayed Generalization
Not on the map. Why?
Abstract (arXiv)
Grokking -- the sudden generalisation that appears long after a model has perfectly memorised its training data -- has been widely observed but lacks a quantitative theory explaining the length of the delay. We show that grokking is a norm-driven representational phase transition in regularised training dynamics, and establish the Norm-Separation Delay Law: Tgrok - Tmem = Theta(γeff-1 log(|θmem|2 / |θpost|2)), where γeff is the optimiser's effective contraction rate (γeff = ηλ for SGD, γeff ≥ ηλ for AdamW). The upper bound follows from a discrete Lyapunov contraction argument; the matching lower bound from dynamical constraints of regularised first-order optimisation. Across 293 training runs spanning modular addition, modular multiplication, and sparse parity, we confirm three falsifiable predictions: inverse scaling with weight decay (R2 = 0.97), inverse scaling with learning rate (R2 = 0.92), and logarithmic dependence on the norm ratio (Pearson r = 0.91). A fourth finding reveals that grokking requires an optimiser capable of decoupling memorisation from contraction: SGD fails entirely at the same hyperparameters where AdamW reliably groks. These results reframe grokking not as a mysterious optimisation artefact but as a predictable consequence of norm separation between competing interpolating representations. We further derive a practical three-input algorithm that predicts grokking delay at memorisation time with 34.6% mean absolute error (bootstrap 95% CI [30.0%, 39.4%], N=60 seeds), enabling principled early stopping.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 20 of its 36 references that are in the map. First posted on arXiv 5 Mar 2026.
Builds on
20 of 36 references.
References · 20
- 2017Attention Is All You Needcited by 59,932
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 667
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
- 2019On Exact Computation with an Infinitely Wide Neural Netcited by 572
- 2019Benign Overfitting in Linear Regressioncited by 544
8 more
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 308
- 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 184
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 142
- 2023A Toy Model of Universality: Reverse Engineering How Networks Learn Group Operationscited by 141
- 2023Explaining grokking through circuit efficiencycited by 105
- 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 90
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 76
- 2023Unifying Grokking and Double Descentcited by 49
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026The Weight Norm Sets the Grokking Timescale: A Causal Delay Lawcited by 4
- 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnosticscited by 2
- 2026Grokking on the Weight-Decay Clock: A Rate Hierarchy from Softly Broken Symmetriescited by 2
- 2026At-Grok Is Not Converged:A Measurement-Validity Audit for Grokking Representation Metricscited by 1
- 2026The Spectral Lifecycle of Transformer Training: Transient Compression Waves, Persistent Spectral Gradients, and the Q/K--V Asymmetrycited by 0
- 2026Is Grokking a Loss of Normal Hyperbolicity of the Interpolation Manifold?cited by 0
- 2026Is Grokking a Loss of Normal Hyperbolicity of the Interpolation Manifold?14 Aug 2026
- 2026Grokking on the Weight-Decay Clock: A Rate Hierarchy from Softly Broken Symmetries27 Jul 2026
- 2026At-Grok Is Not Converged:A Measurement-Validity Audit for Grokking Representation Metrics7 Jul 2026
- 2026The Weight Norm Sets the Grokking Timescale: A Causal Delay Law11 Jun 2026
- 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnostics19 May 2026
- 2026The Spectral Lifecycle of Transformer Training: Transient Compression Waves, Persistent Spectral Gradients, and the Q/K--V Asymmetry3 Apr 2026