mlsanityData to 14 Oct 2026

Is Grokking a Loss of Normal Hyperbolicity of the Interpolation Manifold?

arXiv 2608.14803 ↗First posted 14 Aug 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

A recent line of work recasts the post-memorization phase of grokking as constrained optimization: once a network interpolates the training set, weight decay drives a slow drift along the zero-loss manifold toward lower norm. In the language of dynamical systems, this is a fast-slow system in which the interpolation manifold plays the role of a slow manifold. We ask a question that this framing makes natural but the existing literature does not address: is the sharp generalization transition a loss of normal hyperbolicity of that manifold: a fold- or bifurcation-like event in which a normal restoring direction goes flat? Or does the manifold stay uniformly attracting while generalization happens by smooth drift? We propose a simple, optimizer-agnostic diagnostic: the smallest nonzero singular value σmin+(mathbf J) of the residual Jacobian, which, for the squared loss, equals the slowest normal restoring rate of the manifold. On a two-layer ReLU network trained to grok modular addition under squared loss, σmin+(mathbf J) does not collapse at the transition; it is near zero only before memorization and attains its largest values during the transition. The result holds across five seeds, and the six smallest singular values behave identically; there is no subspace-local collapse either. This is preliminary evidence against the bifurcation hypothesis and in favor of the smooth-contraction picture. We are explicit that a single-setting, gradual-transition experiment under Adam optimizer does not prove the absence of a bifurcation; it constrains where one could hide.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 16 references that are in the map. First posted on arXiv 14 Aug 2026.

Builds on

14 of 16 references.

References · 14

  1. 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
  2. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  3. 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
  4. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
  5. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  6. 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
  7. 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
  8. 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
  9. 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
  10. 2023Unifying Grokking and Double Descentcited by 50
  11. 2025A Theoretical Framework for Grokking: Interpolation followed by Riemannian Norm Minimisationcited by 13
  12. 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networkscited by 11
2 more
  1. 2025The Geometry of Grokking: Norm Minimization on the Zero-Loss Manifoldcited by 9
  2. 2026The Norm-Separation Delay Law of Grokking: A First-Principles Theory of Delayed Generalizationcited by 7

Built on by

Not cited yet.