Implicit Regularization of Discrete Gradient Dynamics in Linear Neural Networks
Not on the map. Why?
Abstract (arXiv)
When optimizing over-parameterized models, such as deep neural networks, a large set of parameters can achieve zero training error. In such cases, the choice of the optimization algorithm and its respective hyper-parameters introduces biases that will lead to convergence to specific minimizers of the objective. Consequently, this choice can be considered as an implicit regularization for the training of over-parametrized models. In this work, we push this idea further by studying the discrete gradient dynamics of the training of a two-layer linear network with the least-squares loss. Using a time rescaling, we show that, with a vanishing initialization and a small enough step size, this dynamics sequentially learns the solutions of a reduced-rank regression with a gradually increasing rank.
Key facts. Cited by 122 papers in the map (data through 7 Oct 2026). First posted on arXiv 30 Apr 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 22; 2023: 24; 2024: 20; 2025: 18; 2026: 21.
122 papers.
Sort:
By area:
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 150
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 83
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 82
- 2021The staircase property: How hierarchical structure can guide deep learningcited by 62
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 61
- 2023Saddle-to-Saddle Dynamics in Diagonal Linear Networkscited by 54
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
- 2023Transformers learn through gradual rank increasecited by 52
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 46
38 more
- 2021Implicit Regularization in Tensor Factorizationcited by 43
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 41
- 2023Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix Sensingcited by 40
- 2021On Margin Maximization in Linear and ReLU Networkscited by 34
- 2021Implicit Sparse Regularization: The Impact of Depth and Early Stoppingcited by 30
- 2023Don't blame Dataset Shift! Shortcut Learning due to Gradients and Cross Entropycited by 30
- 2023Early Neuron Alignment in Two-layer ReLU Networks with Small Initializationcited by 28
- 2023Dynamical versus Bayesian Phase Transitions in a Toy Model of Superpositioncited by 28
- 2022The dynamics of representation learning in shallow, non-linear autoencoderscited by 27
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networkscited by 27
- 2022Incremental Learning in Diagonal Linear Networkscited by 27
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptationcited by 26
- 2022Explicit Regularization in Overparametrized Models via Noise Injectioncited by 25
- 2021The loss landscape of deep linear neural networks: a second-order analysiscited by 24
- 2021More is Less: Inducing Sparsity via Overparameterizationcited by 22
- 2022Freeze then Train: Towards Provable Representation Learning under Spurious Correlations and Feature Noisecited by 22
- 2022Global Convergence of Sub-gradient Method for Robust Matrix Recovery: Small Initialization, Noisy Measurements, and Over-parameterizationcited by 21
- 2026There Will Be a Scientific Theory of Deep Learningcited by 21
- 2022Infinite-width limit of deep linear neural networkscited by 20
- 2023High-dimensional Asymptotics of Denoising Autoencoderscited by 19
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networkscited by 19
- 2023Understanding Unimodal Bias in Multimodal Deep Linear Networkscited by 18
- 2024Mixed Dynamics In Linear Networks: Unifying the Lazy and Active Regimescited by 18
- 2024Deep linear networks for regression are implicitly regularized towards flat minimacited by 17
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attentioncited by 16
- 2025How Muon's Spectral Design Benefits Generalization: A Study on Imbalanced Datacited by 16
- 2023The Implicit Bias of Minima Stability in Multivariate Shallow ReLU Networkscited by 15
- 2025Learning In-context n-grams with Transformers: Sub-n-grams Are Near-stationary Pointscited by 15
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Framecited by 14
- 2025Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architecturescited by 14
- 2021Analysis of feature learning in weight-tied autoencoders via the mean field lenscited by 13
- 2022Provable Acceleration of Heavy Ball beyond Quadratics for a Class of Polyak-Lojasiewicz Functions when the Non-Convexity is Averaged-Outcited by 11
- 2023The Interpolating Information Criterion for Overparameterized Modelscited by 11
- 2025Alternating Gradient Flows: A Theory of Feature Learning in Two-layer Neural Networkscited by 11
- 2021Principal Components Bias in Over-parameterized Linear Models, and its Manifestation in Deep Neural Networkscited by 10
- 2023Robust Implicit Regularization via Weight Normalizationcited by 10
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problemscited by 9
- 2024How Neural Networks Learn the Support is an Implicit Regularization Effect of SGDcited by 8
50 most cited of 122 All 122 (JSON)
- 2026Optimal Learning Rate Scaling Depends on Data in Deep Scalar Linear Networks8 Jul 2026
- 2026Muon learns balanced solutions in matrix factorization without slow saddle-to-saddle dynamics29 Jun 2026
- 2026Incremental Learning in Mirror Flows22 Jun 2026
- 2026Gradient Descent with Large Step Size Restores Symmetry in Deep Linear Networks with Multi-Pathway29 May 2026
- 2026The Stability of Singular Distribution: A Spectral Perspective on the Two-Phase Dynamics of Language Model Pre-training26 May 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026Early Data Exposure Improves Robustness to Subsequent Fine-Tuning12 May 2026
- 2026The critical slowing down in training diffusion models12 May 2026
- 2026There Will Be a Scientific Theory of Deep Learning23 Apr 2026
- 2026Revisiting Sharpness-Aware Minimization: A More Faithful and Effective Implementation9 Mar 2026
- 2026To Use or not to Use Muon: How Simplicity Bias in Optimizers Matters28 Feb 2026
- 2026Grokking Finite-Dimensional Algebra23 Feb 2026
38 more
- 2026Incremental Learning of Sparse Attention Patterns in Transformers22 Feb 2026
- 2026Muon with Spectral Guidance: Efficient Optimization for Scientific Machine Learning18 Feb 2026
- 2026Emergent Low-Rank Training Dynamics in MLPs with Smooth Activations5 Feb 2026
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Model4 Feb 2026
- 2026It's Not a Lottery, It's a Race: Understanding How Gradient Descent Adapts the Network's Capacity to the Task4 Feb 2026
- 2026Sequential Group Composition: A Window into the Mechanics of Deep Learning3 Feb 2026
- 2026On the Sample Efficiency of Inverse Dynamics Models for Semi-Supervised Imitation Learning2 Feb 2026
- 2026Why Loss Re-weighting Works If You Stop Early: Training Dynamics of Unconstrained Features17 Jan 2026
- 2026The Impact of Anisotropic Covariance Structure on the Training Dynamics and Generalization Error of Linear Networks11 Jan 2026
- 2025Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architectures23 Dec 2025
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamics3 Dec 2025
- 2025Tuning-Free Structured Sparse Recovery of Multiple Measurement Vectors using Implicit Regularization3 Dec 2025
- 2025How Muon's Spectral Design Benefits Generalization: A Study on Imbalanced Data27 Oct 2025
- 2025Neural Collapse under Gradient Flow on Shallow ReLU Networks for Orthogonally Separable Data24 Oct 2025
- 2025A Derandomization Framework for Structure Discovery: Applications in Neural Networks and Beyond22 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
- 2025Why is Your Language Model a Poor Implicit Reward Model?10 Jul 2025
- 2025A Log-Linear Analytics Approach to Cost Model Regularization for Inpatient Stays through Diagnostic Code Merging5 Jul 2025
- 2025Alternating Gradient Flows: A Theory of Feature Learning in Two-layer Neural Networks6 Jun 2025
- 2025Saddle-To-Saddle Dynamics in Deep ReLU Networks: Low-Rank Bias in the First Saddle Escape27 May 2025
- 2025A Local Polyak-Lojasiewicz and Descent Lemma of Gradient Descent For Overparametrized Linear Models16 May 2025
- 2025Geometry of Semantics in Next-Token Prediction: How Optimization Implicitly Organizes Linguistic Representations13 May 2025
- 2025Position: Solve Layerwise Linear Models First to Understand Neural Dynamical Phenomena (Neural Collapse, Emergence, Lazy/Rich Regime, and Grokking)28 Feb 2025
- 2025Closed-Form Training Dynamics Reveal Learned Features and Linear Structure in Word2Vec-like Models14 Feb 2025
- 2025Implicit Bias in Matrix Factorization and its Explicit Realization in a New Architecture27 Jan 2025
- 2025Gradient Descent Converges Linearly to Flatter Minima than Gradient Flow in Shallow Linear Networks15 Jan 2025
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Frame2 Nov 2024
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problems18 Oct 2024
- 2024Decoupling Dynamical Richness from Representation Learning: Towards Practical Measurement5 Oct 2024
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networks22 Sep 2024
- 2024Input Space Mode Connectivity in Deep Neural Networks9 Sep 2024
- 2024Lecture Notes on Linear Neural Networks: A Tale of Optimization and Generalization in Deep Learning25 Aug 2024
- 2024Towards Understanding Epoch-wise Double descent in Two-layer Linear Neural Networks13 Jul 2024
- 2024When Are Bias-Free ReLU Networks Effectively Linear Networks?18 Jun 2024
- 2024How Neural Networks Learn the Support is an Implicit Regularization Effect of SGD17 Jun 2024
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learning10 Jun 2024
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptation6 Jun 2024
50 newest of 122 All 122 (JSON)
- 2025Why is Your Language Model a Poor Implicit Reward Model?cited by 6
- 2025Why is Your Language Model a Poor Implicit Reward Model?10 Jul 2025
- 2026The critical slowing down in training diffusion modelscited by 1
- 2026The critical slowing down in training diffusion models12 May 2026
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 150
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 83
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 82
- 2021The staircase property: How hierarchical structure can guide deep learningcited by 62
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 61
- 2023Saddle-to-Saddle Dynamics in Diagonal Linear Networkscited by 54
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
- 2023Transformers learn through gradual rank increasecited by 52
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 46
12 most cited of 119 All 119 (JSON)
- 2026Optimal Learning Rate Scaling Depends on Data in Deep Scalar Linear Networks8 Jul 2026
- 2026Muon learns balanced solutions in matrix factorization without slow saddle-to-saddle dynamics29 Jun 2026
- 2026Incremental Learning in Mirror Flows22 Jun 2026
- 2026Gradient Descent with Large Step Size Restores Symmetry in Deep Linear Networks with Multi-Pathway29 May 2026
- 2026The Stability of Singular Distribution: A Spectral Perspective on the Two-Phase Dynamics of Language Model Pre-training26 May 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026Early Data Exposure Improves Robustness to Subsequent Fine-Tuning12 May 2026
- 2026There Will Be a Scientific Theory of Deep Learning23 Apr 2026
- 2026Revisiting Sharpness-Aware Minimization: A More Faithful and Effective Implementation9 Mar 2026
- 2026To Use or not to Use Muon: How Simplicity Bias in Optimizers Matters28 Feb 2026
- 2026Grokking Finite-Dimensional Algebra23 Feb 2026
- 2026Incremental Learning of Sparse Attention Patterns in Transformers22 Feb 2026
12 newest of 119 All 119 (JSON)