A Function-Space Approach to the Statistical Mechanics of Learning Dynamics
Not on the map. Why?
Abstract (arXiv)
In the kernel regime, neural-network learning inherits its preferences from a frozen spectrum. During feature learning, this spectrum evolves, yet networks retain systematic biases toward simple, smooth directions. We develop a function-space statistical framework explaining the origin of these preferences, treating functions and their learning operators as macroscopic variables, with parameterization entering through the multiplicity of parameter configurations realizing each function. For mean-squared loss, error relaxes exactly under the evolving learning operator M=JJ^ast. Training stochasticity induces a Gaussian weight over function-space states, while parameter multiplicity contributes an entropic operator B, defined by the curvature of its log multiplicity. A local Laplace expansion yields the fluctuation free energy Phifluc(M;B)=fracσ_xi22logdet(M-1+B)+const, analogous to an Occam factor. Under mild statistical conditions, this free energy is rotationally stationary exactly when [M,B]=0, is minimized by pairing large eigenvalues of M with small eigenvalues of B, and generates a local restoring force against mismatch. Learning is therefore biased toward faster relaxation along entropically cheaper directions. This preference strengthens with training noise and vanishes in the deterministic limit, beyond gradient-flow accounts of operator alignment. For ReLU networks, we relate entropic curvature to the minimal rearrangement of activation boundaries required for a functional change and bound this structural cost by directional smoothness. Consequently, smooth directions are preferentially learned faster, in a data-adaptive manner, even as the learning operator evolves.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 51 references that are in the map. First posted on arXiv 9 Sep 2026.
Builds on
28 of 51 references.
References · 28
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 566
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 306
- 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 300
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 230
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representationscited by 185
- 2023The Quantization Model of Neural Scalingcited by 138
16 more
- 2017Stochastic gradient descent performs variational inference, converges to limit cycles for deep networkscited by 109
- 2020Statistical Mechanics of Deep Linear Neural Networks: The Back-Propagating Kernel Renormalizationcited by 71
- 2016A Variational Analysis of Stochastic Gradient Algorithmscited by 57
- 2024How Feature Learning Can Improve Neural Scaling Lawscited by 56
- 2023Grokking as a First Order Phase Transition in Two Layer Networkscited by 48
- 2016Unreasonable Effectiveness of Learning Neural Networks: From Accessible States and Robust Ensembles to Basic Algorithmic Schemescited by 40
- 2018A representer theorem for deep neural networkscited by 38
- 2024Why Do You Grok? A Theoretical Analysis of Grokking Modular Additioncited by 36
- 2021Unveiling the structure of wide flat minima in neural networkscited by 28
- 2023Local Kernel Renormalization as a mechanism for feature learning in overparametrized Convolutional Neural Networkscited by 19
- 2025Adaptive kernel predictors from feature-learning infinite limits of neural networkscited by 13
- 2024Feature learning as alignment: a structural property of gradient descent in non-linear neural networkscited by 10
- 2026Spectral Dynamics in Deep Networks: Feature Learning, Outlier Escape, and Learning Rate Transfercited by 6
- 2026How does feature learning reshape the function space?cited by 1
- 2026Kernel Renormalization in Bayesian Deep Neural Networks: the Equivalent Wishart Ansatz in the Proportional Regimecited by 1
- 2026A Commutator Framework for Selective Spectral Alignment in Deep Neural Networkscited by 1
Built on by
Not cited yet.