Lune

NeurIPS2025Top-tier venue

Fast Last-Iterate Convergence of SGD in the Smooth Interpolation Regime

Amit Attia, Matan Schliserman, Uri Sherman, Tomer Koren

2025Year
18Citations
3Top-tier citations

Abstract

We study population convergence guarantees of stochastic gradient descent (SGD) for smooth convex objectives in the interpolation regime, where the noise at optimum is zero or near zero. The behavior of the last iterate of SGD in this setting -- particularly with large (constant) stepsizes -- has received growing attention in recent years due to implications for the training of over-parameterized models, as well as to analyzing forgetting in continual learning and to understanding the convergence of the randomized Kaczmarz method for solving linear systems. We establish that after TT steps of SGD on β\beta-smooth convex loss functions with stepsize 0<η<2/β0<\eta<2/\beta, the last iterate exhibits expected excess risk O~(1η(2−βη)T1−βη/2+η(2−βη)2Tβη/2σ⋆2)\widetilde{O}(\frac{1}{\eta (2-\beta \eta) T^{1-\beta\eta/2}} + \frac{\eta}{(2-\beta\eta)^2} T^{\beta\eta/2} \sigma_\star^2), where σ⋆2\sigma_\star^2 denotes the variance of the stochastic gradients at the optimum. In particular, for a well-tuned stepsize we obtain a near optimal O~(1/T+σ⋆/T)\widetilde{O}(1/T + \sigma_\star/\sqrt{T}) rate for the last iterate, extending the results of Varre et al. (2021) beyond least squares regression; and when σ⋆=0\sigma_\star=0 we obtain a rate of O(1/T)\smash{O(1/\sqrt T)} with η=1/β\eta=1/\beta, improving upon the best-known O(T−1/4)\smash{O(T^{-1/4})} rate recently established by Evron et al. (2025) in the special case of realizable linear regression.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 43c3dd31-75ce-4a64-bfaf-5fad05c06290

Cited by top-tier papers3

Ask how each one uses it

Builds on18

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines