acceptodds
Under review as a conference paper at ICLR 2027

High-Probability Convergence of Clipped SGD under Heavy-Tailed Noise and -Smoothness

Abstract

Gradient clipping is widely used in language-model training to control heavy-tailed gradient noise and can improve convergence guarantees over stochastic gradient descent **SGD** under -smoothness. Under these joint conditions, a central challenge is to obtain high-probability guarantees without exponential dependence on , where bounds the initial distance to a minimizer. We resolve this challenge for convex objectives, establishing, to the best of our knowledge, the first such guarantees for standard **Clip-SGD**. We assume unbiased stochastic gradients with bounded central -th moment, . Our bounds have only polylogarithmic dependence on the inverse failure probability and recover known deterministic generalized-smoothness rates when the noise vanishes, as well as the standard high-probability rate under heavy-tailed noise in the classical -smooth setting. The convex rate is attained by a computable output that averages iterates whose sampled stochastic gradients are not clipped, requiring neither function values nor extra oracle calls. At constant confidence, we establish a matching lower bound for **Clip-SGD** with any fixed stepsize and clipping level, showing that our convex stochastic rate is optimal up to logarithmic factors in the iteration budget. Our upper-bound analysis uses a directional clipping-bias bound to absorb part of the bias into the progress generated by the clipped population gradient, avoiding an exponentially large local smoothness constant. We also obtain nonconvex high-probability guarantees that recover the noiseless rate, match the classical-smooth stochastic iteration rate when , and have no explicit dependence in the asymptotically dominant stochastic term.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.