English

Optimal scaling laws in learning hierarchical multi-index models

Machine Learning 2026-02-06 v1 Machine Learning

Abstract

In this work, we provide a sharp theory of scaling laws for two-layer neural networks trained on a class of hierarchical multi-index targets, in a genuinely representation-limited regime. We derive exact information-theoretic scaling laws for subspace recovery and prediction error, revealing how the hierarchical features of the target are sequentially learned through a cascade of phase transitions. We further show that these optimal rates are achieved by a simple, target-agnostic spectral estimator, which can be interpreted as the small learning-rate limit of gradient descent on the first-layer weights. Once an adapted representation is identified, the readout can be learned statistically optimally, using an efficient procedure. As a consequence, we provide a unified and rigorous explanation of scaling laws, plateau phenomena, and spectral structure in shallow neural networks trained on such hierarchical targets.

Keywords

Cite

@article{arxiv.2602.05846,
  title  = {Optimal scaling laws in learning hierarchical multi-index models},
  author = {Leonardo Defilippis and Florent Krzakala and Bruno Loureiro and Antoine Maillard},
  journal= {arXiv preprint arXiv:2602.05846},
  year   = {2026}
}
R2 v1 2026-07-01T10:22:47.151Z