Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descent
Not on the map. Why?
Abstract (arXiv)
As part of the effort to understand implicit bias of gradient descent in overparametrized models, several results have shown how the training trajectory on the overparametrized model can be understood as mirror descent on a different objective. The main result here is a characterization of this phenomenon under a notion termed commuting parametrization, which encompasses all the previous results in this setting. It is shown that gradient flow with any commuting parametrization is equivalent to continuous mirror descent with a related Legendre function. Conversely, continuous mirror descent with any Legendre function can be viewed as gradient flow with a related commuting parametrization. The latter result relies upon Nash's embedding theorem.
Key facts. Cited by 38 papers in the map (data through 7 Oct 2026). Builds on 43 of its 92 references that are in the map. First posted on arXiv 8 Jul 2022.
Builds on
43 of 92 references.
References · 43
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2014Convex Optimization: Algorithms and Complexitycited by 569
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2018Learning and Generalization in Overparameterized Neural Networks, Going Beyond Two Layerscited by 340
- 2017Implicit Regularization in Matrix Factorizationcited by 318
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 308
31 more
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 302
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 267
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 259
- 2018Implicit Bias of Gradient Descent on Linear Convolutional Networkscited by 251
- 2020Directional convergence and alignment in deep learningcited by 172
- 2019Towards Explaining the Regularization Effect of Initial Large Learning Rate in Training Neural Networkscited by 162
- 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 148
- 2019Scaling Limits of Wide Neural Networks with Weight Sharing: Gaussian Process Behavior, Gradient Independence, and Neural Tangent Kernel Derivationcited by 146
- 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 118
- 2018Convergence of Gradient Descent on Separable Datacited by 108
- 2021Implicit Bias of SGD for Diagonal Linear Networks: a Provable Benefit of Stochasticitycited by 107
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 106
- 2020Shape Matters: Understanding the Implicit Bias of the Noise Covariancecited by 88
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 82
- 2018Risk and parameter convergence of logistic regressioncited by 81
- 2019Implicit Regularization for Optimal Sparse Recoverycited by 78
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 77
- 2020A Unifying View on Implicit Bias in Training Linear Neural Networkscited by 74
- 2020Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracycited by 70
- 2021On the Implicit Bias of Initialization Shape: Beyond Infinitesimal Mirror Descentcited by 69
- 2020The Implicit Bias for Adaptive Optimization Algorithms on Homogeneous Neural Networkscited by 43
- 2021Fast Margin Maximization via Dual Accelerationcited by 33
- 2021The Benefits of Implicit Regularization from SGD in Least Squares Problemscited by 32
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networkscited by 27
- 2018Hessian Riemannian gradient flows in convex programmingcited by 25
- 2021Understanding Deflation Process in Over-parametrized Tensor Decompositioncited by 20
- 2020Mirrorless Mirror Descent: A Natural Derivation of Mirror Descentcited by 18
- 2019The Implicit Bias of AdaGrad on Separable Datacited by 15
- 2022Non-convex online learning via algorithmic equivalencecited by 10
Built on by
Citing papers per year: 2022: 1; 2023: 8; 2024: 10; 2025: 7; 2026: 12.
38 papers.
Sort:
By area:
- 2023Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flowscited by 44
- 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
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attentioncited by 16
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomycited by 16
- 2024Transformers Learn Low Sensitivity Functions: Investigations and Implicationscited by 13
- 2024Mask in the Mirror: Implicit Sparsificationcited by 12
- 2023mSAM: Micro-Batch-Averaged Sharpness-Aware Minimizationcited by 10
- 2025Mirror, Mirror of the Flow: How Does Regularization Shape Implicit Bias?cited by 10
- 2024Leveraging Continuous Time to Understand Momentum When Training Diagonal Linear Networkscited by 9
- 2022Deep Linear Networks can Benignly Overfit when Shallow Ones Docited by 8
- 2023Implicit Regularization for Group Sparsitycited by 8
26 more
- 2024Implicit Bias of Mirror Flow on Separable Datacited by 7
- 2025Sign-In to the Lottery: Reparameterizing Sparse Training From Scratchcited by 6
- 2025On Learning Verifiers and Implications to Chain-of-Thought Reasoningcited by 6
- 2023Implicit Regularization Leads to Benign Overfitting for Sparse Linear Regressioncited by 4
- 2023Combining Explicit and Implicit Regularization for Efficient Learning in Deep Networkscited by 4
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescalingcited by 4
- 2025Neural Graduated Assignment for Maximum Common Edge Subgraphscited by 3
- 2025On exploration of an interior mirror descent flow for stochastic nonconvex constrained problemcited by 3
- 2025Intrinsic training dynamics of deep neural networkscited by 3
- 2025Diagonal Linear Networks and the Lasso Regularization Pathcited by 3
- 2026Understanding and inverse design of implicit bias in stochastic learning: a geometric perspectivecited by 3
- 2024Improving Implicit Regularization of SGD with Preconditioning for Least Square Problemscited by 2
- 2024A Mirror Descent Perspective of Smoothed Sign Descentcited by 2
- 2024Optimization Insights into Deep Diagonal Linear Networkscited by 2
- 2026Never Saddle for Reparameterized Steepest Descent as Mirror Flowcited by 2
- 2026Policy Regret for Embedding Model Routing: Contextual Bandits with Low-Rank Expertscited by 1
- 2026Incremental Learning in Mirror Flowscited by 1
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensingcited by 0
- 2026Gradient Flow Polarizes Softmax Outputs towards Low-Entropy Solutionscited by 0
- 2026Mirror Descent on Riemannian Manifoldscited by 0
- 2026Implicit Bias of Mirror Flow in Homogeneous Neural Networks: Sparse and Dense Feature Learningcited by 0
- 2026HORST: Composing Optimizer Geometries for Sparse Transformer Trainingcited by 0
- 2026Theoretical Analysis of Sparse Optimization with Reparameterization, Weight Decay, and Adaptive Learning Ratecited by 0
- 2026Online Learning on Hidden-Convex Losses via Algorithmic Equivalence: Optimal Regret, Geometric Barrier, and Bandit Feedbackcited by 0
- 2026Gradient Flow Dynamics and Implicit Bias of Diagonal Linear Networks under Infinitesimal Initializationcited by 0
- 2026Establishing Boundary KKT Convergence of Mirror Descent through Reparameterizationcited by 0
- 2026Establishing Boundary KKT Convergence of Mirror Descent through Reparameterization7 Aug 2026
- 2026Gradient Flow Dynamics and Implicit Bias of Diagonal Linear Networks under Infinitesimal Initialization14 Jul 2026
- 2026Incremental Learning in Mirror Flows22 Jun 2026
- 2026Policy Regret for Embedding Model Routing: Contextual Bandits with Low-Rank Experts12 Jun 2026
- 2026Online Learning on Hidden-Convex Losses via Algorithmic Equivalence: Optimal Regret, Geometric Barrier, and Bandit Feedback25 May 2026
- 2026Theoretical Analysis of Sparse Optimization with Reparameterization, Weight Decay, and Adaptive Learning Rate24 May 2026
- 2026HORST: Composing Optimizer Geometries for Sparse Transformer Training20 May 2026
- 2026Implicit Bias of Mirror Flow in Homogeneous Neural Networks: Sparse and Dense Feature Learning19 May 2026
- 2026Mirror Descent on Riemannian Manifolds18 Mar 2026
- 2026Gradient Flow Polarizes Softmax Outputs towards Low-Entropy Solutions6 Mar 2026
- 2026Never Saddle for Reparameterized Steepest Descent as Mirror Flow2 Mar 2026
- 2026Understanding and inverse design of implicit bias in stochastic learning: a geometric perspective10 Jan 2026
26 more
- 2025Diagonal Linear Networks and the Lasso Regularization Path23 Sep 2025
- 2025Intrinsic training dynamics of deep neural networks10 Aug 2025
- 2025On exploration of an interior mirror descent flow for stochastic nonconvex constrained problem21 Jul 2025
- 2025On Learning Verifiers and Implications to Chain-of-Thought Reasoning28 May 2025
- 2025Neural Graduated Assignment for Maximum Common Edge Subgraphs18 May 2025
- 2025Mirror, Mirror of the Flow: How Does Regularization Shape Implicit Bias?17 Apr 2025
- 2025Sign-In to the Lottery: Reparameterizing Sparse Training From Scratch17 Apr 2025
- 2024Optimization Insights into Deep Diagonal Linear Networks21 Dec 2024
- 2024A Mirror Descent Perspective of Smoothed Sign Descent18 Oct 2024
- 2024Mask in the Mirror: Implicit Sparsification19 Aug 2024
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomy23 Jun 2024
- 2024Implicit Bias of Mirror Flow on Separable Data18 Jun 2024
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learning10 Jun 2024
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attention13 Mar 2024
- 2024Improving Implicit Regularization of SGD with Preconditioning for Least Square Problems13 Mar 2024
- 2024Transformers Learn Low Sensitivity Functions: Investigations and Implications11 Mar 2024
- 2024Leveraging Continuous Time to Understand Momentum When Training Diagonal Linear Networks8 Mar 2024
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescaling24 Nov 2023
- 2023Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flows30 Jun 2023
- 2023Combining Explicit and Implicit Regularization for Efficient Learning in Deep Networks1 Jun 2023
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensing24 Mar 2023
- 2023mSAM: Micro-Batch-Averaged Sharpness-Aware Minimization19 Feb 2023
- 2023Implicit Regularization Leads to Benign Overfitting for Sparse Linear Regression1 Feb 2023
- 2023Implicit Regularization for Group Sparsity29 Jan 2023
- 2023Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix Sensing27 Jan 2023
- 2022Deep Linear Networks can Benignly Overfit when Shallow Ones Do19 Sep 2022
- 2023Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flowscited by 44
- 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
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attentioncited by 16
- 2024Transformers Learn Low Sensitivity Functions: Investigations and Implicationscited by 13
- 2024Mask in the Mirror: Implicit Sparsificationcited by 12
- 2023mSAM: Micro-Batch-Averaged Sharpness-Aware Minimizationcited by 10
- 2025Mirror, Mirror of the Flow: How Does Regularization Shape Implicit Bias?cited by 10
- 2024Leveraging Continuous Time to Understand Momentum When Training Diagonal Linear Networkscited by 9
- 2022Deep Linear Networks can Benignly Overfit when Shallow Ones Docited by 8
- 2023Implicit Regularization for Group Sparsitycited by 8
- 2024Implicit Bias of Mirror Flow on Separable Datacited by 7
12 most cited of 36 All 36 (JSON)
- 2026Establishing Boundary KKT Convergence of Mirror Descent through Reparameterization7 Aug 2026
- 2026Gradient Flow Dynamics and Implicit Bias of Diagonal Linear Networks under Infinitesimal Initialization14 Jul 2026
- 2026Incremental Learning in Mirror Flows22 Jun 2026
- 2026Policy Regret for Embedding Model Routing: Contextual Bandits with Low-Rank Experts12 Jun 2026
- 2026Online Learning on Hidden-Convex Losses via Algorithmic Equivalence: Optimal Regret, Geometric Barrier, and Bandit Feedback25 May 2026
- 2026Theoretical Analysis of Sparse Optimization with Reparameterization, Weight Decay, and Adaptive Learning Rate24 May 2026
- 2026HORST: Composing Optimizer Geometries for Sparse Transformer Training20 May 2026
- 2026Implicit Bias of Mirror Flow in Homogeneous Neural Networks: Sparse and Dense Feature Learning19 May 2026
- 2026Mirror Descent on Riemannian Manifolds18 Mar 2026
- 2026Gradient Flow Polarizes Softmax Outputs towards Low-Entropy Solutions6 Mar 2026
- 2026Never Saddle for Reparameterized Steepest Descent as Mirror Flow2 Mar 2026
- 2026Understanding and inverse design of implicit bias in stochastic learning: a geometric perspective10 Jan 2026
12 newest of 36 All 36 (JSON)