JMLR

High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networks

Authors
Francis Bach Simon Martin Giulio Biroli
Research Topics
Machine Learning High-Dimensional Statistics
Paper Information
  • Journal:
    Journal of Machine Learning Research
  • Added to Tracker:
    Jul 06, 2026
Abstract

We study the high-dimensional training dynamics of a shallow neural network with quadratic activation in a teacher--student setup. We focus on the extensive-width regime, where the teacher and student network widths scale proportionally with the input dimension, and the sample size grows quadratically. This scaling aims to describe overparameterized neural networks in which feature learning still plays a central role. In the high-dimensional limit, we derive a dynamical characterization of the gradient flow, in the spirit of dynamical mean-field theory (DMFT). Under $\ell_2$-regularization, we analyze these equations at long times and characterize the performance and spectral properties of the resulting estimator. This result provides a quantitative understanding of the effect of overparameterization on learning and generalization, and reveals a double descent phenomenon in the presence of label noise, where generalization improves beyond interpolation. In the small regularization limit, we obtain an exact expression for the perfect recovery threshold as a function of the network widths, providing a precise characterization of how overparameterization influences recovery.

Author Details
Francis Bach
Author
Simon Martin
Author
Giulio Biroli
Author
Research Topics & Keywords
Machine Learning
Research Area
High-Dimensional Statistics
Research Area
Citation Information
APA Format
Francis Bach , Simon Martin & Giulio Biroli . High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networks. Journal of Machine Learning Research .
BibTeX Format
@article{paper1361,
  title = { High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networks },
  author = { Francis Bach and Simon Martin and Giulio Biroli },
  journal = { Journal of Machine Learning Research },
  url = { https://www.jmlr.org/papers/v27/26-0672.html }
}