Grokking as a Variance-Limited Phase Transition: Spectral Gating and the Epsilon-Stability Threshold
Not on the map. Why?
Abstract (arXiv)
Standard optimization theories struggle to explain grokking, where generalization occurs long after training convergence. While geometric studies attribute this to slow drift, they often overlook the interaction between the optimizer's noise structure and landscape curvature. This work analyzes AdamW dynamics on modular arithmetic tasks, revealing a ``Spectral Gating'' mechanism that regulates the transition from memorization to generalization. We find that AdamW operates as a variance-gated stochastic system. Grokking is constrained by a stability condition: the generalizing solution resides in a sharp basin (λmaxH) initially inaccessible under low-variance regimes. The ``delayed'' phase represents the accumulation of gradient variance required to lift the effective stability ceiling, permitting entry into this sharp manifold. Our ablation studies identify three complexity regimes: (1) Capacity Collapse (P < 23), where rank-deficiency prevents structural learning; (2) The Variance-Limited Regime (P ≈ 41), where generalization waits for the spectral gate to open; and (3) Stability Override (P > 67), where memorization becomes dimensionally unstable. Furthermore, we challenge the "Flat Minima" hypothesis for algorithmic tasks, showing that isotropic noise injection fails to induce grokking. Generalization requires the textitanisotropic rectification unique to adaptive optimizers, which directs noise into the tangent space of the solution manifold.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 16 of its 22 references that are in the map. First posted on arXiv 16 Mar 2026.
Builds on
16 of 22 references.
References · 16
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2019Limitations of the Empirical Fisher Approximation for Natural Gradient Descentcited by 217
- 2019A Tail-Index Analysis of Stochastic Gradient Noise in Deep Neural Networkscited by 214
- 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 155
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 139
- 2023Explaining grokking through circuit efficiencycited by 111
- 2023Grokking as the Transition from Lazy to Rich Training Dynamicscited by 108
- 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 94
4 more
- 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 83
- 2022Towards understanding how momentum improves generalization in deep learningcited by 40
- 2025A Theoretical Framework for Grokking: Interpolation followed by Riemannian Norm Minimisationcited by 13
- 2025The Geometry of Grokking: Norm Minimization on the Zero-Loss Manifoldcited by 9
Built on by
Citing papers per year: 2026: 1.
1 paper.