A general recipe for data-adaptive anytime-valid envelopes for iterative algorithms
Organizations: University of Pennsylvania · Stanford University
Abstract
Most convergence guarantees for iterative stochastic algorithms are stated at a fixed iteration count and do not justify stopping at a data-dependent time. Recent work has constructed anytime-valid envelopes that do, but deterministic envelopes are typically too conservative for practical stopping, and existing data-adaptive ones are derived case-by-case, each coupling a specific algorithm to a specific noise model and boundary construction. This paper provides a general recipe for constructing such anytime-valid data-adaptive boundaries for a large class of iterative algorithms. Performance measures of many of these algorithms satisfy a one-step recursion that reduces the entire error analysis to a single general requirement: time-uniform concentration for predictably weighted sums of noise. We formalize this through a strongly adapted sub- condition, and derive off-the-shelf boundary constructions via continuous and discrete mixtures, stitching, and product betting martingales. We show how these boundaries can be directly coupled with a separate algorithmic analysis that derives the one-step recursion to provide a class of readymade anytime-valid envelopes across diverse settings including stochastic gradient descent (SGD), temporal difference learning, and Bures-Wasserstein SGD. In particular, for SGD, our framework either recovers or significantly sharpens every existing envelope we are aware of. Our constructions come with explicit implementation details and are substantiated by numerical experiments.
Figures & tables
| Conditions on | ||
|---|---|---|
| \begin{array}[]{@{}l@{}}\|\xi_{t}\|\leq c_{t}\\ \mathbb{E}_{t-1}[\xi_{t}]=0\end{array} | ||
| \begin{array}[]{@{}l@{}}\mathbb{E}_{t-1}[\|\xi_{t}\|^{2}]<\infty\\ \mathbb{E}_{t-1}[\xi_{t}]=0\end{array} | ||
| \begin{array}[]{@{}l@{}}\mathbb{E}_{t-1}[\|\xi_{t}\|^{3}]<\infty\\ \mathbb{E}_{t-1}[\xi_{t}]=0\end{array} | ||
| \begin{array}[]{@{}l@{}}\|\xi_{t}\|\leq c\\ \mathbb{E}_{t-1}[\xi_{t}]=0\end{array} | ||
| \begin{array}[]{@{}l@{}}\|\xi_{t}\|\leq 1/2\\ \mathbb{E}_{t-1}[\xi_{t}]=0\end{array} | \begin{array}[]{@{}l@{}}(\xi_{t}-\widehat{\mu}_{t-1})(\xi_{t}-\widehat{\mu}_{t-1})^{\top},\\ \widehat{\mu}_{t-1}=\frac{1}{t-1}\sum_{s=1}^{t-1}\xi_{s}\end{array} |
| (betting) | |||||
|---|---|---|---|---|---|
| Asm. | Name | Condition | Remarks |
|---|---|---|---|
| 2 | QC γ : Quasar-convexity | , unique; : star-convexity ( StC ) [ Hinder et al., 2020 ] | |
| 3 | SC : Strong convexity | ||
| 4 | RSI : Restricted secant | ; [ Vaswani et al., 2019 ] | |
| 5 | LSC : Local self-concordance | , , [ Bach, 2010 ] | |
| 6 | KŁ ⋆ : Kurdyka–Łojasiewicz | nondecreasing, for ; Example : PŁ [ Scaman et al., 2022 , Polyak, 1963 ] | |
| 7 | DISS : Dissipativity |
| Asm. | Name | Condition | Remarks |
|---|---|---|---|
| Regularity of | |||
| 8 | LC : Lipschitz continuity | ||
| 9 | LS : -smoothness | ||
| 10 | LG : Linear growth | ||
| Metric | Measures | Last iterate |
|---|---|---|
| 8.1 | Squared distance † | |
| 8.2 | Suboptimality | |
| 8.3 | Gradient norm | |
| 8.4 | Average suboptimality | |
| 8.5 | Average squared gradient norm |
| Assumption | Transferred bound | Proof |
| Base envelope: | ||
| LC | LC | |
| LS L | Equation ( 19 ) with | |
| LS | ||
| LG K | LG | |
| Assumption 5 | ||
Stochastic Optimization
Stochastic optimization studies how to find good decisions or model parameters when objectives, data, or feedback are random or noisy.
See all 210 papers