Global Low-Rank, Local Full-Rank: The Holographic Encoding of Learned Algorithms
Not on the map. Why?
Abstract (arXiv)
Grokking -- the abrupt transition from memorization to generalization after extended training -- has been linked to the emergence of low-dimensional structure in learning dynamics. Yet neural network parameters inhabit extremely high-dimensional spaces. How can a low-dimensional learning process produce solutions that resist low-dimensional compression? We investigate this question in multi-task modular arithmetic, training shared-trunk Transformers with separate heads for addition, multiplication, and a quadratic operation modulo 97. Across three model scales (315K--2.2M parameters) and five weight decay settings, we compare three reconstruction methods: per-matrix SVD, joint cross-matrix SVD, and trajectory PCA. Across all conditions, grokking trajectories are confined to a 2--6 dimensional global subspace, while individual weight matrices remain effectively full-rank. Reconstruction from 3--5 trajectory PCs recovers over 95% of final accuracy, whereas both per-matrix and joint SVD fail at sub-full rank. Even when static decompositions capture most spectral energy, they destroy task-relevant structure. These results show that learned algorithms are encoded through dynamically coordinated updates spanning all matrices, rather than localized low-rank components. We term this the holographic encoding principle: grokked solutions are globally low-rank in the space of learning directions but locally full-rank in parameter space, with implications for compression, interpretability, and understanding how neural networks encode computation.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 14 of its 16 references that are in the map. First posted on arXiv 20 Feb 2026.
Builds on
14 of 16 references.
References · 14
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2022Toy Models of Superpositioncited by 928
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
- 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
- 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
- 2023Unifying Grokking and Double Descentcited by 49
- 2026Low-Dimensional and Transversely Curved Optimization Dynamics in Grokkingcited by 10
- 2026The Geometry of Multi-Task Grokking: Transverse Instability, Superposition, and Weight Decay Phase Structurecited by 8
Built on by
Not cited yet.