Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Radial Suppression Accelerates Algorithmic Generalization: A Geometric Analysis of Delayed Generalization

About

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.

Srijan Tiwari, Aditya Chauhan, Manjot Singh• 2026

Related benchmarks

TaskDatasetResultRank
Modular ArithmeticModular Arithmetic P=97 (test)
Final Test Accuracy100
5
Modular AdditionModular Addition f_train=0.5 P=97
Grokking Onset (epochs)2.46e+3
4
3-digit addition3-digit addition NanoGPT implementation (train test)
Grokking Onset9.80e+3
2
Showing 3 of 3 rows

Other info

Follow for update