Lexicographic and Depth-Sensitive Margins in Homogeneous and Non-Homogeneous Deep Models
Not on the map. Why?
Abstract (arXiv)
With an eye toward understanding complexity control in deep learning, we study how infinitesimal regularization or gradient descent optimization lead to margin maximizing solutions in both homogeneous and non-homogeneous models, extending previous work that focused on infinitesimal regularization only in homogeneous models. To this end we study the limit of loss minimization with a diverging norm constraint (the "constrained path"), relate it to the limit of a "margin path" and characterize the resulting solution. For non-homogeneous ensemble models, which output is a sum of homogeneous sub-models, we show that this solution discards the shallowest sub-models if they are unnecessary. For homogeneous models, we show convergence to a "lexicographic max-margin solution", and provide conditions under which max-margin solutions are also attained as the limit of unconstrained gradient descent.
Key facts. Cited by 53 papers in the map (data through 14 Oct 2026). First posted on arXiv 17 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 11; 2023: 10; 2024: 11; 2025: 4; 2026: 6.
53 papers.
Sort:
By area:
- 2022Reconstructing Training Data from Trained Neural Networkscited by 133
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2022On the Implicit Bias in Deep-Learning Algorithmscited by 86
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2021On the Implicit Bias of Initialization Shape: Beyond Infinitesimal Mirror Descentcited by 70
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
- 2023Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix Sensingcited by 42
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
- 2021Understanding the role of importance weighting for deep learningcited by 39
38 more
- 2024Why Do You Grok? A Theoretical Analysis of Grokking Modular Additioncited by 36
- 2021On Margin Maximization in Linear and ReLU Networkscited by 34
- 2022Regression as Classification: Influence of Task Formulation on Neural Network Featurescited by 29
- 2021Properties of the After Kernelcited by 27
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networkscited by 27
- 2023On the Implicit Bias of Adamcited by 27
- 2022The Asymmetric Maximum Margin Bias of Quasi-Homogeneous Neural Networkscited by 25
- 2021Towards Understanding Learning in Neural Networks with Linear Teacherscited by 19
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attentioncited by 19
- 2021Inductive Bias of Multi-Channel Linear Convolutional Networks with Bounded Weight Normcited by 18
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networkscited by 18
- 2021On the Equivalence between Neural Network and Support Vector Machinecited by 17
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimizationcited by 17
- 2021Optimal Gradient-based Algorithms for Non-concave Bandit Optimizationcited by 15
- 2023The Implicit Bias of Minima Stability in Multivariate Shallow ReLU Networkscited by 15
- 2024A Short Survey on Importance Weighting for Machine Learningcited by 15
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networkscited by 11
- 2025Generalization or Hallucination? Understanding Out-of-Context Reasoning in Transformerscited by 10
- 2021Going Beyond Linear RL: Sample Efficient Neural Function Approximationcited by 9
- 2024Implicit Bias of Mirror Flow on Separable Datacited by 9
- 2023ReLU Neural Networks with Linear Layers are Biased Towards Single- and Multi-Index Modelscited by 8
- 2022Thinking Outside the Ball: Optimal Learning with Gradient Descent for Generalized Linear Stochastic Convex Optimizationcited by 7
- 2023A Quadratic Synchronization Rule for Distributed Deep Learningcited by 5
- 2024The Price of Implicit Bias in Adversarially Robust Generalizationcited by 5
- 2026A Theory of How Pretraining Shapes Inductive Bias in Fine-Tuningcited by 5
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescalingcited by 4
- 2023Generator Born from Classifiercited by 4
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adamcited by 4
- 2022The Implicit Bias of Gradient Descent on Generalized Gated Linear Networkscited by 3
- 2024Stealing Training Graphs from Graph Neural Networkscited by 3
- 2021Integral representations of shallow neural network with Rectified Power Unit activation functioncited by 2
- 2026Breaking the Reversal Curse in Autoregressive Language Models via Identity Bridgecited by 2
- 2026Stability Annealing Selects the Implicit Bias of Smoothed Sign Descent: A Rate-Indexed Barrier Path on Separable Datacited by 2
- 2023Applying statistical learning theory to deep learningcited by 1
- 2025The late-stage training dynamics of (stochastic) subgradient descent on homogeneous neural networkscited by 1
- 2022From Optimization Dynamics to Generalization Bounds via Lojasiewicz Gradient Inequalitycited by 0
- 2022Iterative regularization in classification via hinge loss diagonal descentcited by 0
50 most cited of 53 All 53 (JSON)
- 2026Stability Annealing Selects the Implicit Bias of Smoothed Sign Descent: A Rate-Indexed Barrier Path on Separable Data7 Jul 2026
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026A Theory of How Pretraining Shapes Inductive Bias in Fine-Tuning23 Feb 2026
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networks18 Feb 2026
- 2026Breaking the Reversal Curse in Autoregressive Language Models via Identity Bridge2 Feb 2026
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adam2 Feb 2026
- 2025Generalization or Hallucination? Understanding Out-of-Context Reasoning in Transformers12 Jun 2025
- 2025Heavy-Ball Momentum Method in Continuous Time and Discretization Error Analysis3 Jun 2025
- 2025Embedding principle of homogeneous neural network for classification problem18 May 2025
- 2025The late-stage training dynamics of (stochastic) subgradient descent on homogeneous neural networks8 Feb 2025
- 2024Stealing Training Graphs from Graph Neural Networks17 Nov 2024
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networks29 Oct 2024
38 more
- 2024Why Do You Grok? A Theoretical Analysis of Grokking Modular Addition17 Jul 2024
- 2024Implicit Bias of Mirror Flow on Separable Data18 Jun 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learning10 Jun 2024
- 2024The Price of Implicit Bias in Adversarially Robust Generalization7 Jun 2024
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularization31 May 2024
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimization5 Apr 2024
- 2024A Short Survey on Importance Weighting for Machine Learning15 Mar 2024
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attention13 Mar 2024
- 2023Generator Born from Classifier5 Dec 2023
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokking30 Nov 2023
- 2023Applying statistical learning theory to deep learning26 Nov 2023
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescaling24 Nov 2023
- 2023A Quadratic Synchronization Rule for Distributed Deep Learning22 Oct 2023
- 2023On the Implicit Bias of Adam31 Aug 2023
- 2023The Implicit Bias of Minima Stability in Multivariate Shallow ReLU Networks30 Jun 2023
- 2023ReLU Neural Networks with Linear Layers are Biased Towards Single- and Multi-Index Models24 May 2023
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stability19 May 2023
- 2023Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix Sensing27 Jan 2023
- 2022Iterative regularization in classification via hinge loss diagonal descent24 Dec 2022
- 2022Regression as Classification: Influence of Task Formulation on Neural Network Features10 Nov 2022
- 2022The Asymmetric Maximum Margin Bias of Quasi-Homogeneous Neural Networks7 Oct 2022
- 2022On the Implicit Bias in Deep-Learning Algorithms26 Aug 2022
- 2022Reconstructing Training Data from Trained Neural Networks15 Jun 2022
- 2022Neural Collapse: A Review on Modelling Principles and Generalization8 Jun 2022
- 2022Thinking Outside the Ball: Optimal Learning with Gradient Descent for Generalized Linear Stochastic Convex Optimization27 Feb 2022
- 2022From Optimization Dynamics to Generalization Bounds via Lojasiewicz Gradient Inequality22 Feb 2022
- 2022The Implicit Bias of Gradient Descent on Generalized Gated Linear Networks5 Feb 2022
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networks30 Jan 2022
- 2022Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networks27 Jan 2022
- 2021Integral representations of shallow neural network with Rectified Power Unit activation function20 Dec 2021
- 2021On the Equivalence between Neural Network and Support Vector Machine11 Nov 2021
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapse6 Oct 2021
- 2021On Margin Maximization in Linear and ReLU Networks6 Oct 2021
- 2021Going Beyond Linear RL: Sample Efficient Neural Function Approximation14 Jul 2021
- 2021Optimal Gradient-based Algorithms for Non-concave Bandit Optimization9 Jul 2021
- 2021Properties of the After Kernel21 May 2021
- 2021Understanding the role of importance weighting for deep learning28 Mar 2021
50 newest of 53 All 53 (JSON)
- 2024Stealing Training Graphs from Graph Neural Networkscited by 3
- 2024Stealing Training Graphs from Graph Neural Networks17 Nov 2024
- 2022Reconstructing Training Data from Trained Neural Networkscited by 133
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2022On the Implicit Bias in Deep-Learning Algorithmscited by 86
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2021On the Implicit Bias of Initialization Shape: Beyond Infinitesimal Mirror Descentcited by 70
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
- 2023Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix Sensingcited by 42
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
- 2021Understanding the role of importance weighting for deep learningcited by 39
12 most cited of 50 All 50 (JSON)
- 2026Stability Annealing Selects the Implicit Bias of Smoothed Sign Descent: A Rate-Indexed Barrier Path on Separable Data7 Jul 2026
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026A Theory of How Pretraining Shapes Inductive Bias in Fine-Tuning23 Feb 2026
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networks18 Feb 2026
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adam2 Feb 2026
- 2025Heavy-Ball Momentum Method in Continuous Time and Discretization Error Analysis3 Jun 2025
- 2025Embedding principle of homogeneous neural network for classification problem18 May 2025
- 2025The late-stage training dynamics of (stochastic) subgradient descent on homogeneous neural networks8 Feb 2025
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networks29 Oct 2024
- 2024Why Do You Grok? A Theoretical Analysis of Grokking Modular Addition17 Jul 2024
- 2024Implicit Bias of Mirror Flow on Separable Data18 Jun 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
12 newest of 50 All 50 (JSON)