Stopping Rules for Stochastic Gradient Descent via Anytime-Valid Confidence Sequences
Not on the map. Why?
Abstract (arXiv)
The problem of stopping stochastic gradient descent (SGD) in an online manner, based solely on the observed trajectory, is a challenging theoretical problem with significant consequences for applications. While SGD is routinely monitored as it runs, the classical theory of SGD provides guarantees only at pre-specified iteration horizons and offers no valid way to decide, based on the observed trajectory, when further computation is justified. We address this longstanding gap by developing anytime-valid confidence sequences for stochastic gradient methods, which remain valid under continuous monitoring and directly induce statistically valid, trajectory-dependent stopping rules: stop as soon as the current upper confidence bound on an appropriate performance measure falls below a user-specified tolerance. The confidence sequences are constructed using nonnegative supermartingales, are time-uniform, and depend only on observable quantities along the SGD trajectory, without requiring prior knowledge of the optimization horizon. In convex optimization, this yields anytime-valid certificates for weighted suboptimality of projected SGD under general stepsize schedules, without assuming smoothness or strong convexity. In nonconvex optimization, it yields time-uniform certificates for weighted first-order stationarity under smoothness assumptions. We further characterize the stopping-time complexity of the resulting stopping rules under standard stepsize schedules. To the best of our knowledge, this is the first framework that provides statistically valid, time-uniform stopping rules for SGD across both convex and nonconvex settings based solely on its observed trajectory.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 36 of its 64 references that are in the map. First posted on arXiv 15 Dec 2025.
Builds on
36 of 64 references.
References · 36
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
- 2018SPIDER: Near-Optimal Non-Convex Optimization via Stochastic Path Integrated Differential Estimatorcited by 287
- 2014On the Complexity of Best Arm Identification in Multi-Armed Bandit Modelscited by 286
- 2017How to Escape Saddle Points Efficientlycited by 285
- 2015Escaping From Saddle Points --- Online Stochastic Gradient for Tensor Decompositioncited by 243
- 2016Optimal Best Arm Identification with Fixed Confidencecited by 239
- 2018Time-uniform, nonparametric, nonasymptotic confidence sequencescited by 229
- 2017SARAH: A Novel Method for Machine Learning Problems Using Stochastic Recursive Gradientcited by 220
- 2018AdaGrad stepsizes: Sharp convergence over nonconvex landscapescited by 182
- 2020Estimating means of bounded random variables by bettingcited by 169
- 2022Game-theoretic statistics and safe anytime-valid inferencecited by 155
24 more
- 2016Stochastic Variance Reduction for Nonconvex Optimizationcited by 145
- 2020PAGE: A Simple and Optimal Probabilistic Gradient Estimator for Nonconvex Optimizationcited by 119
- 2013Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n)cited by 111
- 2020Stochastic Optimization with Heavy-Tailed Noise via Accelerated Gradient Clippingcited by 104
- 2018Tight Analyses for Non-Smooth Stochastic Gradient Descentcited by 97
- 2021High-probability Bounds for Non-Convex Stochastic Optimization with Heavy Tailscited by 80
- 2018Mixture Martingales Revisited with Applications to Sequential Tests and Confidence Intervalscited by 76
- 2013Accelerated Gradient Methods for Nonconvex Nonlinear and Stochastic Programmingcited by 75
- 2024Hypothesis testing with e-valuescited by 74
- 2017Natasha 2: Faster Non-Convex Optimization Than SGDcited by 49
- 2022High Probability Bounds for a Class of Nonconvex Algorithms with AdaGrad Stepsizecited by 40
- 2016Asymptotic Optimality in Stochastic Optimizationcited by 38
- 2021Nonparametric Two-Sample Testing by Bettingcited by 37
- 2022Anytime-valid off-policy inference for contextual banditscited by 37
- 2019Simple and optimal high-probability bounds for strongly-convex stochastic gradient descentcited by 27
- 2023Auditing Fairness by Bettingcited by 25
- 2016Less than a Single Pass: Stochastically Controlled Stochastic Gradient Methodcited by 18
- 2020Optimal Best-Arm Identification Methods for Tail-Risk Measurescited by 18
- 2025Can SGD Handle Heavy-Tailed Noise?cited by 17
- 2023Asymptotic normality and optimality in nonsmooth stochastic approximationcited by 14
- 2025Instance-Log-Optimality of Portfolio-Based E-Processes and their Sequential Hypothesis Testscited by 12
- 2025Optimistic Interior Point Methods for Sequential Hypothesis Testing by Bettingcited by 4
- 2023The Anytime Convergence of Stochastic Gradient Descent with Momentum: From a Continuous-Time Perspectivecited by 3
- 2025The Root Finding Problem Revisited: Beyond the Robbins-Monro procedurecited by 2
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Beyond Optimal Rates in Stochastic Optimization: Trajectory-Adaptive Stopping Rulescited by 1
- 2026The Exact Time-Uniform Rate Frontier for Stochastic Gradient Descent on Smooth Convex Objectivescited by 1
- 2026ScoreStop: Gradient-based early stopping using functional score testscited by 0
- 2026Exact information accounting for SGD methodscited by 0
- 2026Exact information accounting for SGD methods30 Sep 2026
- 2026The Exact Time-Uniform Rate Frontier for Stochastic Gradient Descent on Smooth Convex Objectives8 Sep 2026
- 2026Beyond Optimal Rates in Stochastic Optimization: Trajectory-Adaptive Stopping Rules26 Aug 2026
- 2026ScoreStop: Gradient-based early stopping using functional score tests1 Jun 2026