Gradient Descent on Logistic Regression with Non-Separable Data and Large Step Sizes
Not on the map. Why?
Abstract (arXiv)
We study gradient descent (GD) dynamics on logistic regression problems with large, constant step sizes. For linearly-separable data, it is known that GD converges to the minimizer with arbitrarily large step sizes, a property which no longer holds when the problem is not separable. In fact, the behaviour can be much more complex -- a sequence of period-doubling bifurcations begins at the critical step size 2/λ, where λ is the largest eigenvalue of the Hessian at the solution. Using a smaller-than-critical step size guarantees convergence if initialized nearby the solution: but does this suffice globally? In one dimension, we show that a step size less than 1/λ suffices for global convergence. However, for all step sizes between 1/λ and the critical step size 2/λ, one can construct a dataset such that GD converges to a stable cycle. In higher dimensions, this is actually possible even for step sizes less than 1/λ. Our results show that although local convergence is guaranteed for all step sizes less than the critical step size, global convergence is not, and GD may instead converge to a cycle depending on the initialization.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 25 of its 40 references that are in the map. First posted on arXiv 7 Jun 2024.
Builds on
25 of 40 references.
References · 25
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 206
- 2023Handbook of Convergence Theorems for (Stochastic) Gradient Methodscited by 158
- 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 151
- 2022Understanding the unstable convergence of gradient descentcited by 84
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 54
- 2022Learning threshold neurons via the "edge of stability"cited by 50
- 2023The Implicit Regularization of Dynamical Stability in Stochastic Gradient Descentcited by 44
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 34
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 29
13 more
- 2020Stochasticity of Deterministic Gradient Descent: Large Learning Rate for Multiscale Objective Functioncited by 26
- 2023Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theorycited by 24
- 2023Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapultcited by 23
- 2023Gradient Descent Monotonically Decreases the Sharpness of Gradient Flow Solutions in Scalar Networks and Beyondcited by 20
- 2023Acceleration by Stepsize Hedging I: Multi-Step Descent and the Silver Stepsize Schedulecited by 19
- 2021Squareplus: A Softplus-Like Algebraic Rectifiercited by 18
- 2022Quadratic models for understanding catapult dynamics of neural networkscited by 17
- 2023From Stability to Chaos: Analyzing Gradient Descent Dynamics in Quadratic Regressioncited by 16
- 2024Directional Smoothness and Gradient Methods: Convergence and Adaptivitycited by 15
- 2021Provable Super-Convergence with a Large Cyclical Learning Ratecited by 10
- 2023Accelerated Gradient Descent via Long Stepscited by 9
- 2023Gradient Descent Converges Linearly for Logistic Regression on Separable Datacited by 7
- 2024Dynamical stability and chaos in artificial neural network trajectories along trainingcited by 6
Built on by
Citing papers per year: 2024: 2; 2025: 4; 2026: 3.
9 papers.
- 2024The Optimization Landscape of SGD Across the Feature Learning Strengthcited by 13
- 2025Large Stepsizes Accelerate Gradient Descent for Regularized Logistic Regressioncited by 7
- 2024From Logistic Regression to the Perceptron Algorithm: Exploring Gradient Descent with Large Step Sizescited by 5
- 2025Gradient Descent with Large Step Sizes: Chaos and Fractal Convergence Regioncited by 5
- 2025Gradient Descent on Logistic Regression: Do Large Step-Sizes Work with Data on the Sphere?cited by 2
- 2025Any-stepsize Gradient Descent for Separable Data under Fenchel-Young Lossescited by 1
- 2026Finite-Sample Performance of Gradient Descent in Logistic Regression with Gaussian Designcited by 1
- 2026Tight Bounds for Logistic Regression with Large Stepsize Gradient Descent in Low Dimensioncited by 0
- 2026Non-asymptotic implicit bias of logistic regression at early-stage gradient descent dynamicscited by 0
- 2026Non-asymptotic implicit bias of logistic regression at early-stage gradient descent dynamics5 Aug 2026
- 2026Finite-Sample Performance of Gradient Descent in Logistic Regression with Gaussian Design19 Jun 2026
- 2026Tight Bounds for Logistic Regression with Large Stepsize Gradient Descent in Low Dimension12 Feb 2026
- 2025Gradient Descent with Large Step Sizes: Chaos and Fractal Convergence Region29 Sep 2025
- 2025Gradient Descent on Logistic Regression: Do Large Step-Sizes Work with Data on the Sphere?15 Jul 2025
- 2025Large Stepsizes Accelerate Gradient Descent for Regularized Logistic Regression3 Jun 2025
- 2025Any-stepsize Gradient Descent for Separable Data under Fenchel-Young Losses7 Feb 2025
- 2024From Logistic Regression to the Perceptron Algorithm: Exploring Gradient Descent with Large Step Sizes11 Dec 2024
- 2024The Optimization Landscape of SGD Across the Feature Learning Strength6 Oct 2024