Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix Sensing
Not on the map. Why?
Abstract (arXiv)
It is believed that Gradient Descent (GD) induces an implicit bias towards good generalization in training machine learning models. This paper provides a fine-grained analysis of the dynamics of GD for the matrix sensing problem, whose goal is to recover a low-rank ground-truth matrix from near-isotropic linear measurements. It is shown that GD with small initialization behaves similarly to the greedy low-rank learning heuristics (Li et al., 2020) and follows an incremental learning procedure (Gissin et al., 2019): GD sequentially learns solutions with increasing ranks until it recovers the ground truth matrix. Compared to existing works which only analyze the first learning phase for rank-1 solutions, our result provides characterizations for the whole learning process. Moreover, besides the over-parameterized regime that many prior works focused on, our analysis of the incremental learning procedure also applies to the under-parameterized regime. Finally, we conduct numerical experiments to confirm our theoretical findings.
Key facts. Cited by 42 papers in the map (data through 14 Oct 2026). Builds on 41 of its 64 references that are in the map. First posted on arXiv 27 Jan 2023.
Builds on
41 of 64 references.
References · 41
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 552
- 2016Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Lojasiewicz Conditioncited by 539
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
- 2017Implicit Regularization in Matrix Factorizationcited by 322
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 306
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 266
- 2015Escaping From Saddle Points --- Online Stochastic Gradient for Tensor Decompositioncited by 243
- 2019SGD on Neural Networks Learns Functions of Increasing Complexitycited by 204
29 more
- 2020Directional convergence and alignment in deep learningcited by 178
- 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 150
- 2017No Spurious Local Minima in Nonconvex Low Rank Problems: A Unified Geometric Analysiscited by 129
- 2016Matrix Completion has No Spurious Local Minimumcited by 125
- 2019Implicit Regularization of Discrete Gradient Dynamics in Linear Neural Networkscited by 124
- 2019Implicit regularization for deep neural networks driven by an Ornstein-Uhlenbeck like processcited by 123
- 2021Label Noise SGD Provably Prefers Flat Global Minimizerscited by 121
- 2020Implicit Regularization in Deep Learning May Not Be Explainable by Normscited by 119
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
- 2015Low-rank Solutions of Linear Matrix Equations via Procrustes Flowcited by 97
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 83
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 78
- 2022Gradient flow dynamics of shallow ReLU networks for square loss and orthogonal inputscited by 76
- 2020Accelerating Ill-Conditioned Low-Rank Matrix Estimation via Scaled Gradient Descentcited by 71
- 2019Optimization for deep learning: theory and algorithmscited by 70
- 2019The Implicit Bias of Depth: How Incremental Learning Drives Generalizationcited by 64
- 2020The Surprising Simplicity of the Early-Time Learning Dynamics of Neural Networkscited by 58
- 2019Lexicographic and Depth-Sensitive Margins in Homogeneous and Non-Homogeneous Deep Modelscited by 53
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
- 2021Implicit Regularization in Tensor Factorizationcited by 44
- 2022Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descentcited by 39
- 2017Global Optimality in Low-rank Matrix Optimizationcited by 36
- 2022Algorithmic Regularization in Model-free Overparametrized Asymmetric Matrix Factorizationcited by 34
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networkscited by 27
- 2017The Global Optimization Geometry of Low-Rank Matrix Optimizationcited by 25
- 2013Gradient methods for convex minimization: better rates under weaker conditionscited by 23
- 2021Provable Generalization of SGD-trained Neural Networks of Any Width in the Presence of Adversarial Label Noisecited by 17
- 2020On implicit regularization: Morse functions and applications to matrix factorizationcited by 11
Built on by
Citing papers per year: 2023: 14; 2024: 13; 2025: 10; 2026: 5.
42 papers.
Sort:
By area:
- 2023Trained Transformers Learn Linear Models In-Contextcited by 337
- 2023Saddle-to-Saddle Dynamics in Diagonal Linear Networkscited by 57
- 2023Transformers learn through gradual rank increasecited by 52
- 2024LoRA Training in the NTK Regime has No Spurious Local Minimacited by 48
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representationcited by 20
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initializationcited by 18
- 2024Mixed Dynamics In Linear Networks: Unifying the Lazy and Active Regimescited by 18
- 2023Learning a Neuron by a Shallow ReLU Network: Dynamics and Implicit Bias for Correlated Inputscited by 16
- 2025Learning In-context n-grams with Transformers: Sub-n-grams Are Near-stationary Pointscited by 16
- 2024On the Crucial Role of Initialization for Matrix Factorizationcited by 15
- 2025Understanding the Learning Dynamics of LoRA: A Gradient Flow Perspective on Low-Rank Adaptation in Matrix Factorizationcited by 13
- 2024Directional Convergence Near Small Initializations and Saddles in Two-Homogeneous Neural Networkscited by 11
30 more
- 2025Towards Understanding Text Hallucination of Diffusion Models via Local Generation Biascited by 11
- 2024The Optimality of (Accelerated) SGD for High-Dimensional Quadratic Optimizationcited by 8
- 2024Early Directional Convergence in Deep Homogeneous Neural Networks for Small Initializationscited by 7
- 2025On Learning Verifiers and Implications to Chain-of-Thought Reasoningcited by 6
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descentcited by 5
- 2023A Quadratic Synchronization Rule for Distributed Deep Learningcited by 5
- 2024Task Diversity Shortens the ICL Plateaucited by 5
- 2024Swing-by Dynamics in Concept Learning and Compositional Generalizationcited by 5
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descentcited by 4
- 2023Efficient Compression of Overparameterized Deep Models through Low-Dimensional Learning Dynamicscited by 3
- 2024Connectivity Shapes Implicit Regularization in Matrix Factorization Models for Matrix Completioncited by 3
- 2024Non-convex matrix sensing: Breaking the quadratic rank barrier in the sample complexitycited by 3
- 2025One Rank at a Time: Cascading Error Dynamics in Sequential Learningcited by 3
- 2025On the Benefits of Weight Normalization for Overparameterized Matrix Sensingcited by 3
- 2023Gradient descent in matrix factorization: Understanding large initializationcited by 2
- 2023Implicit Regularization Makes Overparameterized Asymmetric Matrix Sensing Robust to Perturbationscited by 2
- 2024The Implicit Bias of Heterogeneity towards Invariance: A Study of Multi-Environment Matrix Sensingcited by 2
- 2025Neural Collapse under Gradient Flow on Shallow ReLU Networks for Orthogonally Separable Datacited by 2
- 2026Incremental Learning of Sparse Attention Patterns in Transformerscited by 2
- 2023Greedy Pruning with Group Lasso Provably Generalizes for Matrix Sensingcited by 1
- 2023Algorithmic Regularization in Tensor Optimization: Towards a Lifted Approach in Matrix Sensingcited by 1
- 2024Absence of spurious solutions far from ground truth: A low-rank analysis with high-order lossescited by 1
- 2024Lecture Notes on Linear Neural Networks: A Tale of Optimization and Generalization in Deep Learningcited by 1
- 2025Understanding Incremental Learning with Closed-form Solution to Gradient Flow on Overparamerterized Matrix Factorizationcited by 1
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensingcited by 0
- 2023Sparse Mean Estimation in Adversarial Settings via Incremental Learningcited by 0
- 2026Near-optimal and Efficient First-Order Algorithm for Multi-Task Learning with Shared Linear Representationcited by 0
- 2026Direction-Magnitude Decomposition for Low-Rank Matrix Optimization: Faster Convergence and Saddle-to-saddle Dynamicscited by 0
- 2026Ideal Paths for Approximating Logistic Gradient Descent Trajectories at Large Initializationcited by 0
- 2026Optimizer Geometry Sets the Pace: Spectral Learning Dynamics in Matrix Factorizationcited by 0
- 2026Optimizer Geometry Sets the Pace: Spectral Learning Dynamics in Matrix Factorization3 Oct 2026
- 2026Ideal Paths for Approximating Logistic Gradient Descent Trajectories at Large Initialization2 Oct 2026
- 2026Direction-Magnitude Decomposition for Low-Rank Matrix Optimization: Faster Convergence and Saddle-to-saddle Dynamics30 Jun 2026
- 2026Near-optimal and Efficient First-Order Algorithm for Multi-Task Learning with Shared Linear Representation1 May 2026
- 2026Incremental Learning of Sparse Attention Patterns in Transformers22 Feb 2026
- 2025Neural Collapse under Gradient Flow on Shallow ReLU Networks for Orthogonally Separable Data24 Oct 2025
- 2025On the Benefits of Weight Normalization for Overparameterized Matrix Sensing1 Oct 2025
- 2025Understanding Incremental Learning with Closed-form Solution to Gradient Flow on Overparamerterized Matrix Factorization28 Aug 2025
- 2025Learning In-context n-grams with Transformers: Sub-n-grams Are Near-stationary Points18 Aug 2025
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representation3 Jun 2025
- 2025On Learning Verifiers and Implications to Chain-of-Thought Reasoning28 May 2025
- 2025One Rank at a Time: Cascading Error Dynamics in Sequential Learning28 May 2025
30 more
- 2025Understanding the Learning Dynamics of LoRA: A Gradient Flow Perspective on Low-Rank Adaptation in Matrix Factorization10 Mar 2025
- 2025Towards Understanding Text Hallucination of Diffusion Models via Local Generation Bias5 Mar 2025
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descent1 Feb 2025
- 2024On the Crucial Role of Initialization for Matrix Factorization24 Oct 2024
- 2024Swing-by Dynamics in Concept Learning and Compositional Generalization10 Oct 2024
- 2024Task Diversity Shortens the ICL Plateau7 Oct 2024
- 2024The Optimality of (Accelerated) SGD for High-Dimensional Quadratic Optimization15 Sep 2024
- 2024Lecture Notes on Linear Neural Networks: A Tale of Optimization and Generalization in Deep Learning25 Aug 2024
- 2024Non-convex matrix sensing: Breaking the quadratic rank barrier in the sample complexity20 Aug 2024
- 2024Mixed Dynamics In Linear Networks: Unifying the Lazy and Active Regimes27 May 2024
- 2024Connectivity Shapes Implicit Regularization in Matrix Factorization Models for Matrix Completion22 May 2024
- 2024Early Directional Convergence in Deep Homogeneous Neural Networks for Small Initializations12 Mar 2024
- 2024Absence of spurious solutions far from ground truth: A low-rank analysis with high-order losses10 Mar 2024
- 2024The Implicit Bias of Heterogeneity towards Invariance: A Study of Multi-Environment Matrix Sensing3 Mar 2024
- 2024LoRA Training in the NTK Regime has No Spurious Local Minima19 Feb 2024
- 2024Directional Convergence Near Small Initializations and Saddles in Two-Homogeneous Neural Networks14 Feb 2024
- 2023Efficient Compression of Overparameterized Deep Models through Low-Dimensional Learning Dynamics8 Nov 2023
- 2023Algorithmic Regularization in Tensor Optimization: Towards a Lifted Approach in Matrix Sensing24 Oct 2023
- 2023A Quadratic Synchronization Rule for Distributed Deep Learning22 Oct 2023
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initialization3 Oct 2023
- 2023Implicit Regularization Makes Overparameterized Asymmetric Matrix Sensing Robust to Perturbations4 Sep 2023
- 2023Trained Transformers Learn Linear Models In-Context16 Jun 2023
- 2023Transformers learn through gradual rank increase12 Jun 2023
- 2023Learning a Neuron by a Shallow ReLU Network: Dynamics and Implicit Bias for Correlated Inputs10 Jun 2023
- 2023Gradient descent in matrix factorization: Understanding large initialization30 May 2023
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descent26 May 2023
- 2023Sparse Mean Estimation in Adversarial Settings via Incremental Learning24 May 2023
- 2023Saddle-to-Saddle Dynamics in Diagonal Linear Networks2 Apr 2023
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensing24 Mar 2023
- 2023Greedy Pruning with Group Lasso Provably Generalizes for Matrix Sensing20 Mar 2023
- 2023Trained Transformers Learn Linear Models In-Contextcited by 337
- 2023Saddle-to-Saddle Dynamics in Diagonal Linear Networkscited by 57
- 2023Transformers learn through gradual rank increasecited by 52
- 2024LoRA Training in the NTK Regime has No Spurious Local Minimacited by 48
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representationcited by 20
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initializationcited by 18
- 2024Mixed Dynamics In Linear Networks: Unifying the Lazy and Active Regimescited by 18
- 2023Learning a Neuron by a Shallow ReLU Network: Dynamics and Implicit Bias for Correlated Inputscited by 16
- 2025Learning In-context n-grams with Transformers: Sub-n-grams Are Near-stationary Pointscited by 16
- 2024On the Crucial Role of Initialization for Matrix Factorizationcited by 15
- 2025Understanding the Learning Dynamics of LoRA: A Gradient Flow Perspective on Low-Rank Adaptation in Matrix Factorizationcited by 13
- 2024Directional Convergence Near Small Initializations and Saddles in Two-Homogeneous Neural Networkscited by 11
12 most cited of 39 All 39 (JSON)
- 2026Optimizer Geometry Sets the Pace: Spectral Learning Dynamics in Matrix Factorization3 Oct 2026
- 2026Ideal Paths for Approximating Logistic Gradient Descent Trajectories at Large Initialization2 Oct 2026
- 2026Direction-Magnitude Decomposition for Low-Rank Matrix Optimization: Faster Convergence and Saddle-to-saddle Dynamics30 Jun 2026
- 2026Near-optimal and Efficient First-Order Algorithm for Multi-Task Learning with Shared Linear Representation1 May 2026
- 2026Incremental Learning of Sparse Attention Patterns in Transformers22 Feb 2026
- 2025Neural Collapse under Gradient Flow on Shallow ReLU Networks for Orthogonally Separable Data24 Oct 2025
- 2025On the Benefits of Weight Normalization for Overparameterized Matrix Sensing1 Oct 2025
- 2025Understanding Incremental Learning with Closed-form Solution to Gradient Flow on Overparamerterized Matrix Factorization28 Aug 2025
- 2025Learning In-context n-grams with Transformers: Sub-n-grams Are Near-stationary Points18 Aug 2025
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representation3 Jun 2025
- 2025One Rank at a Time: Cascading Error Dynamics in Sequential Learning28 May 2025
- 2025Understanding the Learning Dynamics of LoRA: A Gradient Flow Perspective on Low-Rank Adaptation in Matrix Factorization10 Mar 2025
12 newest of 39 All 39 (JSON)