mlsanityData to 7 Oct 2026

Late-Stage Generalization Collapse in Grokking: Detecting anti-grokking with Weightwatcher

arXiv 2602.02859 ↗First posted 2 Feb 2026cs.LGcited by 2

Not on the map. Why?

Abstract (arXiv)

Memorization in neural networks lacks a precise operational definition and is often inferred from the grokking regime, where training accuracy saturates while test accuracy remains very low. We identify a previously unreported third phase of grokking in this training regime: anti-grokking, a late-stage collapse of generalization. We revisit two canonical grokking setups: a 3-layer MLP trained on a subset of MNIST and a transformer trained on modular addition, but extended training far beyond standard. In both cases, after models transition from pre-grokking to successful generalization, test accuracy collapses back to chance while training accuracy remains perfect, indicating a distinct post-generalization failure mode. To diagnose anti-grokking, we use the open-source textttWeightWatcher tool based on HTSR/SETOL theory. The primary signal is the emergence of Correlation Traps: anomalously large eigenvalues beyond the Marchenko--Pastur bulk in the empirical spectral density of shuffled weight matrices, which are predicted to impair generalization. As a secondary signal, anti-grokking corresponds to the average HTSR layer quality metric α deviating from 2.0. Neither metric requires access to the test or training data. We compare these signals to alternative grokking diagnostic, including ℓ2 norms, Activation Sparsity, Absolute Weight Entropy, and Local Circuit Complexity. These track pre-grokking and grokking but fail to identify anti-grokking. Finally, we show that Correlation Traps can induce catastrophic forgetting and/or prototype memorization, and observe similar pathologies in large-scale LLMs, like OSS GPT 20/120B.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 13 of its 24 references that are in the map. First posted on arXiv 2 Feb 2026.

Builds on

13 of 24 references.

References · 13

  1. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  2. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
  3. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 229
  4. 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 211
  5. 2020The Heavy-Tail Phenomenon in SGDcited by 122
  6. 2022The Lazy Neuron Phenomenon: On Emergence of Activation Sparsity in Transformerscited by 121
  7. 2020Predicting trends in the quality of state-of-the-art neural networks without access to training or testing datacited by 113
  8. 2023Explaining grokking through circuit efficiencycited by 105
  9. 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 78
  10. 2025SETOL: A Semi-Empirical Theory of (Deep) Learningcited by 10
  11. 2025Grokking and Generalization Collapse: Insights from textttHTSR theorycited by 8
  12. 2023A Theoretical Explanation of Activation Sparsity through Flat Minima and Adversarial Robustnesscited by 7
1 more
  1. 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: 2026: 2.

2 papers.