Exponential Convergence of (Stochastic) Gradient Descent for Separable Logistic Regression

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kale, Sacchit, Manupriya, Piyushi, Marion, Pierre, Bach, Francis, Raj, Anant
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912929343864832
author Kale, Sacchit
Manupriya, Piyushi
Marion, Pierre
Bach, Francis
Raj, Anant
author_facet Kale, Sacchit
Manupriya, Piyushi
Marion, Pierre
Bach, Francis
Raj, Anant
contents Gradient descent and stochastic gradient descent are central to modern machine learning, yet their behavior under large step sizes remains theoretically unclear. Recent work suggests that acceleration often arises near the edge of stability, where optimization trajectories become unstable and difficult to analyze. Existing results for separable logistic regression achieve faster convergence by explicitly leveraging such unstable regimes through constant or adaptive large step sizes. In this paper, we show that instability is not inherent to acceleration. We prove that gradient descent with a simple, non-adaptive increasing step-size schedule achieves exponential convergence for separable logistic regression under a margin condition, while remaining entirely within a stable optimization regime. The resulting method is anytime and does not require prior knowledge of the optimization horizon or target accuracy. We also establish exponential convergence of stochastic gradient descent using a lightweight adaptive step-size rule that avoids line search and specialized procedures, improving upon existing polynomial-rate guarantees. Together, our results demonstrate that carefully structured step-size growth alone suffices to obtain exponential acceleration for both gradient descent and stochastic gradient descent.
format Preprint
id arxiv_https___arxiv_org_abs_2602_18946
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Exponential Convergence of (Stochastic) Gradient Descent for Separable Logistic Regression
Kale, Sacchit
Manupriya, Piyushi
Marion, Pierre
Bach, Francis
Raj, Anant
Machine Learning
Optimization and Control
Gradient descent and stochastic gradient descent are central to modern machine learning, yet their behavior under large step sizes remains theoretically unclear. Recent work suggests that acceleration often arises near the edge of stability, where optimization trajectories become unstable and difficult to analyze. Existing results for separable logistic regression achieve faster convergence by explicitly leveraging such unstable regimes through constant or adaptive large step sizes. In this paper, we show that instability is not inherent to acceleration. We prove that gradient descent with a simple, non-adaptive increasing step-size schedule achieves exponential convergence for separable logistic regression under a margin condition, while remaining entirely within a stable optimization regime. The resulting method is anytime and does not require prior knowledge of the optimization horizon or target accuracy. We also establish exponential convergence of stochastic gradient descent using a lightweight adaptive step-size rule that avoids line search and specialized procedures, improving upon existing polynomial-rate guarantees. Together, our results demonstrate that carefully structured step-size growth alone suffices to obtain exponential acceleration for both gradient descent and stochastic gradient descent.
title Exponential Convergence of (Stochastic) Gradient Descent for Separable Logistic Regression
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2602.18946