Radial Suppression Accelerates Algorithmic Generalization: A Geometric Analysis of Delayed Generalization
Not on the map. Why?
Abstract (arXiv)
Why do neural networks memorize algorithmic training data long before they generalize? We present a geometric case study demonstrating that, on tasks where generalization requires discovering structured low-dimensional circuits, the memorization-generalization delay is driven by radial inflation of hidden representations under cross-entropy optimization. We formalize a radial-angular decomposition of activation-space dynamics and derive three testable propositions: (i) that penalizing radial inflation induces anisotropic, data-dependent weight regularization; (ii) that it suppresses radial gradient energy below the isotropic random baseline, forcing predominantly angular updates; and (iii) that it biases convergence toward flatter minima. To empirically validate these propositions, we study a single-hyperparameter norm penalty that softly constrains activations to a sqrt(d)-radius hypersphere. On modular arithmetic, this penalty accelerates grokking up to 6x across MLPs and Transformers, and halves training steps for a 10M-parameter nanoGPT on 3-digit addition.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 19 of its 20 references that are in the map. First posted on arXiv 30 Jun 2026.
Builds on
19 of 20 references.
References · 19
- 2016Layer Normalizationcited by 5,669
- 2018Spectral Normalization for Generative Adversarial Networkscited by 1,636
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 543
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 230
- 2017Regularizing and Optimizing LSTM Language Modelscited by 205
- 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 187
- 2023The Clock and the Pizza: Two Stories in Mechanistic Explanation of Neural Networkscited by 156
- 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 150
- 2023Teaching Arithmetic to Small Transformerscited by 123
7 more
- 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 83
- 2025Grokking at the Edge of Numerical Stabilitycited by 46
- 2024Grokfast: Accelerated Grokking by Amplifying Slow Gradientscited by 41
- 2025Let Me Grok for You: Accelerating Grokking via Embedding Transfer from a Weaker Modelcited by 13
- 2025Dimensional Collapse in Transformer Attention Outputs: A Challenge for Sparse Dictionary Learningcited by 6
- 2026The Geometric Inductive Bias of Grokking: Bypassing Phase Transitions via Architectural Topologycited by 5
- 2025Minimizing Layerwise Activation Norm Improves Generalization in Federated Learningcited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.