The Geometry of Multi-Task Grokking: Transverse Instability, Superposition, and Weight Decay Phase Structure
Not on the map. Why?
Abstract (arXiv)
Grokking -- the abrupt transition from memorization to generalization long after near-zero training loss -- has been studied mainly in single-task settings. We extend geometric analysis to multi-task modular arithmetic, training shared-trunk Transformers on dual-task (mod-add + mod-mul) and tri-task (mod-add + mod-mul + mod-sq) objectives across a systematic weight decay sweep. Five consistent phenomena emerge. (1) Staggered grokking order: multiplication generalizes first, followed by squaring, then addition, with consistent delays across seeds. (2) Universal integrability: optimization trajectories remain confined to an empirically invariant low-dimensional execution manifold; commutator defects orthogonal to this manifold reliably precede generalization. (3) Weight decay phase structure: grokking timescale, curvature depth, reconstruction threshold, and defect lead covary systematically with weight decay, revealing distinct dynamical regimes and a sharp no-decay failure mode. (4) Holographic incompressibility: final solutions occupy only 4--8 principal trajectory directions yet are distributed across full-rank weights and destroyed by minimal perturbations; SVD truncation, magnitude pruning, and uniform scaling all fail to preserve performance. (5) Transverse fragility and redundancy: removing less than 10% of orthogonal gradient components eliminates grokking, yet dual-task models exhibit partial recovery under extreme deletion, suggesting redundant center manifolds enabled by overparameterization. Together, these results support a dynamical picture in which multi-task grokking constructs a compact superposition subspace in parameter space, with weight decay acting as compression pressure and excess parameters supplying geometric redundancy in optimization pathways.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 24 of its 33 references that are in the map. First posted on arXiv 19 Feb 2026.
Builds on
24 of 33 references.
References · 24
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2022Toy Models of Superpositioncited by 928
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
- 2015Escaping From Saddle Points --- Online Stochastic Gradient for Tensor Decompositioncited by 241
12 more
- 2023The Clock and the Pizza: Two Stories in Mechanistic Explanation of Neural Networkscited by 153
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 142
- 2023Explaining grokking through circuit efficiencycited by 105
- 2023Grokking as the Transition from Lazy to Rich Training Dynamicscited by 102
- 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 90
- 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 78
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 76
- 2019Large Scale Structure of Neural Network Loss Landscapescited by 68
- 2023Unifying Grokking and Double Descentcited by 49
- 2026Low-Dimensional and Transversely Curved Optimization Dynamics in Grokkingcited by 10
- 2026Low-Dimensional Execution Manifolds in Transformer Learning Dynamics: Evidence from Modular Arithmetic Taskscited by 7
- 2026Phase Transitions for Feature Learning in Neural Networkscited by 6
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scalescited by 5
- 2026Spectral Edge Dynamics Reveal Functional Modes of Learningcited by 2
- 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnosticscited by 2
- 2026The Lifecycle of the Spectral Edge: From Gradient Learning to Weight-Decay Compressioncited by 1
- 2026Gradient-Direction Sensitivity Reveals Linear-Centroid Coupling Hidden by Optimizer Trajectoriescited by 1
- 2026At-Grok Is Not Converged:A Measurement-Validity Audit for Grokking Representation Metricscited by 1
- 2026Global Low-Rank, Local Full-Rank: The Holographic Encoding of Learned Algorithmscited by 0
- 2026Emergence via Phase Transitions: Mechanism Landscapes and Universal Convergence Across Complex Systemscited by 0
- 2026At-Grok Is Not Converged:A Measurement-Validity Audit for Grokking Representation Metrics7 Jul 2026
- 2026Emergence via Phase Transitions: Mechanism Landscapes and Universal Convergence Across Complex Systems25 May 2026
- 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnostics19 May 2026
- 2026Gradient-Direction Sensitivity Reveals Linear-Centroid Coupling Hidden by Optimizer Trajectories28 Apr 2026
- 2026The Lifecycle of the Spectral Edge: From Gradient Learning to Weight-Decay Compression8 Apr 2026
- 2026Spectral Edge Dynamics Reveal Functional Modes of Learning6 Apr 2026
- 2026Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scales14 Mar 2026
- 2026Global Low-Rank, Local Full-Rank: The Holographic Encoding of Learned Algorithms20 Feb 2026