paper

Quantifying the Memorization-to-Generalization Transition: Scaling Laws and Phase Structure in Grokking

arXiv:2609.10657

Abstract

Neural networks trained past memorization frequently undergo a delayed transition to generalization, a phenomenon known as grokking. Despite theoretical progress on \emph{why} this transition occurs, the quantitative structure of \emph{when} it occurs in hyperparameter space remains uncharacterized. We map the memorization-to-generalization boundary across 384 configurations of two-hidden-layer MLPs on modular arithmetic, fitting a power-law scaling relation for generalization onset time: (; with interactions). The exponent hierarchy reveals that data complexity () is the dominant driver of regime transition, not model capacity (): doubling data accelerates generalization by , while doubling width yields only . A sharp phase boundary at weight decay separates grokking from non-grokking configurations, and weight norm trajectories show monotonic compression during the transition, consistent with implicit regularization selecting low-complexity solutions. These results provide a quantitative foundation for predicting and controlling regime transitions in overparameterized networks.

Quantifying the Memorization-to-Generalization Transition: Scaling Laws and Phase Structure in Grokking · wovepaper