The Training Process of Many Deep Networks Explores the Same Low-Dimensional Manifold
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
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2016Layer Normalizationcited by 5,579
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2016Wide Residual Networkscited by 3,057
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
27 more
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2014Striving for Simplicity: The All Convolutional Netcited by 983
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2019Benign Overfitting in Linear Regressioncited by 544
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
- 2017Opening the Black Box of Deep Neural Networks via Informationcited by 494
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
- 2019Making Convolutional Networks Shift-Invariant Againcited by 390
- 2019A Baseline for Few-Shot Image Classificationcited by 298
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 267
- 2021Deep learning: a statistical viewpointcited by 239
- 2017Emergence of Invariance and Disentanglement in Deep Representationscited by 213
- 2022Patches Are All You Need?cited by 201
- 2018Just Interpolate: Kernel "Ridgeless" Regression Can Generalizecited by 190
- 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 182
- 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 130
- 2017Stochastic gradient descent performs variational inference, converges to limit cycles for deep networkscited by 108
- 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 67
- 2019Let's Agree to Agree: Neural Networks Share Classification Order on Real Datasetscited by 55
- 2019A Purely Algebraic Justification of the Kabsch-Umeyama Algorithmcited by 24
- 2021Does the Data Induce Capacity Control in Deep Learning?cited by 13
- 2021On the interplay between data structure and loss function in classification problemscited by 12
- 2018PCA of high dimensional random walks with comparison to neural network trainingcited by 11
- 2014Weighted principal component analysis: a weighted covariance eigendecomposition approachcited by 4
Built on by
Citing papers per year: 2023: 2; 2024: 6; 2025: 3; 2026: 9.
20 papers.
Sort:
By area:
- 2025The Universal Weight Subspace Hypothesiscited by 22
- 2025Dynamics of Transient Structure in In-Context Linear Regression Transformerscited by 12
- 2024Density estimation with LLMs: a geometric investigation of in-context learning trajectoriescited by 6
- 2023Learning Capacity: A Measure of the Effective Dimensionality of a Modelcited by 4
- 2024Many Perception Tasks are Highly Redundant Functions of their Input Datacited by 4
- 2026Narrow Fine-Tuning Erodes Safety Alignment in Vision-Language Agentscited by 4
- 2026Geometry-aware similarity metrics for neural representations on Riemannian and statistical manifoldscited by 3
- 2023A Waddington landscape for prototype learning in generalized Hopfield networkscited by 2
- 2024Comparing the information content of probabilistic representation spacescited by 2
- 2026Distributional Spectral Diagnostics for Localizing Grokking Transitionscited by 2
- 2026Position: Weight Space Should Be a First-Class Generative AI Modalitycited by 2
- 2024Circuit design in biology and machine learning. I. Random networks and dimensional reductioncited by 1
8 more
- 2026Model Agreement via Anchoringcited by 1
- 2026Intrinsic Green's Learning: Supervised Learning on Manifolds via Inverse PDEcited by 1
- 2024Sequencing the Neurome: Towards Scalable Exact Parameter Reconstruction of Black-Box Neural Networkscited by 0
- 2024Contrastive Learning to Fine-Tune Feature Extraction Models for the Visual Cortexcited by 0
- 2025An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Modelscited by 0
- 2026Trajectory-Restricted Optimization Conditions and Geometry-Aware Linear Convergencecited by 0
- 2026Bayesian Fine-tuning in Projected Subspacescited by 0
- 2026Scalar Representations of Neural Network Training Dynamicscited by 0
- 2026Intrinsic Green's Learning: Supervised Learning on Manifolds via Inverse PDE8 Jul 2026
- 2026Scalar Representations of Neural Network Training Dynamics29 Jun 2026
- 2026Position: Weight Space Should Be a First-Class Generative AI Modality18 May 2026
- 2026Bayesian Fine-tuning in Projected Subspaces8 May 2026
- 2026Distributional Spectral Diagnostics for Localizing Grokking Transitions7 May 2026
- 2026Trajectory-Restricted Optimization Conditions and Geometry-Aware Linear Convergence18 Apr 2026
- 2026Geometry-aware similarity metrics for neural representations on Riemannian and statistical manifolds30 Mar 2026
- 2026Model Agreement via Anchoring26 Feb 2026
- 2026Narrow Fine-Tuning Erodes Safety Alignment in Vision-Language Agents18 Feb 2026
- 2025The Universal Weight Subspace Hypothesis4 Dec 2025
- 2025An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Models13 May 2025
- 2025Dynamics of Transient Structure in In-Context Linear Regression Transformers29 Jan 2025
8 more
- 2024Contrastive Learning to Fine-Tune Feature Extraction Models for the Visual Cortex8 Oct 2024
- 2024Density estimation with LLMs: a geometric investigation of in-context learning trajectories7 Oct 2024
- 2024Sequencing the Neurome: Towards Scalable Exact Parameter Reconstruction of Black-Box Neural Networks27 Sep 2024
- 2024Circuit design in biology and machine learning. I. Random networks and dimensional reduction18 Aug 2024
- 2024Many Perception Tasks are Highly Redundant Functions of their Input Data18 Jul 2024
- 2024Comparing the information content of probabilistic representation spaces31 May 2024
- 2023A Waddington landscape for prototype learning in generalized Hopfield networks4 Dec 2023
- 2023Learning Capacity: A Measure of the Effective Dimensionality of a Model27 May 2023
- 2025The Universal Weight Subspace Hypothesiscited by 22
- 2025Dynamics of Transient Structure in In-Context Linear Regression Transformerscited by 12
- 2024Density estimation with LLMs: a geometric investigation of in-context learning trajectoriescited by 6
- 2023Learning Capacity: A Measure of the Effective Dimensionality of a Modelcited by 4
- 2024Many Perception Tasks are Highly Redundant Functions of their Input Datacited by 4
- 2026Geometry-aware similarity metrics for neural representations on Riemannian and statistical manifoldscited by 3
- 2023A Waddington landscape for prototype learning in generalized Hopfield networkscited by 2
- 2024Comparing the information content of probabilistic representation spacescited by 2
- 2026Distributional Spectral Diagnostics for Localizing Grokking Transitionscited by 2
- 2026Position: Weight Space Should Be a First-Class Generative AI Modalitycited by 2
- 2024Circuit design in biology and machine learning. I. Random networks and dimensional reductioncited by 1
- 2026Model Agreement via Anchoringcited by 1
12 most cited of 19 All 19 (JSON)
- 2026Intrinsic Green's Learning: Supervised Learning on Manifolds via Inverse PDE8 Jul 2026
- 2026Scalar Representations of Neural Network Training Dynamics29 Jun 2026
- 2026Position: Weight Space Should Be a First-Class Generative AI Modality18 May 2026
- 2026Bayesian Fine-tuning in Projected Subspaces8 May 2026
- 2026Distributional Spectral Diagnostics for Localizing Grokking Transitions7 May 2026
- 2026Trajectory-Restricted Optimization Conditions and Geometry-Aware Linear Convergence18 Apr 2026
- 2026Geometry-aware similarity metrics for neural representations on Riemannian and statistical manifolds30 Mar 2026
- 2026Model Agreement via Anchoring26 Feb 2026
- 2025The Universal Weight Subspace Hypothesis4 Dec 2025
- 2025An Analytical Characterization of Sloppiness in Neural Networks: Insights from Linear Models13 May 2025
- 2025Dynamics of Transient Structure in In-Context Linear Regression Transformers29 Jan 2025
- 2024Contrastive Learning to Fine-Tune Feature Extraction Models for the Visual Cortex8 Oct 2024
12 newest of 19 All 19 (JSON)