The Weight Norm Sets the Grokking Timescale: A Causal Delay Law
Not on the map. Why?
Abstract (arXiv)
Grokking is the delayed onset of generalization in neural networks, arising long after they fit the training data. Whether the weight norm causes this delay is disputed: some studies report a critical norm at the transition, others observe grokking with no fixed norm at all. We settle this by intervening on the norm during training rather than only observing it. Under free training with weight decay, networks grok when the weight norm reaches a value Wc that varies little across seeds and learning rates (CV 1 to 2 percent) and grows with the modular base as a power law. When we instead clamp the norm to a fixed multiple rho of Wc and hold it there, the network still groks, but the delay follows T_grok proportional to exp(alpha rho). One exponent, alpha near 7.5, fits this delay across four moduli (R^2 = 0.996). Over the swept ranges the held norm moves the delay by about 19x and the learning rate by only about 2x, and holding the norm above Wc slows grokking rather than preventing it. A final LayerNorm removes the dependence by decoupling weight scale from the network function; without it the exponential law returns. This pinned-norm delay is the exponential counterpart to the logarithmic delay predicted for a freely contracting norm.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 22 of its 27 references that are in the map. First posted on arXiv 11 Jun 2026.
Builds on
22 of 27 references.
References · 22
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 187
- 2023Explaining grokking through circuit efficiencycited by 111
- 2023Grokking as the Transition from Lazy to Rich Training Dynamicscited by 108
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2024Confidence Regulation Neurons in Language Modelscited by 57
10 more
- 2025Grokking Beyond the Euclidean Norm of Model Parameterscited by 15
- 2024Progress Measures for Grokking on Real-world Taskscited by 8
- 2025What Can Grokking Teach Us About Learning Under Nonstationarity?cited by 8
- 2026The Norm-Separation Delay Law of Grokking: A First-Principles Theory of Delayed Generalizationcited by 7
- 2026A Systematic Empirical Study of Grokking: Depth, Architecture, Activation, and Regularizationcited by 6
- 2026The Geometric Inductive Bias of Grokking: Bypassing Phase Transitions via Architectural Topologycited by 5
- 2026Gated Normalization Removal and Scale Anchoring in Pre-Norm Transformerscited by 4
- 2026Neural Collapse Dynamics: Depth, Activation, Regularisation, and Feature Norm Thresholdcited by 2
- 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnosticscited by 2
- 2025When Data Falls Short: Grokking Below the Critical Thresholdcited by 1
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Structure-Specific Representational Priors Causally Control the Grokking Delaycited by 3
- 2026What Makes a Representational Prior Work? Feature Families, Label-Free Invariances, and Critical Windows in Grokkingcited by 2
- 2026Mapping the Emergence of Regularization-Driven Dynamics in Grokkingcited by 1
- 2026Capability Emergence Can Be Forecast: Per-Seed, In Advance, With Calibrated Intervals, Certified False Alarms, and a Blind Pre-Registered Gatecited by 0
- 2026Mapping the Emergence of Regularization-Driven Dynamics in Grokking26 Aug 2026
- 2026Capability Emergence Can Be Forecast: Per-Seed, In Advance, With Calibrated Intervals, Certified False Alarms, and a Blind Pre-Registered Gate17 Jul 2026
- 2026What Makes a Representational Prior Work? Feature Families, Label-Free Invariances, and Critical Windows in Grokking14 Jul 2026
- 2026Structure-Specific Representational Priors Causally Control the Grokking Delay5 Jul 2026