Search papers, labs, and topics across Lattice.
This study investigates the phenomenon of grokking in deep neural networks (DNNs) by analyzing first-order phase transitions induced by L2 regularization strength. The authors demonstrate that networks can become trapped in metastable states, which can only be escaped through stochastic noise from stochastic gradient descent (SGD), leading to delayed generalization. Key findings reveal that the number of metastable states corresponds to the number of learnable features, suggesting that task complexity enhances the potential for hysteresis in learning dynamics.
Grokking in DNNs is driven by noise-induced escapes from metastable states, revealing a surprising connection between regularization strength and the emergence of learnable features.
Deep neural networks (DNNs) exhibit first order phase transitions under variations of the L2 regularization strength, with each transition marking the onset of a new learnable feature. Below a critical regularization strength, all features are in principle learnable, but coexisting metastable states, separated by energy barriers, can trap the network and impede convergence. A strength of DNNs is their ability to generalize. But many open questions remain, among them the origin of so called grokking: the abrupt, delayed onset of generalization after prolonged apparent overfitting. We show for linear DNNs that grokking is consistent with hysteresis in first-order L2 phase transitions: using L2 regularization to engineer deliberate trapping, we demonstrate that a model in a low-accuracy metastable state escapes only when SGD noise drives it across an energy barrier, with escape times following Arrhenius scaling. We reproduce grokking-like delayed convergence across two orders of magnitude in escape time by deliberately trapping models in metastable phases. Using sparse sub-sampling we also reproduce the canonical grokking curve where test error eventually approaches the final training error. Our work suggests that the number of metastable states equals the number of learnable features -- one per singular value of the data covariance -- the potential for hysteresis grows naturally with task complexity. We provide evidence that the same mechanism likely operates in general nonlinear DNNs. Our results provide routes toward more efficient learning schemes.
