mlsanityData to 7 Oct 2026

Grokking and Generalization Collapse: Insights from textttHTSR theory

arXiv 2506.04434 ↗First posted 4 Jun 2025cs.LG cs.AI stat.MLcited by 8

Not on the map. Why?

Abstract (arXiv)

We study the well-known grokking phenomena in neural networks (NNs) using a 3-layer MLP trained on 1 k-sample subset of MNIST, with and without weight decay, and discover a novel third phase -- anti-grokking -- that occurs very late in training and resembles but is distinct from the familiar pre-grokking phases: test accuracy collapses while training accuracy stays perfect. This late-stage collapse is distinct, from the known pre-grokking and grokking phases, and is not detected by other proposed grokking progress measures. Leveraging Heavy-Tailed Self-Regularization HTSR through the open-source WeightWatcher tool, we show that the HTSR layer quality metric α alone delineates all three phases, whereas the best competing metrics detect only the first two. The anti-grokking is revealed by training for 107 and is invariably heralded by α < 2 and the appearance of Correlation Traps -- outlier singular values in the randomized layer weight matrices that make the layer weight matrix atypical and signal overfitting of the training set. Such traps are verified by visual inspection of the layer-wise empirical spectral densities, and by using Kolmogorov--Smirnov tests on randomized spectra. Comparative metrics, including activation sparsity, absolute weight entropy, circuit complexity, and l2 weight norms track pre-grokking and grokking but fail to distinguish grokking from anti-grokking. This discovery provides a way to measure overfitting and generalization collapse without direct access to the test data. These results strengthen the claim that the HTSR α provides universal layer-convergence target at α ≈ 2 and underscore the value of using the HTSR alpha (α) metric as a measure of generalization.

Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 7 of its 13 references that are in the map. First posted on arXiv 4 Jun 2025.

Builds on

7 of 13 references.

References · 7

  1. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 229
  2. 2022The Lazy Neuron Phenomenon: On Emergence of Activation Sparsity in Transformerscited by 121
  3. 2023Explaining grokking through circuit efficiencycited by 105
  4. 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 78
  5. 2023A Theoretical Explanation of Activation Sparsity through Flat Minima and Adversarial Robustnesscited by 7
  6. 2024Progress Measures for Grokking on Real-world Taskscited by 7
  7. 2021The Impact of Activation Sparsity on Overfitting in Convolutional Neural Networkscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 2; 2026: 6.

8 papers.