English

The Norm-Separation Delay Law of Grokking: A First-Principles Theory of Delayed Generalization

Artificial Intelligence 2026-05-05 v2 Machine Learning

Abstract

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: TgrokTmem=Θ(γeff1log(θmem2/θpost2))T_{\mathrm{grok}} - T_{\mathrm{mem}} = \Theta(\gamma_{\mathrm{eff}}^{-1} \log(\|\theta_{\mathrm{mem}}\|^2 / \|\theta_{\mathrm{post}}\|^2)), where γeff\gamma_{\mathrm{eff}} is the optimiser's effective contraction rate (γeff=ηλ\gamma_{\mathrm{eff}} = \eta\lambda for SGD, γeffηλ\gamma_{\mathrm{eff}} \ge \eta\lambda 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.97R^2 = 0.97), inverse scaling with learning rate (R2=0.92R^2 = 0.92), and logarithmic dependence on the norm ratio (Pearson r=0.91r = 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=60N=60 seeds), enabling principled early stopping.

Keywords

Cite

@article{arxiv.2603.13331,
  title  = {The Norm-Separation Delay Law of Grokking: A First-Principles Theory of Delayed Generalization},
  author = {Truong Xuan Khanh and Truong Quynh Hoa and Luu Duc Trung and Phan Thanh Duc},
  journal= {arXiv preprint arXiv:2603.13331},
  year   = {2026}
}

Comments

38 pages, 6 figs