Algebraic Representability as the Limiting Regime of Grokking: An Exactly Solvable Model with Holomorphic Activations
Chon-Fai Kam, Xavier Cadet, Miloud Bessafi, Frederic Cadet
cs.LG, stat.ML
Submitted: 2026-07-15
License: http://creativecommons.org/licenses/by/4.0/
The gist: Neural networks trained on modular arithmetic exhibit grokking, a delayed transition from memorisation to generalisation known to depend on model capacity: too little and the network memorises slowly
Terminology
Abstract
Neural networks trained on modular arithmetic exhibit grokking, a delayed transition from memorisation to generalisation known to depend on model capacity: too little and the network memorises slowly or not at all, too much and it generalises almost immediately. What happens at the extreme of this spectrum, when the architecture's expressible function class collapses to a finite-dimensional algebraic variety? We study two-layer networks with a holomorphic monomial activation sigma(z)=z k, trained on modular tasks encoded via roots of unity. Here the network output, regardless of hidden width, is confined to a (k+1)-dimensional subspace of characters of (Z p) squared, an O(k/p 2) slice of the full function space. We give a complete algebraic characterisation of this subspace: a task is representable if and only if its discrete Fourier support lies on the diagonal u+v = k (mod p), which for linear-phase targets reduces to the arithmetic criterion m+n=k. This is not merely a constraint on eventual generalisation but on memorisation itself: because the outputs are algebraically confined, a non-representable target cannot be fit even on the training set, and we prove a positive lower bound on the training loss, independent of width. Across 585 runs the algebraic prediction matches the observed outcome with 99.8% accuracy, with no memorisation regime and no grokking; outcomes split cleanly into instant success and outright failure. This binary behaviour is the limiting case of the capacity-grokking relationship: when the expressible class shrinks to a fixed algebraic object, the question of when a network will grok dissolves into whether it can represent the target at all. A bottleneck ablation connects this extreme to standard networks, tracing a continuous path from representational failure, through memorisation without generalisation, to grokking with a shrinking gap as capacity grows.
Sources
- A Basin-Selection Perspective on Grokking via Singular Learning Theory
- Grokking Modular Polynomials
- Grokking modular arithmetic
- Neural Tangent Kernel: Convergence and Generalization in Neural Networks
- On the Expressive Power of Deep Polynomial Neural Networks
- Grokking as the Transition from Lazy to Rich Training Dynamics
- Towards Understanding Grokking: An Effective Theory of Representation Learning
- Progress measures for grokking via mechanistic interpretability
- Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets
- Grokking as a First Order Phase Transition in Two Layer Networks
- Model Capacity Determines Grokking through Competing Memorisation and Generalisation Speeds
- Benefits of depth in neural networks
- Deep Complex Networks
- Grokking phase transitions in learning local rules with gradient descent
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks