mlsanityData to 14 Oct 2026

High-dimensional learning of narrow neural networks

arXiv 2409.13904 ↗First posted 20 Sep 2024stat.ML cond-mat.dis-nn cs.LGcited by 12

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  3. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  4. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  5. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  6. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  7. 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
  8. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
  9. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
  10. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  11. 2018On Lazy Training in Differentiable Programmingcited by 675
  12. 2019The generalization error of random features regression: Precise asymptotics and double descent curvecited by 413
38 more
  1. 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
  2. 2019Mean-field theory of two-layers neural networks: dimension-free bounds and kernel limitcited by 187
  3. 2018A mathematical theory of semantic development in deep neural networkscited by 186
  4. 2022High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representationcited by 150
  5. 2015High-Dimensional Asymptotics of Prediction: Ridge Regression and Classificationcited by 148
  6. 2015Statistical physics of inference: Thresholds and algorithmscited by 128
  7. 2020Generalisation error in learning with random features and the hidden manifold modelcited by 127
  8. 2021Learning curves of generic features maps for realistic datasets with a teacher-student modelcited by 126
  9. 2017Optimal Errors and Phase Transitions in High-Dimensional Generalized Linear Modelscited by 116
  10. 2020The Gaussian equivalence of generative models for learning with shallow neural networkscited by 113
  11. 2019Dynamics of stochastic gradient descent for two-layer neural networks in the teacher-student setupcited by 112
  12. 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
  13. 2022Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networkscited by 105
  14. 2021Generalization Error Rates in Kernel Regression: The Crossover from the Noiseless to Noisy Regimecited by 96
  15. 2017A Random Matrix Approach to Neural Networkscited by 80
  16. 2016Understanding Deep Convolutional Networkscited by 76
  17. 2022Universality of empirical risk minimizationcited by 75
  18. 2020Statistical Mechanics of Deep Linear Neural Networks: The Back-Propagating Kernel Renormalizationcited by 71
  19. 2020A Random Matrix Analysis of Random Fourier Features: Beyond the Gaussian Kernel, a Precise Phase Transition, and the Corresponding Double Descentcited by 70
  20. 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 69
  21. 2023How Two-Layer Neural Networks Learn, One (Giant) Step at a Timecited by 68
  22. 2023Depthwise Hyperparameter Transfer in Residual Networks: Dynamics and Scaling Limitcited by 66
  23. 2020The role of regularization in classification of high-dimensional noisy Gaussian mixturecited by 63
  24. 2017Classification and Geometry of General Perceptual Manifoldscited by 62
  25. 2020Dynamical mean-field theory for stochastic gradient descent in Gaussian mixture classificationcited by 60
  26. 2018The committee machine: Computational to statistical gaps in learning a two-layers neural networkcited by 58
  27. 2024Asymptotic theory of in-context learning by linear attentioncited by 55
  28. 2021Learning Gaussian Mixtures with Generalised Linear Models: Precise Asymptotics in High-dimensionscited by 54
  29. 2021Separation of Scales and a Thermodynamic Description of Feature Learning in Some CNNscited by 53
  30. 2018Entropy and mutual information in models of deep neural networkscited by 51
  31. 2020Generalization error in high-dimensional perceptrons: Approaching Bayes error with convex optimizationcited by 51
  32. 2022Neural networks trained with SGD learn distributions of increasing complexitycited by 50
  33. 2023A Theory of Non-Linear Feature Learning with One Gradient Step in Two-Layer Neural Networkscited by 49
  34. 2018The phase transition for the existence of the maximum likelihood estimate in high-dimensional logistic regressioncited by 46
  35. 2020A Precise High-Dimensional Asymptotic Theory for Boosting and Minimum-ℓ1-Norm Interpolated Classifierscited by 44
  36. 2023Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networkscited by 43
  37. 2023Bayes-optimal Learning of Deep Random Networks of Extensive-widthcited by 41
  38. 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

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 10; 2026: 2.

12 papers.