Fast Margin Maximization via Dual Acceleration
Not on the map. Why?
Abstract (arXiv)
We present and analyze a momentum-based gradient method for training linear classifiers with an exponentially-tailed loss (e.g., the exponential or logistic loss), which maximizes the classification margin on separable data at a rate of widetildeO(1/t2). This contrasts with a rate of O(1/log(t)) for standard gradient descent, and O(1/t) for normalized gradient descent. This momentum-based method is derived via the convex dual of the maximum-margin problem, and specifically by applying Nesterov acceleration to this dual, which manages to result in a simple and intuitive method in the primal. This dual view can also be used to derive a stochastic variant, which performs adaptive non-uniform sampling via the dual variables.
Key facts. Cited by 33 papers in the map (data through 14 Oct 2026). Builds on 20 of its 34 references that are in the map. First posted on arXiv 1 Jul 2021.
Builds on
20 of 34 references.
References · 20
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2017Spectrally-normalized margin bounds for neural networkscited by 686
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2014In Search of the Real Inductive Bias: On the Role of Implicit Regularization in Deep Learningcited by 342
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 268
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 266
- 2018Learning Overparameterized Neural Networks via Stochastic Gradient Descent on Structured Datacited by 261
- 2018Implicit Bias of Gradient Descent on Linear Convolutional Networkscited by 251
- 2020Directional convergence and alignment in deep learningcited by 178
8 more
- 2018Convergence of Gradient Descent on Separable Datacited by 110
- 2018Risk and parameter convergence of logistic regressioncited by 85
- 2014Global convergence of the Heavy-ball method for convex optimizationcited by 61
- 2014Linear Coupling: An Ultimate Unification of Gradient and Mirror Descentcited by 58
- 2013Margins, Shrinkage, and Boostingcited by 35
- 2021Properties of the After Kernelcited by 27
- 2014Towards A Deeper Geometric, Analytic and Algorithmic Understanding of Marginscited by 4
- 2020Linear Separation via Optimismcited by 2
Built on by
Citing papers per year: 2021: 1; 2022: 7; 2023: 8; 2024: 11; 2025: 5; 2026: 1.
33 papers.
Sort:
- 2022On the Implicit Bias in Deep-Learning Algorithmscited by 86
- 2023Transformers as Support Vector Machinescited by 74
- 2023Max-Margin Token Selection in Attention Mechanismcited by 58
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 41
- 2022Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descentcited by 39
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 35
- 2024Implicit Bias and Fast Convergence Rates for Self-attentioncited by 31
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Timecited by 30
- 2023Benign Overfitting in Linear Classifiers and Leaky ReLU Networks from KKT Conditions for Margin Maximizationcited by 29
- 2022Mirror Descent Maximizes Generalized Margin and Can Be Implemented Efficientlycited by 26
- 2021Does Momentum Change the Implicit Regularization on Separable Data?cited by 22
21 more
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attentioncited by 19
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimizationcited by 17
- 2023A Unified Approach to Controlling Implicit Regularization via Mirror Descentcited by 14
- 2024Implicit Optimization Bias of Next-Token Prediction in Linear Modelscited by 14
- 2024Transformers Learn Low Sensitivity Functions: Investigations and Implicationscited by 14
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
- 2024Non-asymptotic Convergence of Training Transformers for Next-token Predictioncited by 13
- 2025The Rich and the Simple: On the Implicit Bias of Adam and SGDcited by 13
- 2024The Implicit Bias of Gradient Descent on Separable Multiclass Datacited by 11
- 2025How Transformers Learn Regular Language Recognition: A Theoretical Study on Training Dynamics and Implicit Biascited by 11
- 2024Optimizing Attention with Mirror Descent: Generalized Max-Margin Token Selectioncited by 9
- 2022On Generalization of Decentralized Learning with Separable Datacited by 8
- 2025Minimax Optimal Convergence of Gradient Descent in Logistic Regression via Large and Adaptive Stepsizescited by 7
- 2022On Accelerated Perceptrons and Beyondcited by 6
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescalingcited by 4
- 2024The Oracle Complexity of Simplex-based Matrix Gamescited by 4
- 2025Constant Stepsize Local GD for Logistic Regression: Acceleration by Instabilitycited by 2
- 2022On the Optimization of Margin Distributioncited by 0
- 2022Iterative regularization in classification via hinge loss diagonal descentcited by 0
- 2023Faster Margin Maximization Rates for Generic and Adversarially Robust Optimization Methodscited by 0
- 2026On the Rate of Convergence of GD in Non-linear Neural Networks: An Adversarial Robustness Perspectivecited by 0
- 2026On the Rate of Convergence of GD in Non-linear Neural Networks: An Adversarial Robustness Perspective2 Mar 2026
- 2025Constant Stepsize Local GD for Logistic Regression: Acceleration by Instability16 Jun 2025
- 2025The Rich and the Simple: On the Implicit Bias of Adam and SGD29 May 2025
- 2025How Transformers Learn Regular Language Recognition: A Theoretical Study on Training Dynamics and Implicit Bias2 May 2025
- 2025Minimax Optimal Convergence of Gradient Descent in Logistic Regression via Large and Adaptive Stepsizes5 Apr 2025
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Data7 Feb 2025
- 2024The Oracle Complexity of Simplex-based Matrix Games9 Dec 2024
- 2024The Implicit Bias of Gradient Descent on Separable Multiclass Data2 Nov 2024
- 2024Optimizing Attention with Mirror Descent: Generalized Max-Margin Token Selection18 Oct 2024
- 2024Non-asymptotic Convergence of Training Transformers for Next-token Prediction25 Sep 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularization31 May 2024
21 more
- 2024Implicit Regularization of Gradient Flow on One-Layer Softmax Attention13 Mar 2024
- 2024Transformers Learn Low Sensitivity Functions: Investigations and Implications11 Mar 2024
- 2024Implicit Optimization Bias of Next-Token Prediction in Linear Models28 Feb 2024
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiency24 Feb 2024
- 2024Implicit Bias and Fast Convergence Rates for Self-attention8 Feb 2024
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescaling24 Nov 2023
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Time14 Sep 2023
- 2023Transformers as Support Vector Machines31 Aug 2023
- 2023A Unified Approach to Controlling Implicit Regularization via Mirror Descent24 Jun 2023
- 2023Max-Margin Token Selection in Attention Mechanism23 Jun 2023
- 2023Faster Margin Maximization Rates for Generic and Adversarially Robust Optimization Methods27 May 2023
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stability19 May 2023
- 2023Benign Overfitting in Linear Classifiers and Leaky ReLU Networks from KKT Conditions for Margin Maximization2 Mar 2023
- 2022Iterative regularization in classification via hinge loss diagonal descent24 Dec 2022
- 2022On Accelerated Perceptrons and Beyond17 Oct 2022
- 2022On Generalization of Decentralized Learning with Separable Data15 Sep 2022
- 2022On the Implicit Bias in Deep-Learning Algorithms26 Aug 2022
- 2022Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descent8 Jul 2022
- 2022Mirror Descent Maximizes Generalized Margin and Can Be Implemented Efficiently25 May 2022
- 2022On the Optimization of Margin Distribution29 Apr 2022
- 2021Does Momentum Change the Implicit Regularization on Separable Data?8 Oct 2021
- 2022On the Implicit Bias in Deep-Learning Algorithmscited by 86
- 2023Transformers as Support Vector Machinescited by 74
- 2023Max-Margin Token Selection in Attention Mechanismcited by 58
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 41
- 2022Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descentcited by 39
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 35
- 2024Implicit Bias and Fast Convergence Rates for Self-attentioncited by 31
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Timecited by 30
- 2023Benign Overfitting in Linear Classifiers and Leaky ReLU Networks from KKT Conditions for Margin Maximizationcited by 29
- 2022Mirror Descent Maximizes Generalized Margin and Can Be Implemented Efficientlycited by 26
- 2021Does Momentum Change the Implicit Regularization on Separable Data?cited by 22
12 most cited of 33 All 33 (JSON)
- 2026On the Rate of Convergence of GD in Non-linear Neural Networks: An Adversarial Robustness Perspective2 Mar 2026
- 2025Constant Stepsize Local GD for Logistic Regression: Acceleration by Instability16 Jun 2025
- 2025The Rich and the Simple: On the Implicit Bias of Adam and SGD29 May 2025
- 2025How Transformers Learn Regular Language Recognition: A Theoretical Study on Training Dynamics and Implicit Bias2 May 2025
- 2025Minimax Optimal Convergence of Gradient Descent in Logistic Regression via Large and Adaptive Stepsizes5 Apr 2025
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Data7 Feb 2025
- 2024The Oracle Complexity of Simplex-based Matrix Games9 Dec 2024
- 2024The Implicit Bias of Gradient Descent on Separable Multiclass Data2 Nov 2024
- 2024Optimizing Attention with Mirror Descent: Generalized Max-Margin Token Selection18 Oct 2024
- 2024Non-asymptotic Convergence of Training Transformers for Next-token Prediction25 Sep 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularization31 May 2024
12 newest of 33 All 33 (JSON)