Proving the t^(−2β/(1+β)) feature-learning loss exponent for hard tasks under power-law kernel spectra
Difficulty 47: the target is a single explicit exponent with a precise heuristic derivation, and 2025 work already proves feature-learning scaling laws on isotropic data, but no technique controls moving features against a power-law input spectrum; it sits below CDCL proof search (54) and above the smoothed simplex exponent (38). Application value 38, below the retired broad neural-scaling-laws entry (55): a proof would justify compute-allocation rules in the feature-learning regime, but practice relies on fitted exponents. Leverage 40: the methods (high-dimensional SGD with evolving kernels) are reusable across training-dynamics theory, but few other results are stated conditionally on this exponent.
The problem
Can we prove how fast a network's error shrinks when it learns its own features on hard tasks?
- 1Bigger training, error falls as power law
- 2Lazy networks: provable rate
- 3Feature learning predicted to be faster
- 4Prediction rests on physics arguments
- 5Open: rigorous proof for real setting
Background. Empirically, the loss of large models falls as a power law in training time, data or model size. A solvable setting: take the infinite-width neural tangent kernel (NTK), the kernel that governs training when the network barely moves its features. Suppose its eigenvalues decay like and the target function has coefficients satisfying a with exponent . The task is called when , meaning the target lies outside the function space the initial kernel can easily represent. In the lazy (kernel) limit, training for time with unlimited width and data gives loss .
Path to understanding
foundations → problemThe research entries this problem rests on, in the order to read them. Each links to its explainer.
- Start with
- Step 2
- Fast matrix multiplicationAlgorithm
- Step 3
- The transformer architectureAlgorithm · host
- Then
- Proving the t^(−2β/(1+β)) feature-learning loss exponent for hard tasks under power-law kernel spectraThe problem
State of the art
Proved in the lazy and linear regimes. Kernel ridge regression with power-law spectra has minimax-optimal rates (Caponnetto and De Vito, 2007). For random-feature models trained by one-pass SGD with power-law covariance, Paquette, Paquette, Xiao and Pennington (arXiv:2405.15074, 2024) prove loss asymptotics and identify the phases of compute-optimal scaling; Lin, Wu, Kakade, Bartlett and Lee (arXiv:2406.08466, 2024) prove data, parameter and compute scaling for SGD on linear regression with power-law spectra.
Proved with feature learning, but on isotropic data. Ren, Nichani, Wu and Lee (arXiv:2504.19983, NeurIPS 2025) analyse online SGD for a two-layer student learning with , orthonormal , an even activation of information exponent above 2 and ; they prove sharp recovery times for each direction and derive scaling exponents in samples, SGD steps and width. Ben Arous, Erdogdu, Vural and Wu (arXiv:2508.03688, NeurIPS 2025) prove scaling laws for quadratic-activation networks with extensive width and power-law second-layer coefficients. In both, the power law comes from the target's coefficient decay over orthogonal directions of isotropic data, so the hard-task mechanism above (feature learning against a power-law input spectrum) is not covered.
Realistic applications
Direct benefit is limited: a proof would tell practitioners which exponents depend on architecture-level feature learning and which on data spectra, informing compute-allocation rules, but practical scaling-law fits will remain empirical. The methods (high-dimensional SGD dynamics with evolving kernels) are reusable in the theory of training dynamics.
What counts as solved
Resolved by a proof of (ii): matching upper and lower bounds on the loss exponent of a nonlinear two-layer network trained by online SGD in the mean-field parametrisation on power-law-spectrum data, with the hard-task exponent identified explicitly (whether or not it equals ). A rigorous proof of (i) for the Bordelon–Atanasov–Pehlevan model alone, or a rigorous counterexample to its predicted exponent, is major partial progress.
How it gets verified
Expert reviewOnly expert review of the argument is possible today.
The target is a theorem about online SGD for a two-layer network in the mean-field parametrisation on power-law data, with matching upper and lower loss-exponent bounds. Mathlib has no SGD dynamics or mean-field limits, so it cannot be formalised today. Probabilistic-analysis experts review the proof; numerical simulations of the exponent can support it but do not count as resolution.
Claims under verification
+ add a claimNo claimed results on the Verification board for this problem yet.
Where the work is
This is a theorem-proving task in high-dimensional probability; numerical experiments only guide conjectures.
Bottlenecks
ranked by votes- #1TheoryMoving features against an anisotropic power-law spectrum
Rigorous feature-learning analyses (Ren et al.; Ben Arous et al.) rely on isotropic inputs and orthogonal teacher directions, which decouple the directions. With a power-law input spectrum, the first layer must align with infinitely many coupled eigendirections at different rates, and Gaussian-equivalence and random-matrix tools for fixed features no longer apply.
Community grades
Sign in to add your own grade.
Connected problems
Progress
- Apr 28, 2025Rigorous feature-learning scaling laws on isotropic datasource
Ren, Nichani, Wu and Lee prove scaling exponents for online SGD on two-layer networks learning power-law additive targets with isotropic Gaussian inputs (NeurIPS 2025).
- Sep 26, 2024DMFT prediction of the hard-task exponentsource
Bordelon, Atanasov and Pehlevan predict that feature learning improves the time exponent from β to 2β/(1+β) for hard tasks (non-rigorous; ICLR 2025).
- Jun 12, 2024Scaling laws proved for SGD linear regressionsource
Lin et al. give matching upper and lower bounds in data, parameters and compute for power-law spectra (lazy/linear setting).
- Jan 23, 2020Kaplan et al. power-law scalingsource
Loss follows power laws in parameters, data and compute over several orders of magnitude.