math.OCApr 15, 2026

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

Authors: Guy Kornowski, Ohad Shamir

Organizations: Institute for Advanced Study · Weizmann Institute of Science · University of Toronto and the Vector Institute

Abstract

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.

Explore similar work

Oct 14, 2026

On the Last-Iterate Convergence of Clipped Gradient Methods

In stochastic (strongly) convex optimization, establishing last-iterate convergence guarantees for first-order methods has recently garnered increasing attention, as this not only deepens the theoretical understanding of these algorithms but also aligns more closely with practice. Notably, people have demonstrated that one of the most famous, simple, and popular methods, Stochastic Gradient Descent (SGD\mathtt{SGD}), provably achieves last-iterate convergence. However, the existing results may have limited applicability, since they typically rely on strong conditions on the gradient noise (e.g., an exponentially decaying tail) rather than the more realistic heavy-tailed noise. Unfortunately, when faced with heavy-tailed noise, SGD\mathtt{SGD} is known to exhibit undesirable behavior or even fail to converge. To address the challenge of heavy-tailed noise, people have proposed Clipped-SGD\mathtt{Clipped}\text{-}\mathtt{SGD}, an algorithm that combines SGD\mathtt{SGD} with a simple mechanism, gradient clipping. Although Clipped-SGD\mathtt{Clipped}\text{-}\mathtt{SGD} performs well in practice, its last-iterate convergence property remains highly underexplored. In this work, we prove the first optimal last-iterate rates in high probability for Clipped-SGD\mathtt{Clipped}\text{-}\mathtt{SGD} under heavy-tailed noise, thereby closing a gap in the literature.
Jun 23, 2026

New Bounds for the Last Iterate of the Stochastic subGradient Method

We study the last iterate of the stochastic subgradient method for one-dimensional convex Lipschitz objectives. For a fixed horizon nn, we consider the standard fixed stepsizes η=Θ(1/n)η=Θ(1/\sqrt n). We prove that, for such stepsize policies, under additive i.i.d. subgradient noise with uniformly bounded variance, the last iterate features an optimization error of order 1/n1/\sqrt n, thereby removing the extra (log⁡n)(\log n) factor present in existing generic bounds. On the other hand, we show that without the i.i.d. assumption, the optimization error can be of order (log⁡n)/n(\log n)/\sqrt n. Thus, under the uniformly bounded variance assumption alone, the last iterate of SsGM is suboptimal even in dimension one, resolving negatively an open problem posed in Koren and Segal, COLT, 2020.
Oct 18, 2026

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

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.