arXiv · 2412.09810
The Complexity Dynamics of Grokking
Abstract
We demonstrate the existence of a complexity phase transition in neural networks by studying the grokking phenomenon, where networks suddenly transition from memorization to generalization long after overfitting their training data. To characterize this phase transition, we introduce a theoretical framework for measuring complexity based on rate-distortion theory and Kolmogorov complexity, which can be understood as principled lossy compression for networks. We find that properly regularized networks exhibit a sharp phase transition: complexity rises during memorization, then falls as the network discovers a simpler underlying pattern that generalizes. In contrast, unregularized networks remain trapped in a high-complexity memorization phase. We establish an explicit connection between our complexity measure and generalization bounds, providing a theoretical foundation for the link between lossy compression and generalization. Our framework achieves compression ratios 30-40x better than na\"ive approaches, enabling precise tracking of complexity dynamics. Finally, we introduce a regularization method based on spectral entropy that encourages networks toward low-complexity representations by penalizing their intrinsic dimension.
Explore related subjects
Keep this discovery
Branton DeMoss, Silvia Sapora, Jakob Foerster, Nick Hawes, Ingmar Posner. 2024-12-13. The Complexity Dynamics of Grokking. https://doi.org/10.1016/j.physd.2025.134859
Cite the original work for its findings. Save a collection to share your selection of sources.