Learning deep linear neural networks: Riemannian gradient flows and convergence to global minimizers
Not on the map. Why?
Abstract (arXiv)
We study the convergence of gradient flows related to learning deep linear neural networks (where the activation function is the identity map) from data. In this case, the composition of the network layers amounts to simply multiplying the weight matrices of all layers together, resulting in an overparameterized problem. The gradient flow with respect to these factors can be re-interpreted as a Riemannian gradient flow on the manifold of rank-r matrices endowed with a suitable Riemannian metric. We show that the flow always converges to a critical point of the underlying functional. Moreover, we establish that, for almost all initializations, the flow converges to a global minimum on the manifold of rank k matrices for some k≤ r.
Key facts. Cited by 61 papers in the map (data through 14 Oct 2026). First posted on arXiv 12 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 9; 2022: 8; 2023: 12; 2024: 10; 2025: 15; 2026: 7.
61 papers.
Sort:
By area:
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 78
- 2023Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flowscited by 45
- 2025Symmetry in Neural Network Parameter Spacescited by 36
- 2021Geometry of Linear Convolutional Networkscited by 27
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networkscited by 27
- 2021The loss landscape of deep linear neural networks: a second-order analysiscited by 24
- 2021Convergence of gradient descent for learning linear neural networkscited by 22
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensionscited by 21
- 2022Infinite-width limit of deep linear neural networkscited by 20
- 2023Function Space and Critical Points of Linear Convolutional Networkscited by 19
- 2023Optimization Dynamics of Equivariant and Augmented Neural Networkscited by 18
- 2023Neural (Tangent Kernel) Collapsecited by 18
38 more
- 2024Plastic Learning with Deep Fourier Featurescited by 14
- 2023Robust low-rank training via approximate orthonormal constraintscited by 13
- 2023Robust Implicit Regularization via Weight Normalizationcited by 12
- 2023Geometry-aware training of factorized layers in tensor Tucker formatcited by 12
- 2021Convergence analysis for gradient flows in the training of artificial neural networks with ReLU activationcited by 11
- 2024Provable Emergence of Deep Neural Collapse and Low-Rank Bias in L2-Regularized Nonlinear Networkscited by 11
- 2024Keep the Momentum: Conservation Laws beyond Euclidean Gradient Flowscited by 11
- 2021Principal Components Bias in Over-parameterized Linear Models, and its Manifestation in Deep Neural Networkscited by 10
- 2021Stable Recovery of Entangled Weights: Towards Robust Identification of Deep Neural Networks from Minimal Samplescited by 8
- 2022Deep Linear Networks for Matrix Completion -- An Infinite Depth Limitcited by 8
- 2024Understanding the training of infinitely deep and wide ResNets with Conditional Optimal Transportcited by 8
- 2025An entropy formula for the Deep Linear Networkcited by 8
- 2023Critical Points and Convergence Analysis of Generative Deep Linear Networks Trained with Bures-Wasserstein Losscited by 7
- 2024Federated Dynamical Low-Rank Training with Global Loss Convergence Guaranteescited by 7
- 2025Transformative or Conservative? Conservation laws for ResNets and Transformerscited by 7
- 2022Side Effects of Learning from Low-dimensional Data Embedded in a Euclidean Spacecited by 6
- 2021How and When Random Feedback Works: A Case Study of Low-Rank Matrix Factorizationcited by 5
- 2022Gradient descent provably escapes saddle points in the training of shallow ReLU networkscited by 5
- 2022Finite Sample Identification of Wide Shallow Neural Networks with Biasescited by 5
- 2023Neural Hilbert Ladders: Multi-Layer Neural Networks in Function Spacecited by 5
- 2021First-Order Methods for Convex Optimizationcited by 4
- 2023Combining Explicit and Implicit Regularization for Efficient Learning in Deep Networkscited by 4
- 2023Implicit regularization in AI meets generalized hardness of approximation in optimization -- Sharp results for diagonal linear networkscited by 4
- 2024Convergence of continuous-time stochastic gradient descent with applications to deep neural networkscited by 4
- 2025Intrinsic training dynamics of deep neural networkscited by 4
- 2025Regularization Implies balancedness in the deep linear networkcited by 4
- 2024On the Convergence of Gradient Descent for Large Learning Ratescited by 3
- 2025Understanding Learning Invariance in Deep Linear Networkscited by 3
- 2024On the Role of Initialization on the Implicit Bias in Deep Linear Networkscited by 2
- 2025The Riemannian Geometry Associated to Gradient Flows of Linear Convolutional Networkscited by 2
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamicscited by 2
- 2022A Dynamics Theory of Implicit Regularization in Deep Low-Rank Matrix Factorizationcited by 1
- 2024Lecture Notes on Linear Neural Networks: A Tale of Optimization and Generalization in Deep Learningcited by 1
- 2025Error Bound Analysis for the Regularized Loss of Deep Linear Neural Networkscited by 1
- 2025The Effect of Depth on the Expressivity of Deep Linear State-Space Modelscited by 1
- 2025Wormhole Dynamics in Deep Neural Networkscited by 1
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Modelcited by 1
- 2026Conservation Laws for Modern Neural Architecturescited by 1
50 most cited of 61 All 61 (JSON)
- 2026Benign Loss Landscapes Can Coexist with Worst-Case Hardness11 Sep 2026
- 2026Cascading Through the Hierarchy: Regularizer-Induced Feature Detection as Phase Transitions in Deep Linear Neural Networks6 Aug 2026
- 2026Conservation Laws for Modern Neural Architectures16 Jun 2026
- 2026On the Entropy Formula for Real, Complex, and Quaternionic Deep Linear Networks15 Jun 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Model4 Feb 2026
- 2026Convergence of gradient flow for learning convolutional neural networks13 Jan 2026
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamics3 Dec 2025
- 2025On the Convergence of Overparameterized Problems: Inherent Properties of the Compositional Structure of Neural Networks12 Nov 2025
- 2025Regularization Implies balancedness in the deep linear network3 Nov 2025
- 2025Gradient Flow Convergence Guarantee for General Neural Network Architectures28 Sep 2025
- 2025Geodesics in the Deep Linear Network18 Sep 2025
38 more
- 2025An entropy formula for the Deep Linear Network11 Sep 2025
- 2025Wormhole Dynamics in Deep Neural Networks20 Aug 2025
- 2025Intrinsic training dynamics of deep neural networks10 Aug 2025
- 2025The Riemannian Geometry Associated to Gradient Flows of Linear Convolutional Networks8 Jul 2025
- 2025The Effect of Depth on the Expressivity of Deep Linear State-Space Models24 Jun 2025
- 2025Understanding Learning Invariance in Deep Linear Networks16 Jun 2025
- 2025Symmetry in Neural Network Parameter Spaces16 Jun 2025
- 2025Transformative or Conservative? Conservation laws for ResNets and Transformers6 Jun 2025
- 2025Digital implementations of deep feature extractors are intrinsically informative20 Feb 2025
- 2025Error Bound Analysis for the Regularized Loss of Deep Linear Neural Networks16 Feb 2025
- 2024A new Input Convex Neural Network with application to options pricing19 Nov 2024
- 2024Plastic Learning with Deep Fourier Features27 Oct 2024
- 2024Convergence of continuous-time stochastic gradient descent with applications to deep neural networks11 Sep 2024
- 2024Lecture Notes on Linear Neural Networks: A Tale of Optimization and Generalization in Deep Learning25 Aug 2024
- 2024Federated Dynamical Low-Rank Training with Global Loss Convergence Guarantees25 Jun 2024
- 2024Keep the Momentum: Conservation Laws beyond Euclidean Gradient Flows21 May 2024
- 2024Understanding the training of infinitely deep and wide ResNets with Conditional Optimal Transport19 Mar 2024
- 2024On the Convergence of Gradient Descent for Large Learning Rates20 Feb 2024
- 2024Provable Emergence of Deep Neural Collapse and Low-Rank Bias in L2-Regularized Nonlinear Networks6 Feb 2024
- 2024On the Role of Initialization on the Implicit Bias in Deep Linear Networks4 Feb 2024
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensions7 Nov 2023
- 2023Implicit regularization in AI meets generalized hardness of approximation in optimization -- Sharp results for diagonal linear networks13 Jul 2023
- 2023Neural Hilbert Ladders: Multi-Layer Neural Networks in Function Space3 Jul 2023
- 2023Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flows30 Jun 2023
- 2023Robust low-rank training via approximate orthonormal constraints2 Jun 2023
- 2023Combining Explicit and Implicit Regularization for Efficient Learning in Deep Networks1 Jun 2023
- 2023Geometry-aware training of factorized layers in tensor Tucker format30 May 2023
- 2023Neural (Tangent Kernel) Collapse25 May 2023
- 2023Robust Implicit Regularization via Weight Normalization9 May 2023
- 2023Function Space and Critical Points of Linear Convolutional Networks12 Apr 2023
- 2023Optimization Dynamics of Equivariant and Augmented Neural Networks23 Mar 2023
- 2023Critical Points and Convergence Analysis of Generative Deep Linear Networks Trained with Bures-Wasserstein Loss6 Mar 2023
- 2022A Dynamics Theory of Implicit Regularization in Deep Low-Rank Matrix Factorization29 Dec 2022
- 2022Infinite-width limit of deep linear neural networks29 Nov 2022
- 2022Finite Sample Identification of Wide Shallow Neural Networks with Biases8 Nov 2022
- 2022Deep Linear Networks for Matrix Completion -- An Infinite Depth Limit22 Oct 2022
- 2022Gradient descent provably escapes saddle points in the training of shallow ReLU networks3 Aug 2022
- 2022Side Effects of Learning from Low-dimensional Data Embedded in a Euclidean Space1 Mar 2022
50 newest of 61 All 61 (JSON)
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 78
- 2023Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flowscited by 45
- 2025Symmetry in Neural Network Parameter Spacescited by 36
- 2021Geometry of Linear Convolutional Networkscited by 27
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networkscited by 27
- 2021The loss landscape of deep linear neural networks: a second-order analysiscited by 24
- 2021Convergence of gradient descent for learning linear neural networkscited by 22
- 2022Infinite-width limit of deep linear neural networkscited by 20
- 2023Function Space and Critical Points of Linear Convolutional Networkscited by 19
- 2023Optimization Dynamics of Equivariant and Augmented Neural Networkscited by 18
- 2023Neural (Tangent Kernel) Collapsecited by 18
- 2024Plastic Learning with Deep Fourier Featurescited by 14
12 most cited of 60 All 60 (JSON)
- 2026Benign Loss Landscapes Can Coexist with Worst-Case Hardness11 Sep 2026
- 2026Cascading Through the Hierarchy: Regularizer-Induced Feature Detection as Phase Transitions in Deep Linear Neural Networks6 Aug 2026
- 2026Conservation Laws for Modern Neural Architectures16 Jun 2026
- 2026On the Entropy Formula for Real, Complex, and Quaternionic Deep Linear Networks15 Jun 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Model4 Feb 2026
- 2026Convergence of gradient flow for learning convolutional neural networks13 Jan 2026
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamics3 Dec 2025
- 2025On the Convergence of Overparameterized Problems: Inherent Properties of the Compositional Structure of Neural Networks12 Nov 2025
- 2025Regularization Implies balancedness in the deep linear network3 Nov 2025
- 2025Gradient Flow Convergence Guarantee for General Neural Network Architectures28 Sep 2025
- 2025Geodesics in the Deep Linear Network18 Sep 2025
12 newest of 60 All 60 (JSON)