Sharp feature-learning transitions and Bayes-optimal neural scaling laws in extensive-width networks
Not on the map. Why?
Abstract (arXiv)
We study the information-theoretic limits of learning a one-hidden-layer teacher network with hierarchical features from noisy queries, in the context of knowledge transfer to a smaller student model. We work in the high-dimensional regime where the teacher width k scales linearly with the input dimension d -- a setting that captures large-but-finite-width networks and has only recently become analytically tractable. Using a heuristic leave-one-out decoupling argument, validated numerically throughout, we derive asymptotically sharp characterizations of the Bayes-optimal generalization error and individual feature overlaps via a system of closed fixed-point equations. These equations reveal that feature learnability is governed by a sequence of sharp phase transitions: as data grows, teacher features become recoverable sequentially, each through a discontinuous jump in overlap. This sequential acquisition underlies a precise notion of textiteffective width kc -- the number of learnable features at a given data budget n -- which unifies two distinct scaling regimes: a feature-learning regime in which the Bayes-optimal generalization error varepsilonrm BO scales as n1/(2β)-1, and a refinement regime in which it scales as n-1, where β>1/2 is the exponent of the power-law feature hierarchy. Both laws collapse to the single relation varepsilonrm BO=Theta(kc d/n). We further show empirically that a student trained with textscAdam near the effective width kc achieves these optimal scaling laws (up to a small algorithmic gap), and provide an information-theoretic account of the associated scaling in model size.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 34 of its 39 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
34 of 39 references.
References · 34
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2020Knowledge Distillation: A Surveycited by 1,603
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 582
- 2021Explaining Neural Scaling Lawscited by 363
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2019Linearized two-layers neural networks in high dimensioncited by 155
- 2023SGD learning on neural networks: leap complexity and saddle-to-saddle dynamicscited by 143
- 2023The Quantization Model of Neural Scalingcited by 138
22 more
- 2020Online stochastic gradient descent on non-convex losses from high-dimensional inferencecited by 119
- 2017Optimal Errors and Phase Transitions in High-Dimensional Generalized Linear Modelscited by 116
- 2024A Dynamical Model of Neural Scaling Lawscited by 111
- 2021Generalization Error Rates in Kernel Regression: The Crossover from the Noiseless to Noisy Regimecited by 96
- 2021Learning Curve Theorycited by 85
- 20244+3 Phases of Compute-Optimal Neural Scaling Lawscited by 68
- 2023Smoothing the Landscape Boosts the Signal for SGD: Optimal Sample Complexity for Learning Single Index Modelscited by 61
- 2018The committee machine: Computational to statistical gaps in learning a two-layers neural networkcited by 58
- 2024Scaling Laws in Linear Regression: Compute, Parameters, and Datacited by 57
- 2024Computational-Statistical Gaps in Gaussian Single-Index Modelscited by 44
- 2025Emergence and scaling laws in SGD learning of shallow neural networkscited by 32
- 2025Learning quadratic neural networks in high dimensions: SGD dynamics and scaling lawscited by 25
- 2024Dimension-free deterministic equivalents and scaling laws for random feature regressioncited by 23
- 2024Bayes-optimal learning of an extensive-width neural network from quadratically many samplescited by 23
- 2024An exactly solvable model for emergence and scaling laws in the multitask sparse parity problemcited by 22
- 2025Scaling Laws and Spectra of Shallow Neural Networks in the Feature Learning Regimecited by 20
- 2024Fundamental computational limits of weak learnability in high-dimensional multi-index modelscited by 19
- 2025Learning curves theory for hierarchically compositional data with power-law distributed featurescited by 17
- 2025Statistical mechanics of extensive-width Bayesian neural networks near interpolationcited by 9
- 2026Optimal scaling laws in learning hierarchical multi-index modelscited by 9
- 2025Information-theoretic reduction of deep neural networks to linear models in the overparametrized proportional regimecited by 6
- 2026A Noise Sensitivity Exponent Controls Large Statistical-to-Computational Gaps in Single- and Multi-Index Modelscited by 2
Built on by
Not cited yet.