The Cost of an Unknown Horizon for the Last Iterate of Clipped SGD under Heavy-Tailed Noise
arXiv:2610.33381
Abstract
We study the last iterate of clipped SGD, and of every rule that queries its current iterate and steps along the resulting stochastic gradient, when training is stopped at a data-dependent time, so that guarantees must hold at all times simultaneously. On smooth convex objectives under heavy-tailed noise with a bounded central th moment, , we determine the optimal time-uniform rate of these rules. A clipped SGD schedule without horizon or confidence inputs attains at all times, with , also almost surely, and no such rule attains for any , even with adaptive step sizes and thresholds. This rate lies between the optimal rate for a known horizon , , and the previous time-uniform rate (Chezhegov et al., ICLR 2026); under finite variance, the exponents are , , and . Their conjecture that a better schedule reaches the optimal rate thus holds with a known horizon (Liu; Jiang et al.) but fails without one. Querying gradients at the running average attains at all times, whereas heavy-ball momentum with geometric forgetting and polynomially bounded steps cannot improve on .
arXiv comment44 pages, including appendices and references. Lean 4 proofs of Theorems 2, 4, and 8 and of the known-horizon bound of Theorem 1, and the script of the numerical illustration, are included as ancillary files; the Lean code is also at https://github.com/Thefootofhappiness/clipped-sgd-last-iterate-lean