High-dimensional learning of narrow neural networks
Not on the map. Why?
Abstract (arXiv)
Recent years have been marked with the fast-pace diversification and increasing ubiquity of machine learning applications. Yet, a firm theoretical understanding of the surprising efficiency of neural networks to learn from high-dimensional data still proves largely elusive. In this endeavour, analyses inspired by statistical physics have proven instrumental, enabling the tight asymptotic characterization of the learning of neural networks in high dimensions, for a broad class of solvable models. This manuscript reviews the tools and ideas underlying recent progress in this line of work. We introduce a generic model -- the sequence multi-index model -- which encompasses numerous previously studied models as special instances. This unified framework covers a broad class of machine learning architectures with a finite number of hidden units, including multi-layer perceptrons, autoencoders, attention mechanisms; and tasks, including (un)supervised learning, denoising, contrastive learning, in the limit of large data dimension, and comparably large number of samples. We explicate in full detail the analysis of the learning of sequence multi-index models, using statistical physics techniques such as the replica method and approximate message-passing algorithms. This manuscript thus provides a unified presentation of analyses reported in several previous works, and a detailed overview of central techniques in the field of statistical physics of machine learning. This review should be a useful primer for machine learning theoreticians curious of statistical physics approaches; it should also be of value to statistical physicists interested in the transfer of such ideas to the study of neural networks.
Key facts. Cited by 12 papers in the map (data through 14 Oct 2026). Builds on 100 of its 197 references that are in the map. First posted on arXiv 20 Sep 2024.
Builds on
100 of 197 references.
References · 100
- 2017Attention Is All You Needcited by 60,361
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2019The generalization error of random features regression: Precise asymptotics and double descent curvecited by 413
38 more
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
- 2019Mean-field theory of two-layers neural networks: dimension-free bounds and kernel limitcited by 187
- 2018A mathematical theory of semantic development in deep neural networkscited by 186
- 2022High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representationcited by 150
- 2015High-Dimensional Asymptotics of Prediction: Ridge Regression and Classificationcited by 148
- 2015Statistical physics of inference: Thresholds and algorithmscited by 128
- 2020Generalisation error in learning with random features and the hidden manifold modelcited by 127
- 2021Learning curves of generic features maps for realistic datasets with a teacher-student modelcited by 126
- 2017Optimal Errors and Phase Transitions in High-Dimensional Generalized Linear Modelscited by 116
- 2020The Gaussian equivalence of generative models for learning with shallow neural networkscited by 113
- 2019Dynamics of stochastic gradient descent for two-layer neural networks in the teacher-student setupcited by 112
- 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
- 2022Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networkscited by 105
- 2021Generalization Error Rates in Kernel Regression: The Crossover from the Noiseless to Noisy Regimecited by 96
- 2017A Random Matrix Approach to Neural Networkscited by 80
- 2016Understanding Deep Convolutional Networkscited by 76
- 2022Universality of empirical risk minimizationcited by 75
- 2020Statistical Mechanics of Deep Linear Neural Networks: The Back-Propagating Kernel Renormalizationcited by 71
- 2020A Random Matrix Analysis of Random Fourier Features: Beyond the Gaussian Kernel, a Precise Phase Transition, and the Corresponding Double Descentcited by 70
- 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 69
- 2023How Two-Layer Neural Networks Learn, One (Giant) Step at a Timecited by 68
- 2023Depthwise Hyperparameter Transfer in Residual Networks: Dynamics and Scaling Limitcited by 66
- 2020The role of regularization in classification of high-dimensional noisy Gaussian mixturecited by 63
- 2017Classification and Geometry of General Perceptual Manifoldscited by 62
- 2020Dynamical mean-field theory for stochastic gradient descent in Gaussian mixture classificationcited by 60
- 2018The committee machine: Computational to statistical gaps in learning a two-layers neural networkcited by 58
- 2024Asymptotic theory of in-context learning by linear attentioncited by 55
- 2021Learning Gaussian Mixtures with Generalised Linear Models: Precise Asymptotics in High-dimensionscited by 54
- 2021Separation of Scales and a Thermodynamic Description of Feature Learning in Some CNNscited by 53
- 2018Entropy and mutual information in models of deep neural networkscited by 51
- 2020Generalization error in high-dimensional perceptrons: Approaching Bayes error with convex optimizationcited by 51
- 2022Neural networks trained with SGD learn distributions of increasing complexitycited by 50
- 2023A Theory of Non-Linear Feature Learning with One Gradient Step in Two-Layer Neural Networkscited by 49
- 2018The phase transition for the existence of the maximum likelihood estimate in high-dimensional logistic regressioncited by 46
- 2020A Precise High-Dimensional Asymptotic Theory for Boosting and Minimum-ℓ1-Norm Interpolated Classifierscited by 44
- 2023Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networkscited by 43
- 2023Bayes-optimal Learning of Deep Random Networks of Extensive-widthcited by 41
- 2016Unreasonable Effectiveness of Learning Neural Networks: From Accessible States and Robust Ensembles to Basic Algorithmic Schemescited by 40
50 most cited of 100 All 100 (JSON)
Built on by
Citing papers per year: 2025: 10; 2026: 2.
12 papers.
- 2025A solvable model of learning generative diffusion: theory and insightscited by 14
- 2025Asymptotics of SGD in Sequence-Single Index Models and Single-Layer Attention Networkscited by 9
- 2025Preventing Model Collapse Under Overparametrization: Optimal Mixing Ratios for Interpolation Learning and Ridge Regressioncited by 9
- 2025Statistical Advantage of Softmax Attention: Insights from Single-Location Regressioncited by 7
- 2025Mitigating the Curse of Detail: Scaling Arguments for Feature Learning and Sample Complexitycited by 4
- 2025Bayes optimal learning of attention-indexed modelscited by 3
- 2025A statistical physics framework for optimal learningcited by 3
- 2026High-Dimensional Theory of LoRA Fine-Tuning in a Solvable Attention Modelcited by 3
- 2025Generalization performance of narrow one-hidden layer networks in the teacher-student settingcited by 2
- 2025Deep Neural Nets as Hamiltonianscited by 0
- 2025Minimax Rates for Learning Pairwise Interactions in Attention-Style Modelscited by 0
- 2026Feature Learning in Linear-Width Two-Layer Networks: Two vs. One Step of Gradient Descentcited by 0
- 2026High-Dimensional Theory of LoRA Fine-Tuning in a Solvable Attention Model4 Jun 2026
- 2026Feature Learning in Linear-Width Two-Layer Networks: Two vs. One Step of Gradient Descent18 May 2026
- 2025Mitigating the Curse of Detail: Scaling Arguments for Feature Learning and Sample Complexity3 Dec 2025
- 2025Minimax Rates for Learning Pairwise Interactions in Attention-Style Models13 Oct 2025
- 2025Preventing Model Collapse Under Overparametrization: Optimal Mixing Ratios for Interpolation Learning and Ridge Regression26 Sep 2025
- 2025Statistical Advantage of Softmax Attention: Insights from Single-Location Regression26 Sep 2025
- 2025A statistical physics framework for optimal learning10 Jul 2025
- 2025Generalization performance of narrow one-hidden layer networks in the teacher-student setting1 Jul 2025
- 2025Asymptotics of SGD in Sequence-Single Index Models and Single-Layer Attention Networks3 Jun 2025
- 2025Bayes optimal learning of attention-indexed models2 Jun 2025
- 2025Deep Neural Nets as Hamiltonians31 Mar 2025
- 2025A solvable model of learning generative diffusion: theory and insights7 Jan 2025