mlsanityData to 7 Oct 2026

The Training Process of Many Deep Networks Explores the Same Low-Dimensional Manifold

arXiv 2305.01604 ↗First posted 2 May 2023cs.LG cond-mat.dis-nncited by 20

Not on the map. Why?

Abstract (arXiv)

We develop information-geometric techniques to analyze the trajectories of the predictions of deep networks during training. By examining the underlying high-dimensional probabilistic models, we reveal that the training process explores an effectively low-dimensional manifold. Networks with a wide range of architectures, sizes, trained using different optimization methods, regularization techniques, data augmentation techniques, and weight initializations lie on the same manifold in the prediction space. We study the details of this manifold to find that networks with different architectures follow distinguishable trajectories but other factors have a minimal influence; larger networks train along a similar manifold as that of smaller networks, just faster; and networks initialized at very different parts of the prediction space converge to the solution along a similar manifold.

Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Builds on 39 of its 85 references that are in the map. First posted on arXiv 2 May 2023.

Builds on

39 of 85 references.

References · 39

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  4. 2020Language Models are Few-Shot Learnerscited by 32,199
  5. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  6. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  7. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
  8. 2016Layer Normalizationcited by 5,579
  9. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
  10. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  11. 2016Wide Residual Networkscited by 3,057
  12. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
27 more
  1. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  2. 2016Understanding deep learning requires rethinking generalizationcited by 1,657
  3. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
  4. 2014Striving for Simplicity: The All Convolutional Netcited by 983
  5. 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
  6. 2019Benign Overfitting in Linear Regressioncited by 544
  7. 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
  8. 2017Opening the Black Box of Deep Neural Networks via Informationcited by 494
  9. 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
  10. 2019Making Convolutional Networks Shift-Invariant Againcited by 390
  11. 2019A Baseline for Few-Shot Image Classificationcited by 298
  12. 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 267
  13. 2021Deep learning: a statistical viewpointcited by 239
  14. 2017Emergence of Invariance and Disentanglement in Deep Representationscited by 213
  15. 2022Patches Are All You Need?cited by 201
  16. 2018Just Interpolate: Kernel "Ridgeless" Regression Can Generalizecited by 190
  17. 2019Mean-field theory of two-layers neural networks: dimension-free bounds and kernel limitcited by 187
  18. 2018A mathematical theory of semantic development in deep neural networkscited by 182
  19. 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 130
  20. 2017Stochastic gradient descent performs variational inference, converges to limit cycles for deep networkscited by 108
  21. 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 67
  22. 2019Let's Agree to Agree: Neural Networks Share Classification Order on Real Datasetscited by 55
  23. 2019A Purely Algebraic Justification of the Kabsch-Umeyama Algorithmcited by 24
  24. 2021Does the Data Induce Capacity Control in Deep Learning?cited by 13
  25. 2021On the interplay between data structure and loss function in classification problemscited by 12
  26. 2018PCA of high dimensional random walks with comparison to neural network trainingcited by 11
  27. 2014Weighted principal component analysis: a weighted covariance eigendecomposition approachcited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2023: 2; 2024: 6; 2025: 3; 2026: 9.

20 papers.

Sort:

By area:

  1. 2025The Universal Weight Subspace Hypothesiscited by 22
  2. 2025Dynamics of Transient Structure in In-Context Linear Regression Transformerscited by 12
  3. 2024Density estimation with LLMs: a geometric investigation of in-context learning trajectoriescited by 6
  4. 2023Learning Capacity: A Measure of the Effective Dimensionality of a Modelcited by 4
  5. 2024Many Perception Tasks are Highly Redundant Functions of their Input Datacited by 4
  6. 2026Narrow Fine-Tuning Erodes Safety Alignment in Vision-Language Agentscited by 4
  7. 2026Geometry-aware similarity metrics for neural representations on Riemannian and statistical manifoldscited by 3
  8. 2023A Waddington landscape for prototype learning in generalized Hopfield networkscited by 2
  9. 2024Comparing the information content of probabilistic representation spacescited by 2
  10. 2026Distributional Spectral Diagnostics for Localizing Grokking Transitionscited by 2
  11. 2026Position: Weight Space Should Be a First-Class Generative AI Modalitycited by 2
  12. 2024Circuit design in biology and machine learning. I. Random networks and dimensional reductioncited by 1
8 more
  1. 2026Model Agreement via Anchoringcited by 1
  2. 2026Intrinsic Green's Learning: Supervised Learning on Manifolds via Inverse PDEcited by 1
  3. 2024Sequencing the Neurome: Towards Scalable Exact Parameter Reconstruction of Black-Box Neural Networkscited by 0
  4. 2024Contrastive Learning to Fine-Tune Feature Extraction Models for the Visual Cortexcited by 0
  5. 2025An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Modelscited by 0
  6. 2026Trajectory-Restricted Optimization Conditions and Geometry-Aware Linear Convergencecited by 0
  7. 2026Bayesian Fine-tuning in Projected Subspacescited by 0
  8. 2026Scalar Representations of Neural Network Training Dynamicscited by 0