cs.LGOct 18, 2026

Near-Optimal Last-Iterate Convergence of SGD in the Smooth Interpolation Regime

Authors: Matan Schliserman, Tomer Koren

Organizations: Blavatnik School of Computer Science and AI, Tel Aviv University · Blavatnik School of Computer Science and AI, Tel Aviv University, and Google Research Tel Aviv

Abstract

We study the last iterate of stochastic gradient descent (SGD) with a constant stepsize on smooth convex objectives in the low-noise and interpolation regimes, where the variance σ⋆2σ_\star^2 of the stochastic gradients at the optimum is small or zero. We establish that after TT steps of SGD on ββ-smooth convex losses with any constant stepsize 0<η<2/β0<η<2/β, the last iterate attains expected excess risk O~(1η(2−βη)3T+η(2−βη)4σ⋆2)\widetilde O\bigl(\tfrac{1}{η(2-βη)^3T}+\tfracη{(2-βη)^4}σ_\star^2\bigr), which gives the rate O~(1/T)\smash{\widetilde O(1/T)} for every constant stepsize in the interpolation regime (σ⋆=0σ_\star=0), matching the Ω(1/T)Ω(1/T) lower bound up to logarithmic factors and improving upon the O(1/T)O(1/\sqrt T) rate of Attia et al. (2026) for η=1/βη=1/β. Our analysis is based on a new technique, which compares the second moments of the stochastic gradients along the SGD trajectory across only logarithmically many time scales. For quadratic losses in the interpolation regime, an adaptation of this technique gives an O(1/T)O(1/T) rate for every constant stepsize, extending the recent optimal rate for the randomized Kaczmarz method of Derezinski and Dong (2026) to stepsizes above the greedy stepsize with an arguably simpler proof.

Figures & tables

Explore similar work

Apr 15, 2026

Gradient Descent's Last Iterate is Often (slightly) Suboptimal

We consider the well-studied setting of minimizing a convex Lipschitz function using either gradient descent (GD) or its stochastic variant (SGD), and examine the last iterate convergence. By now, it is known that standard stepsize choices lead to a last iterate convergence rate of log⁡T/T\log T/\sqrt{T} after TT steps. A breakthrough result of Jain et al. [2019] recovered the optimal 1/T1/\sqrt{T} rate by constructing a non-standard stepsize sequence. However, this sequence requires choosing TT in advance, as opposed to common stepsize schedules which apply for any time horizon. Moreover, Jain et al. conjectured that without prior knowledge of TT, no stepsize sequence can ensure the optimal error for SGD's last iterate, a claim which so far remained unproven. We prove this conjecture, and in fact show that even in the noiseless case of GD, it is impossible to avoid an excess poly-log factor in TT when considering an anytime last iterate guarantee.
Apr 10, 2026

Last-Iterate Convergence of Randomized Kaczmarz and SGD with Greedy Step Size

We study last-iterate convergence of SGD with greedy step size over smooth quadratics in the interpolation regime, a setting which captures the classical Randomized Kaczmarz algorithm as well as other popular iterative linear system solvers. For these methods, we show that the tt-th iterate attains an O(1/t3/4)O(1/t^{3/4}) convergence rate, addressing a question posed by Attia, Schliserman, Sherman, and Koren, who gave an O(1/t1/2)O(1/t^{1/2}) guarantee for this setting. In the proof, we introduce the family of stochastic contraction processes, whose behavior can be described by the evolution of a certain deterministic eigenvalue equation, which we analyze via a careful discrete-to-continuous reduction.
Oct 13, 2026

Last-Iterate Convergence of Clipped SGD with Linear Decay

In this paper, we study the last-iterate convergence of clipped stochastic gradient descent under heavy-tailed noise. We assume that the stochastic gradients are unbiased and that the noise has a bounded central pp-th moment, where 1<p≤21<p\le2. For smooth convex objectives, a linearly decreasing learning rate gives a high-probability error bound of order O(LR2log⁡(1/δ)/T+σR(log⁡(1/δ)/T)(p−1)/p)O(LR^2\log(1/δ)/T+σR(\log(1/δ)/T)^{(p-1)/p}). We further establish last-iterate guarantees for nonsmooth composite objectives and smooth composite objectives with proximal updates. In addition, when a causal adversary replaces at most qq gradients, the same clipped update achieves a bound with log⁡(1/δ)\log(1/δ) replaced by log⁡(1/δ)+q\log(1/δ)+q. The corresponding lower bounds for the clean and corrupted oracle models are also provided.