REVIEW 4 major objections 4 minor 16 references
Grokking and Generalization Collapse: Insights from \texttt{HTSR} theory
T0 review · 4 major / 4 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read After very long training without weight decay, a grokked network's test accuracy collapses to near chance while training accuracy stays perfect, and the spectral exponent alpha falling below 2 heralds the collapse.
desk verdict One genuine new observation — late-training collapse after grokking under zero weight decay — but it rests on a single seed-0 run, so the phenomenon is promising, not yet established. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The central object is $\alpha$, the exponent of the heavy-tailed power-law fit to the empirical spectral density of each layer's Gram matrix $X = \tfrac{1}{N}W^{\top}W$, so that $\rho_{\mathrm{emp}}(\lambda) \sim \lambda^{-\alpha}$ over a chosen tail region. The argument sets the random-matrix Marchenko--Pastur density as the null model for uncorrelated weights and treats deviations from it as learned structure; $\alpha \approx 2$ is the claimed universal layer-convergence target, with $2 \lesssim \alpha \lesssim 5$ indicating well-conditioned layers, $\alpha \gtrsim 5$ indicating underfit or random-like layers, and $\alpha < 2$ indicating very heavy tails and overfitting. The companion mechanism is the correlation trap: an eigenvalue of the elementwise-randomized weight matrix lying far outside the best-fit Marchenko--Pastur edge $\lambda_+$, interpreted as an anomalous rank-one-or-higher perturbation that makes the layer's weight distribution atypical. The paper uses these tools to read off which phase the network is in at each stage of training.
What would settle it
Retrain the same MLP on the same 1,000-sample MNIST subset with zero weight decay and the same hyperparameters across ten random seeds out to $10^7$ steps, logging test accuracy and layer $\alpha$; if even one seed keeps test accuracy high past $10^7$ without an $\alpha<2$ episode, or shows a collapse with $\alpha$ never below $2$, the 'invariable herald' claim is falsified.
Extended reading notes
Core claim
The central claim is that training a depth-3, width-200 ReLU MLP on 1,000 MNIST samples with mean-squared error and no weight decay for $10^7$ steps exposes a third phase beyond pre-grokking and grokking: test accuracy falls to roughly $0.5$ while training accuracy remains perfect and the $\ell^2$ weight norm keeps growing. The paper argues that pre-grokking is a mixed-layer state, where some layers approach $\alpha \approx 4$ while others stay near $\alpha \approx 5$, that grokking peaks when all important layers approach $\alpha \approx 2$, and that anti-grokking is marked by one or more layers crossing below $\alpha = 2$, with correlation traps visible in the spectrum of the randomized weights. It also claims that only the HTSR $\alpha$ separates all three phases: activation sparsity, absolute weight entropy, approximate local circuit complexity, and the $\ell^2$ norm track the first two phases but fail to distinguish grokking from anti-grokking. In the weight-decay control run the $\alpha$ curve plateaus near $2$ and test accuracy plateaus instead of collapsing, which the paper reads as corroboration that $\alpha$ marks the phase.
Load-bearing premise
The headline claim is carried by one long run: a single random seed, one 3-layer MLP, and one 1,000-sample MNIST subset, with no seed-averaged curves, so the assertion that $\alpha<2$ and correlation traps invariably herald collapse has only one trajectory behind it.
Editorial extensions
If this is right
- The paper's claim implies that, without weight decay, training far past grokking does not stay at peak test accuracy; the model can silently enter anti-grokking, so training duration itself becomes a generalization risk.
- Tracking $\alpha$ per layer can warn before test accuracy falls, because the crossing of $\alpha<2$ precedes or coincides with collapse and the metric needs only the weights, not labels or test data.
- Explanations of grokking built on the $\ell^2$ norm, activation sparsity, entropy, or circuit complexity miss the late failure mode; the $\alpha=2$ cutoff is the benchmarked signal that separates healthy grokking from collapse.
- In the weight-decay control, $\alpha$ plateaus near $2$ and test accuracy stays on a plateau rather than collapsing, suggesting that weight decay keeps layers out of the $\alpha<2$ regime and may protect against anti-grokking.
Reading between the lines
- The paper's evidence for the word 'invariably' is a single seed-0 trajectory; seed-averaged runs with varied optimizers and learning rates would determine whether anti-grokking is a robust phase or an artifact of one initialization.
- Because alpha is a function of weights alone, the same diagnostic could be applied to transformers and convolutional networks under long training, although the paper studies only one MLP on MNIST and leaves that extension implicit.
- The correlation-trap description, a large mean shift that makes the weight distribution atypical, resonates with rank-collapse and benign-overfitting ideas in other settings, but connecting those ideas is the reader's inference, not the paper's.
- If the alpha-about-2 target is as universal as claimed, an alpha-guided regularizer or early-stopping rule is a natural testable application; the paper lists such strategies as future work without demonstrating them.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies grokking dynamics in a 3-layer MLP (width 200) trained on a 1,000-sample MNIST subset for up to 10^7 steps with zero weight decay, and reports a novel third phase called "anti-grokking": after an initial grokking phase, test accuracy collapses to chance while training accuracy remains perfect. The authors claim that their heavy-tailed self-regularization (HTSR) layer quality metric alpha alone delineates all three phases (pre-grokking, grokking, anti-grokking), that alpha falling below 2 "invariably" heralds the collapse, and that this collapse is accompanied by "correlation traps"—outlier eigenvalues of randomized weight matrices detected by the WeightWatcher tool and validated by Kolmogorov-Smirnov tests. They further argue that competing metrics (activation sparsity, absolute weight entropy, approximate local circuit complexity, l2 norm) fail to distinguish grokking from anti-grokking.
Significance. If the central claims hold, the paper would contribute a striking new empirical phenomenon—late-training generalization collapse without weight decay—and a practical spectral progress measure that can detect it without test-set access. The use of an established open-source tool, the explicit comparison against several prior progress measures, and the attempt to validate correlation traps with KS tests are strengths. However, the significance is currently contingent on the robustness of a single long trajectory, and the paper's strongest language ("invariably," "definitive signal") goes beyond the evidence provided. The claimed novelty of the anti-grokking phase deserves serious scrutiny, but the manuscript needs substantial additional experimental evidence and more careful hedging before the main claims can be accepted.
major comments (4)
- [Appendix A, Table 3; §4.1, Fig. 1] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
- [§4.1, Fig. 4; §3.1] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
- [§4.2, Table 2; Appendix D; Appendix E] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
- [§3.1, §4.1, Refs. [8,9,10,11]] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
minor comments (4)
- [Appendix D, Table 4] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
- [§4.1 and Figure 5 caption] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
- [Throughout] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
- [Equation (9)] This comment is identical to the one above but appears twice in the array; the JSON schema does not require unique comments, but the author should treat this as a single point. This duplicate is a formatting artifact; please ignore the second occurrence.
Circularity Check
No definitional circularity: alpha is measured from weights independently of test accuracy, and the self-cited HTSR threshold functions as external theory rather than an input that constructs the result.
full rationale
The three phases (pre-grokking, grokking, anti-grokking) are defined by the independently observed training and test accuracy curves (Figure 1), not by alpha. Alpha is computed from layer weight spectra via PL fits and randomized-matrix MP comparisons; it is not fitted to the test accuracy labels, and no equation in the paper defines test collapse in terms of alpha or vice versa. The claim that alpha < 2 heralds collapse imports the alpha = 2 cutoff and correlation-trap terminology from prior HTSR/SETOL work by the same authors (refs. [9,10,11]), but those works are external evidence rather than a reduction of this paper's observations to its own inputs, and the paper adds its own KS-test and visual spectral validation of the traps. The main weaknesses are statistical and evidentiary: the headline anti-grokking finding rests on a single seed-0 WD=0 trajectory, the word 'invariably' is not supported by seed-averaged curves, Table 2 labels the WD>0 endpoint 'Anti-grokking (Collapse)' even though Appendix C reports a plateau rather than a collapse, and Appendix E itself concedes that alpha below 2 does not strictly imply suboptimal generalization. These are correctness and robustness concerns, not circularity-by-construction.
Assumptions & free parameters
assumptions (4)
- domain assumption ESD of trained layer weights follows a power-law tail rho ~ lambda^{-alpha}, and alpha ranges correspond to training phases with alpha=2 optimal and alpha<2 overfit.
- standard math Elementwise randomization W to Wrand produces i.i.d. weights whose ESD should fit the Marchenko-Pastur distribution, and deviations indicate correlation traps.
- domain assumption Power-law exponent alpha estimated via Clauset MLE with automated lambda_min selection is reliable.
- domain assumption The 10^7-step WD=0 training dynamics are a normal optimizer trajectory, not a numerical artifact.
invented entities (2)
-
correlation trap
independent evidence
-
anti-grokking
independent evidence
Cite this review
Pith. "Pith review of Grokking and Generalization Collapse: Insights from \texttt{HTSR} theory." pith.science (2026). https://pith.science/paper/OEMPTWTW
@misc{pith2026250604434,
author = {Pith},
title = {Pith review of: Grokking and Generalization Collapse: Insights from \textttHTSR theory},
year = {2026},
howpublished = {\url{https://pith.science/paper/OEMPTWTW}},
note = {Machine review of arXiv:2506.04434}
}
abstract
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 -- \emph{anti-grokking} -- that occurs very late in training and resembles but is distinct from the familiar \emph{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 $\alpha$ alone delineates all three phases, whereas the best competing metrics detect only the first two. The \emph{anti-grokking} is revealed by training for $10^7$ and is invariably heralded by $\alpha < 2$ and the appearance of \emph{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 $l^2$ 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 \emph{HTSR} $\alpha$ provides universal layer-convergence target at $\alpha \approx 2$ and underscore the value of using the HTSR alpha $(\alpha)$ metric as a measure of generalization.
Figures
Figures from the paper (4 more)
Reference graph
Works this paper leans on
-
[1]
Aaron Clauset, Cosma Rohilla Shalizi, and Mark E.J. Newman. Power-law distributions in empirical data. SIAM Review, 51(4):661–703, 2009
work page 2009
-
[2]
Progress measures for grokking on real-world tasks, 2024
Satvik Golechha. Progress measures for grokking on real-world tasks, 2024
work page 2024
-
[3]
The impact of ac- tivation sparsity on overfitting in convolutional neural networks
Karim Huesmann, Luis Garcia Rodriguez, Lars Linsen, and Benjamin Risse. The impact of ac- tivation sparsity on overfitting in convolutional neural networks. InPattern Recognition. ICPR International Workshops and Challenges: Virtual Event, January 10–15, 2021, Proceedings, Part III, volume 12663 of Lecture Notes in Computer Science , pages 130–145. Spring...
work page 2021
-
[4]
Deep networks always grok and here is why, 2024
Ahmed Imtiaz Humayun, Randall Balestriero, and Richard Baraniuk. Deep networks always grok and here is why, 2024
work page 2024
-
[5]
Reddi, Ke Ye, Felix Chern, Felix Yu, Ruiqi Guo, and Sanjiv Kumar
Zonglin Li, Chong You, Srinadh Bhojanapalli, Daliang Li, Ankit Singh Rawat, Sashank J. Reddi, Ke Ye, Felix Chern, Felix Yu, Ruiqi Guo, and Sanjiv Kumar. The lazy neuron phe- nomenon: On emergence of activation sparsity in transformers. In The Eleventh International Conference on Learning Representations (ICLR) , 2023. arXiv:2210.06313
arXiv 2023
-
[6]
Ziming Liu, Ouail Kitouni, Niklas S. Nolte, Eric J. Michaud, Max Tegmark, and Mike Williams. Towards understanding grokking: An effective theory of representation learning. In Surbhi Koyejo, Sham Kakade (formerly Mohamed), Aarti Agarwal, Danielle Belgrave, Kyunghyun Cho, and Alice Oh, editors, Advances in Neural Information Processing Systems , volume 35,...
work page 2022
-
[7]
Marchenko and Leonid Andreevich Pastur
Vladimir A. Marchenko and Leonid Andreevich Pastur. Distribution of eigenvalues for some sets of random matrices. Matematicheskii Sbornik, 72(114)(4):507–536, 1967
work page 1967
-
[8]
Charles H. Martin. WeightWatcher: Analyze Deep Learning Models without Training or Data. https://github.com/CalculatedContent/WeightWatcher, 2018-2024. Ver- sion 0.7.5.5 used in this study. Accessed May 12, 2025
work page 2018
Show all 16 references
-
[9]
Martin, Christopher Hinrichs, and Michael W
Charles H. Martin, Christopher Hinrichs, and Michael W. Mahoney. SETOL: A Semi- Empirical Theory of (Deep) Learning.https://github.com/CalculatedContent/ setol_paper/blob/main/setol_draft.pdf, 2025. Preprint. 9
2025
-
[10]
Martin and Michael W
Charles H. Martin and Michael W. Mahoney. Implicit self-regularization in deep neural net- works: Evidence from heavy-tailed spectral analysis, 2021
2021
-
[11]
Martin and Michael W
Charles H. Martin and Michael W. Mahoney. Predicting trends in the quality of state-of-the-art deep learning models, 2021
2021
-
[12]
A tale of two circuits: grokking as competition of sparse and dense subnetworks, 2023
William Merrill, Nikolaos Tsilivis, and Aman Shukla. A tale of two circuits: grokking as competition of sparse and dense subnetworks, 2023
2023
-
[13]
Progress measures for grokking via mechanistic interpretability, 2023
Neel Nanda, Lawrence Chan, Tom Lieberum, Jess Smith, and Jacob Steinhardt. Progress measures for grokking via mechanistic interpretability, 2023
2023
-
[14]
Theoretical explanation of activation sparsity through flat minima and adversarial robustness, 2023
Ze Peng, Lei Qi, Yinghuan Shi, and Yang Gao. Theoretical explanation of activation sparsity through flat minima and adversarial robustness, 2023
2023
-
[15]
Grokking: Generalization beyond overfitting on small algorithmic datasets, 2022
Alethea Power, Yuri Burda, Harri Edwards, Igor Babuschkin, and Vedant Misra. Grokking: Generalization beyond overfitting on small algorithmic datasets, 2022
2022
-
[16]
Explaining grokking through circuit efficiency, 2023
Vikrant Varma, Rohin Shah, Zachary Kenton, J ´anos Kram´ar, and Ramana Kumar. Explaining grokking through circuit efficiency, 2023. 10 Appendices A Experimental Setup We train a Multi-Layer Perceptron (MLP) on a subset of the MNIST dataset using the hyperparame- ters detailed ...
2023
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.