alphaFeedback

WorksComputer Science

Mapping the Solution-Set Structure of Grokking on Latin Squares

AI contributions
Not recorded for this paper
View details

Abstract

Grokking is a delayed transition from memorisation to generalisation. A network first fits observed examples while remaining near chance on held-out inputs, then generalises abruptly. Prior work has focused mainly on this delay, finding that the network's main loss becomes weight decay after the fitting phase. We name the solution to this optimization problem as the minimum-norm endpoint, and study which tasks make all minimum-norm endpoints generalise from partial observations. Specifically, we model pairwise input tasks as edge colourings of Kq,qK_{q,q}, with Latin colourings as our structured family, and classify each observed instance as all-good, mixed, or no-good by its endpoint set. Furthermore, we prove two contrasting results: (1) Finite-abelian-group families are asymptotically all-good from partial observations; (2) Under a uniform random Latin square, all-good tasks become asymptotically rare for observation-based learners, leaving mixed or no-good behaviour. Moreover, experiments show that relabellings of addition generalise despite their dense polynomial representations. Finally, on tasks of composite order, finite-training results suggest that predictors learn a coarse part of the output before the full label.

Introduction

Grokking is delayed generalisation: a network fits its training examples, remains near chance on held-out inputs for a long period, and then generalises abruptly (Power et al., [42]). The classical instance is modular addition, where the network observes some pairs (x,y)(x, y) with their sums x+y mod qx + y \bmod q and must infer the rest. Such a learning curve combines two things, the quality of the predictor that training eventually reaches and the time taken to reach it, and existing work examines how learned representations, regularisation, and the training data shape this behaviour (Nanda et al., [38]; Mohamadi et al., [36]; Gu et al., [17]). However, a run that generalises shows that there exists one solution does; it does not show that every solution a training could have selected would. We isolate the generalisation behaviour and ask it at the level of the task. Defining the minimum-norm endpoint as the solution to the weight decay-induced optimization problem, our central question is therefore: which task structures make the entire minimum-norm endpoint set generalise?

Answering it needs modelling of the task and endpoint set. A pairwise input task assigns an output to every pair of inputs, and we encode it as an edge colouring of the complete bipartite graph Kq,qK_{q,q}: the two input alphabets are the two vertex classes, each pair (u,v)(u,v) is an edge, which we also call a cell, and the output is its colour. Written out, the colouring is a q×qq \times q table, and at prime-power order it is a polynomial over Fq\mathbb{F}_q; table, graph, and polynomial are three coordinate views on one object. The colouring is Latin when it is proper, so that no vertex sees a colour twice; in table coordinates every symbol appears exactly once in each row and column, so the Latin colourings are the Latin squares of combinatorics. Every group multiplication table is one, and the family therefore contains modular addition together with its natural competitors. Permuting the vertices within either class, permuting the colours, or swapping the two classes only changes how the table is written; we call such a move a relabelling, and the orbit of a colouring under relabellings is our task object. Training observes a uniformly random subset of the cells of fixed size.

The network reads an edge through its two endpoints, with one embedding table per vertex class, a hidden ReLU layer, and a head over the colours. For the endpoint of training we follow implicit-bias theory, which motivates the minimum-norm interpolator as the idealised late-training object ([46, 30, 39]). A partially observed colouring, however, has many minimum-norm interpolators: all meet the margins on the observed cells and are free to disagree on the rest, and a handful of training runs yields a few selected predictors without certifying that they belong to this set or how the set behaves as a whole. We therefore take the whole set, the endpoint set, as the object of study, and describe it by what holds across it at a fixed held-out threshold τ\tau: an observed instance is all-good if every minimum-norm interpolator generalises, no-good if none does, and mixed if both kinds coexist. Averaged over the random observations, the three probabilities form the task’s phase profile, a point of a triangle (Figure 1, middle and right).

Image not extracted

Figure 1. Coordinates, endpoint sets, and the two poles. Left: x+yx+y mod 97; a relabelling of it, 9,118 terms and degree 190; and a random-matching colouring outside the orbit. Middle: the minimum-norm interpolators Sq(D)\mathcal{S}_q(\mathcal{D}) agree on the observed cells, may differ elsewhere; their extremes Aq−≤Aq+A_q^{-} \leq A_q^{+} cut τ\tau into all-good, mixed, no-good. Right: each orbit maps to a simplex point; abelian group families sit at the all-good vertex (cyclic addition illustrated), while the uniform Latin ensemble sits on the opposite edge, where the mixed/no-good split is open.

Relabelling induces a norm-preserving bijection between endpoint sets and preserves the law of the observations, so the phase profile is well defined on orbits: it is a map from colouring orbits to the triangle. The network realises the bijection by permuting the rows of its embedding tables and of its head. This separates task structure from its polynomial description. Take x+yx+y mod 97 and relabel it. The endpoint set is transported by the bijection, but the polynomial coordinate becomes a polynomial over F97\mathbb{F}_{97} with 9,118 nonzero terms and total degree 190, with essentially the same support density and the same degree as a proper colouring assembled from random perfect matchings, which lies outside the orbit. Networks trained on the relabelled colouring memorise and then generalise as they do on x+yx+y; networks trained on the random-matching colouring end below chance (Figure 1, left). Relabelling preserves the phase while changing polynomial sparsity and degree, and in this comparison those coordinate statistics also fail to distinguish the trained predictors’ outcomes. This is why we work with orbits.

Our results place two structures at opposite poles of the triangle. The all-good pole contains every finite abelian group family: at any fixed positive observation fraction, the worst endpoint accuracy converges to one in probability as the order grows. Cyclic addition is one instance, and non-isomorphic abelian groups supply distinct all-good orbits. At the opposite pole, a uniformly random Latin colouring defeats every learner that reads only the observations, at any fixed positive accuracy; consequently the colourings with a non-negligible all-good component have vanishing uniform mass. Generalisation is thus guaranteed throughout a structured family yet rare in the ambient space.

The core of our theory is a lemma of carrier criterion: a network that scores the full grid with controlled margins at a known norm cost. It bounds the cost of every minimum-norm endpoint on every mask, and a finite atlas fixed before the observations counts the harmful low-norm alternatives, whose number is too small for any of them to evade a random mask. The criterion does not depend on the colouring, and abelian characters provide carriers uniformly across group structures. Therefore, we have proven that satisfying this lemma is a sufficient condition for all good. We conjecture that the carrier criterion is also necessary: every asymptotically all-good Latin family admits a full-grid carrier in the same norm-and-margin regime.

Experiments at composite order 64 compare the guaranteed families with non-abelian and non-group candidates and show that coarse output factors are recovered before exact labels. These trained predictors probe structure; they do not measure the phase of the whole endpoint set. Whether a typical random colouring has some good endpoint, and whether any family carries non-vanishing mixed mass, remain open.

Contributions.

  • A task-level formulation: Latin colourings of Kq,qK_{q,q} compared up to relabelling, each observed instance judged by its whole minimum-norm solution set.

  • Two poles: every finite abelian group family is asymptotically all-good, while the uniform Latin ensemble defeats every learner at any fixed positive held-out accuracy.

  • A colouring-independent carrier criterion for all-goodness, together with the conjecture that this criterion is necessary.

  • Experiments in the same model class: relabelled addition groks like addition while looking like noise, and at composite order structured tasks recover coarse output factors before exact labels.

Paper Organization. Section 2 reviews prior work, Section 3 defines the model, Section 4 states and proves the two poles and the abelian extension, Section 5 reports the composite-order experiments and states the conjecture, and Section 6 concludes. Experiments that speak directly to the theory follow the theory itself.

Related Work

Grokking is reported in non-neural models [32], deep networks [14, 33], in-context and multi-hop reasoning [19, 56, 1], and LLM pretraining [27]; its onset needs a critical data fraction [64, 21] and its delay can be shortened [49, 26, 40, 24, 59]. Related approaches include representation learning, norm and optimisation bias, and circuit competition. Representation-level work studies Fourier circuits and progress measures [38, 16, 15, 3], spectral dynamics [63, 61], embedding bottlenecks [2], architectural bias [44] (Yıldırım 2026), task symmetry [23], embedding uniformity coupled to the training distribution [17], feature-emergence scaling laws [50], and topological signatures [48].

Norm- and optimisation-based accounts appeal to implicit bias toward max-margin, minimum-norm endpoints [46, 30, 39, 31, 36, 22], to norm minimisation on the zero-loss manifold [37, 51], to the edge of linear separability [4], to numerical stability [43], sharpness [35], Jacobian alignment [55], and quantitative delay laws [52, 47, 11].

Competition accounts pit a slow generalising circuit against a fast memorising one [54, 34], linking grokking to double descent and emergent abilities [9, 21, 28], with phase-transition, statistical, basin-selection, and capacity variants [7, 6, 8, 45, 29]. Provable results include generalisation guarantees for bounded-norm quadratic networks on modular addition [36] and delay bounds in ridge regression [58]. Gu et al. [17] separately analyse the accuracy ceiling and embedding dynamics; Appendix E.2 discusses their ceiling and transition-time measurements. Our setting fixes a ReLU architecture and a uniform mask law, compares Latin colourings up to relabelling, and quantifies over every minimum-norm response. These choices specify the scope of our endpoint guarantees.

Latin-Square Tasks and the Endpoint Set

This section builds the objects the introduction used informally: the task and its observed part, the endpoint set, its three phases, and the relabelling invariance that makes the phase a property of the task rather than of its presentation.

Tasks, masks, and three coordinates

Let q≥2q \ge2, [q]={0,…,q−1}[q] = \{0,\ldots,q-1\}, Xq=[q]2X_q = [q]^2, and N=q2N = q^2. A pairwise classification task is a colouring C:Xq→[q]C : X_q \to[q]; the set of all colourings is Cq\mathcal{C}_q. A colouring is Latin when every symbol occurs exactly once in each row and each column; Lq⊂Cq\mathcal{L}_q \subset\mathcal{C}_q denotes the Latin colourings of order qq. For every qq, modular addition Cqadd(u,v)=u+v(modq)C_q^{\mathrm{add}}(u,v) = u+v \pmod q is Latin and is the cyclic prototype of the abelian all-good family proved below.

Fix α∈(0,1)\alpha\in(0,1). The observed cells form a uniformly random mask of fixed size,

Ω∼νq:=Unif⁡{Ω′⊆Xq:∣Ω′∣=⌊αN⌋},T=Xq∖Ω;(1)\Omega\sim\nu_q := \operatorname{Unif}\{\Omega' \subseteq X_q : |\Omega'| = \lfloor\alpha N \rfloor\}, \qquad T = X_q \setminus\Omega; \tag*{(1)}

TT is the held-out set and D(C,Ω)={(u,v,C(u,v)):(u,v)∈Ω}\mathcal{D}(C,\Omega) = \{(u,v,C(u,v)) : (u,v) \in\Omega\} the dataset. All asymptotic statements hold at fixed α\alpha.

Proposition 3.1 (Two coordinates, and a third at prime-power order). For every q≥2q \ge2, a label table C∈CqC \in\mathcal{C}_q and a qq-edge-colouring of Kq,qK_{q,q} (colour C(u,v)C(u,v) on the edge UuVvU_uV_v) determine each other. When qq is a prime power and [q][q] is identified with Fq\mathbb{F}_q, both are also equivalent to the unique reduced polynomial PC∈Fq[X,Y]P_C \in\mathbb{F}_q[X,Y] with deg⁡XPC,deg⁡YPC≤q−1\deg_X P_C,\deg_Y P_C \le q-1 and PC(u,v)=C(u,v)P_C(u,v)=C(u,v). CC is Latin if and only if the edge colouring is proper, i.e. each colour class is a perfect matching.

The minimum-norm endpoint set

The model is a bias-free one-hidden-layer ReLU network with a separate embedding table for each input, EU,EV:[q]→RdqE_U,E_V : [q] \to\mathbb{R}^{d_q}, a hidden layer W∈Rhq×dqW \in\mathbb{R}^{h_q \times d_q}, and a head A∈Rq×hqA \in\mathbb{R}^{q \times h_q}, where dq=2q+1d_q = 2q+1 and hq=q2h_q = q^2; the width allows one hidden unit per cell, which is what feasibility needs, and the embedding dimension exceeds the rank of any reduced representation. Writing θ=(EU,EV,W,A)\theta=(E_U,E_V,W,A),

Fθ(u,v)=AReLU⁡(W(EU(u)+EV(v)))∈Rq,∥θ∥2=∥EU∥F2+∥EV∥F2+∥W∥F2+∥A∥F2.(2)F_\theta(u,v)=A\operatorname{ReLU}(W(E_U(u)+E_V(v))) \in\mathbb{R}^q,\qquad\|\theta\|^2=\|E_U\|_F^2+\|E_V\|_F^2+\|W\|_F^2+\|A\|_F^2. \tag*{(2)}

As a function class the pair embedding contains the common design that shares one table EE and concatenates [E(u);E(v)][E(u);E(v)], since W[E(u);E(v)]=W1E(u)+W2E(v)W[E(u);E(v)]=W_1E(u)+W_2E(v); the norms differ, so the endpoint sets need not coincide. The endpoint set of an observed dataset is the set of global minimisers of the parameter norm under unit margins on the observed cells,

Sq(D)=arg⁡min⁡θ 12∥θ∥2s.t.Fθ,y(u,v)≥1+max⁡a≠yFθ,a(u,v)((u,v,y)∈D).(3)\mathcal{S}_q(\mathcal{D})=\underset{\theta}{\arg\min}\ \frac{1}{2}\|\theta\|^2\quad\text{s.t.}\quad F_{\theta,y}(u,v)\ge1+\max_{a\ne y}F_{\theta,a}(u,v)\qquad((u,v,y)\in\mathcal{D}). \tag*{(3)}

Max-margin and implicit-bias results for homogeneous networks motivate this as the idealised late-training object ([46, 30, 39]). Sq(D)\mathcal{S}_q(\mathcal{D}) is a set: all its members meet the margins on Ω\Omega, and they may predict differently on TT. Appendix § 8.§ 2 treats the extension to L2L_2-regularised biases.

Three of the four parameter blocks can be folded into one matrix, which is the form the proofs work with. Assemble Φ=[EU(0),…,EU(q−1),EV(0),…,EV(q−1)]∈Rdq×2q\Phi=[E_U(0),\ldots,E_U(q-1),E_V(0),\ldots,E_V(q-1)]\in\mathbb{R}^{d_q\times2q}, put Z=WΦZ=W\Phi and ξ(u,v)=eu+eq+v∈R2q\xi_{(u,v)}=e_u+e_{q+v}\in\mathbb{R}^{2q}, so that Fθ(x)=AReLU⁡(Zξx)F_\theta(x)=A\operatorname{ReLU}(Z\xi_x). Call (A,Z)(A,Z) a reduced representation and define

J(F)=inf⁡{∥A∥F∥Z∥∗:F=AReLU⁡(Zξ), any finite width},J(F)=\inf\{\|A\|_F\|Z\|_*:F=A\operatorname{ReLU}(Z\xi),\ \text{any finite width}\},
J∗=min⁡{∥A∥F∥Z∥∗:Z∈Rhq×2q, AReLU⁡(Zξ) meets the margins of (3)}.(4)J_*=\min\{\|A\|_F\|Z\|_*:Z\in\mathbb{R}^{h_q\times2q},\ A\operatorname{ReLU}(Z\xi)\text{ meets the margins of (3)}\}. \tag*{(4)}

Lemma 3.2 (Reduced normalisation). The optimal value of (3) is V∗=32J∗2/3V_*=\frac{3}{2}J_*^{2/3}, and every global minimiser is balanced: ∥A∥F2=∥W∥F2=∥Φ∥F2=J∗2/3\|A\|_F^2=\|W\|_F^2=\|\Phi\|_F^2=J_*^{2/3} and ∥A∥F∥WΦ∥∗=J∗\|A\|_F\|W\Phi\|_*=J_*.

In words: the nuclear norm of ZZ is the cost of its cheapest factorisation into a hidden layer and an embedding, and the scale freedom of the ReLU then balances the head against the rest, so an endpoint is simply a score table whose product ∥A∥F∥Z∥∗\|A\|_F\|Z\|_* is as small as the observations allow. This product is the one quantity the main proof tracks.

Phases of the endpoint set and the phase profile

Fix CC and Ω\Omega and write Sq=Sq(D(C,Ω))S_q = S_q(\mathcal{D}(C,\Omega)). Strict held-out accuracy counts a score tie as an error, and the best and worst endpoint accuracies are its extrema over the endpoint set:

References

  1. [1]Roman Abramov, Felix Steinbauer, and Gjergji Kasneci. Grokking in the wild: Data augmentation for real-world multi-hop reasoning with transformers. In Int. Conf. Mach. Learn. (ICML), 2025. arXiv:2504.20752. 3arxiv.org/abs/2504.20752
  2. [2]Hilal AlQuabeh, Velibor Bojkovic, Munichiso Nwadike, and Kentaro Inui. Mechanistic insights into grokking from the embedding layer. arXiv preprint arXiv:2505.15624, 2025. 3arxiv.org/abs/2505.15624
  3. [3]Boaz Barak, Benjamin Edelman, Surbhi Goel, Sham Kakade, Eran Malach, and Cyril Zhang. Hidden progress in deep learning: SGD learns parities near the computational limit. In Adv. Neural Inform. Process. Syst. (NeurIPS), volume 35, pp. 21750–21764, 2022. 3arxiv.org/abs/2207.08799
  4. [4]Alon Beck, Noam Levi, and Yohai Bar-Sinai. Grokking at the edge of linear separability. In Int. Conf. Mach. Learn. (ICML), 2025. arXiv:2410.04489. 3arxiv.org/abs/2410.04489
  5. [5]Matthias C. Caro and Ishaun Datta. Pseudo-dimension of quantum circuits. Quantum Machine Intelligence, 2(2):14, 2020. arXiv:2002.01490. 7, 26DOI
  6. [6]Breno W. Carvalho, Artur S. d’Avila Garcez, Luís C. Lamb, and Emílio Vital Brazil. Grokking explained: A statistical phenomenon. arXiv preprint arXiv:2502.01774, 2025. 3arxiv.org/abs/2502.01774
  7. [7]Kenzo Clauw, Sebastiano Stramaglia, and Daniele Marinazzo. Information-theoretic progress measures reveal grokking is an emergent phase transition. arXiv preprint arXiv:2408.08944, 2024. 3arxiv.org/abs/2408.08944
  8. [8]Ben Cullen, Sergio Estan-Ruiz, Riya Danait, and Jiayi Li. A basin-selection perspective on grokking via singular learning theory. arXiv preprint arXiv:2603.01192, 2026. 3arxiv.org/abs/2603.01192
  9. [9]Xander Davies, Lauro Langosco, and David Krueger. Unifying grokking and double descent. arXiv preprint arXiv:2303.06173, 2023. 3arxiv.org/abs/2303.06173
  10. [10]G. P. Egorychev. The solution of van der Waerden’s problem for permanents. Advances in Mathematics, 42(3):299–305, 1981. 8, 26DOI
  11. [11]Ibrahim Talha Ersoy and Karoline Wiesner. Noise-driven escape from metastable phases explains grokking in deep neural networks. arXiv preprint arXiv:2606.17120, 2026. 3arxiv.org/abs/2606.17120
  12. [12]Pavel Etingof, Oleg Golberg, Sebastian Hensel, Tiankai Liu, Alex Schwендner, Dmitry Vaintrob, and Elena Yudovina. Introduction to representation theory, 2011. URL https://math.mit.edu/~etingof/rep lect.pdf. Lecture notes, January 10. 26, 33
  13. [14]Simin Fan, Razvan Pascanu, and Martin Jaggi. Deep grokking: Would deep neural networks generalize better? arXiv preprint arXiv:2405.19454, 2024. 3, 31arxiv.org/abs/2405.19454
  14. [15]Hiroki Furuta, Gouki Minegishi, Yusuke Iwasawa, and Yutaka Matsuo. Interpreting grokked transformers in complex modular arithmetic. arXiv preprint arXiv:2402.16726, 2024. 3
  15. [16]Andrey Gromov. Grokking modular arithmetic. arXiv preprint arXiv:2301.02679, 2023. 3arxiv.org/abs/2301.02679
  16. [17]Zihan Gu, Ruoyu Chen, Hua Zhang, Yue Hu, and Xiaochun Cao. Beyond progress measures: Theoretical insights into the mechanism of grokking. arXiv preprint arXiv:2504.03162, 2025. 1, 3, 31arxiv.org/abs/2504.03162
  17. [18]Leonid Gurvits. A proof of hyperbolic van der Waerden conjecture: the right generalization is the ultimate simplification. Electronic Colloquium on Computational Complexity (ECCC), 2005. Report TR05-103; arXiv:math/0504397. 8, 26DOI
  18. [19]Tianyu He, Darshil Doshi, Aritra Das, and Andrey Gromov. Learning to grok: Emergence of in-context learning and skill composition in modular arithmetic tasks. arXiv preprint arXiv:2406.02550, 2024. 3arxiv.org/abs/2406.02550
  19. [20]Wassily Hoeffding. Probability inequalities for sums of bounded random variables. Journal of the American Statistical Association, 58(301):13–30, 1963. 19, 22, 26
  20. [21]Yufei Huang, Shengding Hu, Xu Han, Zhiyuan Liu, and Maosong Sun. Unified view of grokking, double descent and emergent abilities: A perspective from circuits competition. arXiv preprint arXiv:2402.15175, 2024. 3arxiv.org/abs/2402.15175
  21. [22]Ahmed Imtiaz Humayun, Randall Balestriero, and Richard Baraniuk. Deep networks always grok and here is why. In Int. Conf. Mach. Learn. (ICML), 2024. 3arxiv.org/abs/2402.15555
  22. [23]Hyeonbin Hwang and Yeachan Park. Intrinsic task symmetry drives generalization in algorithmic tasks. arXiv preprint arXiv:2603.01968, 2026. 3arxiv.org/abs/2603.01968
  23. [24]Daniel Kunin, Allan Raventos, Clémentine Carla Juliette Dominé, Feng Chen, David Klindt, Andrew M Saxe, and Surya Ganguli. Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learning. In High-dimensional Learning Dynamics 2024: The Emergence of Structure and Reasoning, 2024. 3arxiv.org/abs/2406.06158
  24. [25]Antoine Ledent, Waleed Mustafa, Yunwen Lei, and Marius Kloft. Norm-based generalisation bounds for deep multi-class convolutional neural networks. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 35, pp. 8279–8287, 2021. arXiv:1905.12430. 7, 25arxiv.org/abs/1905.12430
  25. [26]Jaerin Lee, Bong Gyun Kang, Kihoon Kim, and Kyoung Mu Lee. Grokfast: Accelerated grokking by amplifying slow gradients. arXiv preprint arXiv:2405.20233, 2024. 3arxiv.org/abs/2405.20233
  26. [27]Ziyue Li, Chenrui Fan, and Tianyi Zhou. Grokking in LLM pretraining? monitor memorization-to-generalization without test. arXiv preprint arXiv:2506.21551, 2025. 3arxiv.org/abs/2506.21551
  27. [28]Ziming Liu, Ouail Kitouni, Niklas S Nolte, Eric Michaud, Max Tegmark, and Mike Williams. Towards understanding grokking: An effective theory of representation learning. In Adv. Neural Inform. Process. Syst. (NeurIPS), volume 35, pp. 34651–34663, 2022. 3arxiv.org/abs/2205.10343
  28. [29]Róisín Luo, Christian Gagné, Jonas Ngnawé, Ihsan Ullah, and Karyn Morrissey. A stochastic–geometric theory of scaling laws in grokking. arXiv preprint arXiv:2606.30388, 2026. 3arxiv.org/abs/2606.30388
  29. [30]Kaifeng Lyu and Jian Li. Gradient descent maximizes the margin of homogeneous neural networks. In International Conference on Learning Representations (ICLR), 2020. 2, 3, 4arxiv.org/abs/1906.05890
  30. [31]Kaifeng Lyu, Jikai Jin, Zhiyuan Li, Simon Shaolei Du, Jason D. Lee, and Wei Hu. Dichotomy of early and late phase implicit biases can provably induce grokking. In Int. Conf. Learn. Represent. (ICLR), 2024. 3arxiv.org/abs/2311.18817
  31. [32]Neil Mallinar, Daniel Beaglehole, Libin Zhu, Adityanarayanan Radhakrishnan, Parthe Pandit, and Mikhail Belkin. Emergence in non-neural models: grokking modular arithmetic via average gradient outer product. arXiv preprint arXiv:2407.20199, 2024. 3arxiv.org/abs/2407.20199
  32. [33]Shalima Binta Manir and Anamika Paul Rupa. A systematic empirical study of grokking: Depth, architecture, activation, and regularization. arXiv preprint arXiv:2603.25009, 2026. 3, 31arxiv.org/abs/2603.25009
  33. [34]William Merrill, Nikolaos Tsilivis, and Aman Shukla. A tale of two circuits: Grokking as competition of sparse and dense subnetworks. arXiv preprint arXiv:2303.11873, 2023. 3arxiv.org/abs/2303.11873
  34. [35]Jack Miller, Patrick Gleeson, Charles O’Neill, Thang Bui, and Noam Levi. Measuring sharpness in grokking. arXiv preprint arXiv:2402.08946, 2024. 3arxiv.org/abs/2402.08946
  35. [36]Mohamad Amin Mohamadi, Zhiyuan Li, Lei Wu, and Danica J Sutherland. Why do you grok? a theoretical analysis of grokking modular addition. arXiv preprint arXiv:2407.12332, 2024. 1, 3
  36. [37]Tiberiu Muşat. The geometry of grokking: Norm minimization on the zero-loss manifold. arXiv preprint arXiv:2511.01938, 2025. 3arxiv.org/abs/2511.01938
  37. [38]Neel Nanda, Lawrence Chan, Tom Lieberum, Jess Smith, and Jacob Steinhardt. Progress measures for grokking via mechanistic interpretability. In Int. Conf. Learn. Represent. (ICLR), 2023. 1, 3arxiv.org/abs/2301.05217
  38. [39]Behnam Neyshabur, Ryota Tomioka, and Nathan Srebro. Norm-based capacity control in neural networks. In Proceedings of the 28th Conference on Learning Theory (COLT), pp. 1376–1401, 2015. 2, 3, 4
  39. [40]Yeachan Park, Minseok Kim, and Yeoneung Kim. Acceleration of grokking in learning arithmetic operations via kolmogorov-arnold representation. arXiv preprint arXiv:2405.16658, 2024. 3arxiv.org/abs/2405.16658
  40. [41]Gilles Pisier. Remarques sur un résultat non publié de B. Maurey. In Séminaire d’Analyse Fonctionnelle (Polytechnique), pp. 1–12. École Polytechnique, 1981. Exposé no. V. 20, 26
  41. [42]Alethea Power, Yuri Burda, Harri Edwards, Igor Babuschkin, and Vedant Misra. Grokking: Generalization beyond overfitting on small algorithmic datasets. arXiv preprint arXiv:2201.02177, 2022. 1arxiv.org/abs/2201.02177
  42. [43]Lucas Prieto, Melih Barsbey, Pedro A. M. Mediano, and Tolga Birdal. Grokking at the edge of numerical stability. In Int. Conf. Learn. Represent. (ICLR), 2025. arXiv:2501.04697. 3arxiv.org/abs/2501.04697
  43. [44]Jaisidh Singh, Diganta Misra, and Antonio Orvieto. Explaining grokking in transformers through the lens of inductive bias. arXiv preprint arXiv:2602.06702, 2026. 3
  44. [45]Yiding Song and Hamming Ye. Model capacity determines grokking through competing memorisation and generalisation speeds. arXiv preprint arXiv:2605.09724, 2026. 3
  45. [46]Daniel Soudry, Elad Hoffer, Mor Shpigel Nacson, Suriya Gunasekar, and Nathan Srebro. The implicit bias of gradient descent on separable data. Journal of Machine Learning Research, 19(70):1–57, 2018. 2, 3, 4arxiv.org/abs/1710.10345
  46. [47]Hu Tan, Kuo Gai, and Shihua Zhang. Deciphering two training clocks in grokking via deep linear network theory with conditional ReLU reduction. arXiv preprint arXiv:2606.05863, 2026. 3
  47. [48]Yifan Tang, Qiquan Wang, Inês García-Redondo, and Anthea Monod. Topological signatures of grokking. arXiv preprint arXiv:2605.06352, 2026. 3arxiv.org/abs/2605.06352
  48. [49]Vimal Thilak, Etai Littwin, Shuangfei Zhai, Omid Saremi, Roni Paiss, and Joshua M. Susskind. The slingshot mechanism: An empirical study of adaptive optimizers and the grokking phenomenon. In Has It Trained Yet? NeurIPS 2022 Workshop, 2022. 3arxiv.org/abs/2206.04817
  49. [50]Yuandong Tian. Provable scaling laws of feature emergence from learning dynamics of grokking. arXiv preprint arXiv:2509.21519, 2025. 3arxiv.org/abs/2509.21519
  50. [51]Pascal Jr Tikeng Notsawo, Guillaume Dumas, and Guillaume Rabusseau. Grokking beyond the Euclidean norm of model parameters. In Int. Conf. Mach. Learn. (ICML), 2025. arXiv:2506.05718. 3arxiv.org/abs/2506.05718
  51. [52]Xuan Khanh Truong, Quynh Hoa Truong, Duc Trung Luu, and Thanh Duc Phan. The norm-separation delay law of grokking: A first-principles theory of delayed generalization. arXiv preprint arXiv:2603.13331, 2026. 3
  52. [53]J. H. van Lint and R. M. Wilson. A Course in Combinatorics. Cambridge University Press, 2nd edition, 2001. 8
  53. [54]Vikrant Varma, Rohin Shah, Zachary Kenton, János Kramár, and Ramana Kumar. Explaining grokking through circuit efficiency. arXiv preprint arXiv:2309.02390, 2023. 3arxiv.org/abs/2309.02390
  54. [55]Thomas Walker, Ahmed Imtiaz Humayun, Randall Balestriero, and Richard Baraniuk. GrokAlign: Geometric characterisation and acceleration of grokking. arXiv preprint arXiv:2506.12284, 2025. 3arxiv.org/abs/2506.12284
  55. [56]Boshi Wang, Xiang Yue, Yu Su, and Huan Sun. Grokked transformers are implicit reasoners: A mechanistic journey to the edge of generalization. arXiv preprint arXiv:2405.15071, 2024. 3arxiv.org/abs/2405.15071
  56. [57]Hugh E. Warren. Lower bounds for approximation by nonlinear manifolds. Transactions of the American Mathematical Society, 133(1):167–178, 1968. 26
  57. [58]Mingyue Xu, Gal Vardi, and Itay Safran. To grok grokking: Provable grokking in ridge regression. In Proc. Int. Conf. Mach. Learn. (ICML), 2026. arXiv:2601.19791. 3arxiv.org/abs/2601.19791
  58. [59]Zhiwei Xu, Zhiyu Ni, Yixin Wang, and Wei Hu. Let me grok for you: Accelerating grokking via embedding transfer from a weaker model. In Int. Conf. Learn. Represent. (ICLR), 2025. arXiv:2504.13292. 3arxiv.org/abs/2504.13292
  59. [61]David Yunis, Kumar Kshitij Patel, Samuel Wheeler, Pedro Savarese, Gal Vardi, Karen Livescu, Michael Maire, and Matthew R Walter. Approaching deep learning through the spectral dynamics of weights. arXiv preprint arXiv:2408.11804, 2024. 3arxiv.org/abs/2408.11804
  60. [62]Tong Zhang. Covering number bounds of certain regularized linear function classes. Journal of Machine Learning Research, 2:527–550, 2002. 7, 25
  61. [63]Zhangchen Zhou, Yaoyu Zhang, and Zhi-Qin John Xu. A rationale from frequency perspective for grokking in training neural network. arXiv preprint arXiv:2405.17479, 2024. 3arxiv.org/abs/2405.17479
  62. [64]Xuekai Zhu, Yao Fu, Bowen Zhou, and Zhouhan Lin. Critical data size of language models from a grokking perspective. arXiv preprint arXiv:2401.10463, 2024. 3arxiv.org/abs/2401.10463

Paper details

vlnc:W_anv4j92z
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Not recorded for this paper · View statement →

Looking for related papers…
Contents