High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networks
Authors
Research Topics
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
AuthorSimon Martin
AuthorGiulio Biroli
AuthorResearch Topics & Keywords
Machine Learning
Research AreaHigh-Dimensional Statistics
Research AreaCitation 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 }
}