Optimal scaling laws in learning hierarchical multi-index models

Read full text
Authors: Leonardo Defilippis, Florent Krząkała, Bruno Loureiro, Antoine Maillard
Cited by: 0

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.

Fields of Study & Subjects

Stochastic Gradient Optimization Techniques Neural Networks and Applications Machine Learning and Algorithms

Publication details

Journal / source
arXiv (Cornell University)
Publication date
5 February 2026
Language
EN
Document type
Preprint