mlsanityData to 7 Oct 2026

Gradient Descent on Logistic Regression with Non-Separable Data and Large Step Sizes

arXiv 2406.05033 ↗First posted 7 Jun 2024cs.LG math.OCcited by 9

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

  1. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
  2. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
  3. 2020The large learning rate phase of deep learning: the catapult mechanismcited by 206
  4. 2023Handbook of Convergence Theorems for (Stochastic) Gradient Methodscited by 158
  5. 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 151
  6. 2022Understanding the unstable convergence of gradient descentcited by 84
  7. 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 54
  8. 2022Learning threshold neurons via the "edge of stability"cited by 50
  9. 2023The Implicit Regularization of Dynamical Stability in Stochastic Gradient Descentcited by 44
  10. 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
  11. 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 34
  12. 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 29
13 more
  1. 2020Stochasticity of Deterministic Gradient Descent: Large Learning Rate for Multiscale Objective Functioncited by 26
  2. 2023Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theorycited by 24
  3. 2023Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapultcited by 23
  4. 2023Gradient Descent Monotonically Decreases the Sharpness of Gradient Flow Solutions in Scalar Networks and Beyondcited by 20
  5. 2023Acceleration by Stepsize Hedging I: Multi-Step Descent and the Silver Stepsize Schedulecited by 19
  6. 2021Squareplus: A Softplus-Like Algebraic Rectifiercited by 18
  7. 2022Quadratic models for understanding catapult dynamics of neural networkscited by 17
  8. 2023From Stability to Chaos: Analyzing Gradient Descent Dynamics in Quadratic Regressioncited by 16
  9. 2024Directional Smoothness and Gradient Methods: Convergence and Adaptivitycited by 15
  10. 2021Provable Super-Convergence with a Large Cyclical Learning Ratecited by 10
  11. 2023Accelerated Gradient Descent via Long Stepscited by 9
  12. 2023Gradient Descent Converges Linearly for Logistic Regression on Separable Datacited by 7
  13. 2024Dynamical stability and chaos in artificial neural network trajectories along trainingcited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 2; 2025: 4; 2026: 3.

9 papers.