math.OCNov 5, 2026

Last-Iterate Convergence of Stochastic Gradient Methods for Unbounded Problems

Authors: Kevin Kurian Thomas Vaidyan, Ahmet Alacaoglu, Michael P. Friedlander

Organizations: Department of Computer Science, University of British Columbia · Department of Computer Science and Mathematics, University of British Columbia

Abstract

Last-iterate convergence guarantees for stochastic gradient methods have attracted significant attention as practical implementations typically return the last iterate instead of an average of the iterates. In this paper, we study two approaches for analyzing the last-iterate convergence of proximal and projected stochastic gradient methods without bounded domain or bounded variance assumptions, where black-box variational inference serves as a running example. Under two related assumptions relaxing the bounded variance, we show near-optimal anytime rate O~(1/T)\widetilde{\mathcal{O}}(1/\sqrt{T}) for the last iterate. Moreover, when the horizon TT is known, we remove the logarithmic factor and obtain the optimal order O(1/T)\mathcal{O}(1/\sqrt{T}), matching the order known under bounded variance or bounded domain assumptions.

Figures & tables

Explore similar work

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.
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. We prove that Θ(log⁡(T)/T)Θ(\sqrt{\log(T)/T}) is the optimal anytime rate attainable with a fixed stepsize schedule (that does not depend on the total number of steps TT). The upper bound holds already for SGD, and improves on the log⁡(T)/T\log(T)/\sqrt{T} rate of Shamir and Zhang [2013], as well as the log⁡(T)/T\sqrt{\log(T)/T} rate of Zamani and Glineur [2023] which was proven only for deterministic GD. The matching lower bound holds already for GD, and confirms the conjecture of Jain et al. [2019], namely that the optimal 1/T1/\sqrt{T} rate for the last iterate cannot be obtained without prior knowledge of TT.
Sep 26, 2026

Revisiting AdaGrad in Stochastic Convex Optimization: Last Iterates, High Probability, and Lower Bounds

AdaGrad and AdaGrad-Norm are widely used adaptive methods, but their precise behavior in stochastic convex optimization remains less understood. We first show that AdaGrad-Norm and AdaGrad do not admit any universal last-iterate rate, even under sub-Gaussian noise and bounded iterates. We then prove that bounded variance alone is too weak: even with bounded iterates, it cannot yield high-probability average-iterate rates, and without bounded iterates it may not even guarantee convergence in expectation. Besides, we construct tight Ω(log⁡T/T)Ω(\log T/\sqrt T) lower bounds for both AdaGrad-Norm and AdaGrad under sub-Gaussian noise, showing that log⁡T\log T in existing average-iterate upper bounds is unavoidable. Finally, we show that this log⁡T\log T loss disappears once bounded-iterate condition is imposed: under a general ABC condition, both methods achieve rates O(1/T){O}(1/\sqrt{T}).