Gradient Descent on Logistic Regression with Non-Separable Data and Large Step Sizes
cs.LG, math.OC
Submitted: 2024-06-07
Updated: 2026-09-02
License: http://creativecommons.org/licenses/by/4.0/
The gist: We study gradient descent (GD) dynamics on logistic regression problems with large, constant step sizes.
Terminology
Abstract
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.
Sources
- Acceleration by Stepsize Hedging I: Multi-Step Descent and the Silver Stepsize Schedule
- Layer Normalization
- Self-concordant analysis for logistic regression
- Squareplus: A Softplus-Like Algebraic Rectifier
- Beyond the Edge of Stability via Two-step Gradient Updates
- Handbook of Convergence Theorems for (Stochastic) Gradient Methods
- Accelerated Gradient Descent via Long Steps
- Directional Smoothness and Gradient Methods: Convergence and Adaptivity
- Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapult
- Quadratic models for understanding catapult dynamics of neural networks
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks