Making Gradient Descent Optimal for Strongly Convex Stochastic Optimization 1: SGD with α-Suffix Averaging Attains E[F(w̄ᵅ_T)] − F(w*) ≤ (2 + 2.5 log(1/(1−α)))G²/(αλT)Research Paper
Motivation
Stochastic gradient descent (SGD) is the default method for minimizing an expected loss when only noisy gradients are available, which is the situation of nearly every large-scale learning problem. When is strongly convex (for instance, any convex loss with an regularizer), information-theoretic lower bounds show that no method using noisy gradients can guarantee an error better than order , and some algorithms achieve this rate.
For plain SGD with step sizes and the standard averaged output, however, the classical analysis gives only . Rakhlin, Shamir and Sridharan (arXiv:1109.5647, ICML 2012) showed that the logarithm is real for full averaging on non-smooth problems, and that a minimal change to the output, averaging only the last fraction of the iterates, restores the optimal rate. This mission formalizes that positive result, the paper's main theorem.
Timeline of the relevant results:
- 2007. Hazan, Agarwal and Kale (Machine Learning 69) prove an regret bound for online gradient descent on strongly convex losses, which by online-to-batch conversion gives for averaged SGD.
- 2009. Nemirovski, Juditsky, Lan and Shapiro (SIAM J. Optim. 19) show an rate for the last SGD iterate when is in addition smooth.
- 2011. Hazan and Kale (COLT 2011) obtain for non-smooth strongly convex problems with a different, epoch-based algorithm.
- 2012. Rakhlin, Shamir and Sridharan prove that SGD with full averaging can suffer , and that α-suffix averaging achieves (Theorem 5).
Setting
Let be a closed convex set containing , and let . The function is -strongly convex on () if
A point is a minimizer of over .
Information about comes from a stochastic gradient oracle : at a query point , a sample is drawn and is returned. The oracle is unbiased in the sense that its mean is a subgradient of relative to : for all . Its second moment is bounded: at every .
Projected SGD with step sizes runs on an i.i.d. sample :
where is the Euclidean projection onto . For with an integer, the α-suffix average is
In Lean the iterates are UnderstandingML.sgdStrongIterates lam W g S, the oracle mean is oracleMean D g, the suffix average is suffixAverage lam W g S k with , and expectations are integrals against the product law UnderstandingML.iidLaw D T.
Formalization targets
Goal: Theorem 5
together with the finiteness of . The constants are the paper's.
Milestones, in the order the proof uses them
- Lemma 2 (p. 12): at implies .
- Lemma 1 (p. 4): for every .
- (3) (p. 6): .
- App. B.6, convexity display (p. 18): .
- App. B.6, penultimate bound (p. 18): with .
- App. B.6, harmonic bound (p. 18): .
Significance
Theorem 5 shows that the suboptimal factor of averaged SGD on non-smooth strongly convex problems is an artifact of which iterates are averaged, not of the gradient steps themselves: the same iterates, averaged over a constant fraction at the end, reach the minimax-optimal rate up to constants. This matches Hazan and Kale's guarantee with an unmodified algorithm, and the same paper's lower bound shows that full averaging cannot match it.
The result is proved in the paper; nothing in it is formalized. The platform holds the Understanding Machine Learning SGD model and its bound for full averaging (UnderstandingML.sgd_strongly_convex), but no rate for SGD on a non-smooth objective. The mission produces a machine-checked proof of Theorem 5 with its explicit constants and, along the way, of Lemma 1 (the mean-square convergence of the last iterate), which is reused by every analysis of SGD with step sizes , including the paper's smooth-case Theorems 1 and 2.
Difficulty
The obvious argument, the online-gradient-descent regret bound summed over all rounds and divided by , produces and so cannot avoid the logarithm. Discarding early iterates does not by itself help: the resulting bound depends on the distance of the iterate at which the suffix starts, at the scale , and the regret analysis says nothing about individual iterates of a non-smooth problem.
The formal difficulties are probabilistic. The iterates are random vectors built by composing a measurable oracle with a projection defined by choice; conditioning on the past to evaluate requires the product structure of the sample; and every expectation must be shown finite, including that of with not assumed measurable.
Formalization scope
- Space. with the Euclidean inner product (
UnderstandingML.Vec d); the paper allows any Hilbert space. - Domain.
Wclosed and convex with0 ∈ W(the paper's ; closedness makes well defined). - Strong convexity. Mathlib's
StrongConvexOn W lam Fwith modulus , equivalent to the paper's (1) when subgradients relative to exist at every point, which the oracle provides. - Oracle.
IsSubgradientOracleOn W F D g: for , is integrable and its mean is a subgradient of relative to . The oracle is jointly measurable and is a probability measure. The subgradient condition is not global: is only assumed convex on . - Second moment. at every (the paper's "regardless of how the iterates evolve"). Theorem 5's statement on p. 6 prints ; the standing assumption on p. 3 and the proof use , which is what is formalized.
- Indices.
sgdStrongIterates … tis the paper's ; the suffix average runs over indices , , passed as a natural number with , which encodes that and are integers. - Logarithm.
Real.log, the natural logarithm. - Expectations. Every Bochner integral in a conclusion is accompanied by an integrability conjunct, and nonnegative quantities in Lemma 1 are lower Lebesgue integrals, so no bound holds through Lean's default value for non-integrable functions. The bounds are stated for the expectation over ; a pathwise version, or one in which itself (rather than its mean) is a subgradient, would be a different and false or easier theorem.
Infrastructure a complete development needs: measurability of the projected iterates in the sample, the one-step conditional-expectation identity for the product law, and the integral comparison . The first two are reusable for the companion missions on the same paper and for any SGD analysis built on sgdStrongIterates. Proofs of any milestone, and of reusable lemmas about iidLaw and sgdStrongIterates, are welcome.
Selected references
- A. Rakhlin, O. Shamir, K. Sridharan, Making Gradient Descent Optimal for Strongly Convex Stochastic Optimization, ICML 2012; arXiv:1109.5647v7, 2012. https://arxiv.org/abs/1109.5647
- E. Hazan, A. Agarwal, S. Kale, Logarithmic regret algorithms for online convex optimization, Machine Learning 69 (2007) 169–192. https://doi.org/10.1007/s10994-007-5016-8
- E. Hazan, S. Kale, Beyond the regret minimization barrier: an optimal algorithm for stochastic strongly-convex optimization, COLT 2011. https://proceedings.mlr.press/v19/hazan11a.html
- A. Nemirovski, A. Juditsky, G. Lan, A. Shapiro, Robust stochastic approximation approach to stochastic programming, SIAM J. Optim. 19 (2009) 1574–1609. https://doi.org/10.1137/070704277
- S. Shalev-Shwartz, S. Ben-David, Understanding Machine Learning: From Theory to Algorithms, Cambridge University Press, 2014, Ch. 14. https://doi.org/10.1017/CBO9781107298019