Primal-dual subgradient methods for convex problems 2: Stochastic Simple Averages Reach E[φ(x̄ₖ)] − φ* ≤ β̂ₖ₊₁(γd(x*) + L²/(2σγ))/(k+1)Research Paper
Motivation
Many decision problems under uncertainty take the form of minimizing an expected cost over a convex feasible set, where is a random parameter whose distribution can be sampled but not integrated in closed form. Evaluating even approximately can cost exponentially many evaluations of in the dimension of , so the classical deterministic notion of an approximate solution is unusable. A practical alternative, going back to Nemirovski and Yudin, asks for a random output whose expected objective value is close to the optimum: . Stochastic subgradient methods achieve this; Nesterov's paper Primal-dual subgradient methods for convex problems (Math. Program. 120 (2009)) shows that the same guarantee holds for a stochastic version of his method of simple dual averages, a scheme that aggregates all past subgradients in the dual space instead of taking projected steps with decreasing step sizes.
Dual averaging has since become a standard tool in stochastic and online convex optimization (regularized dual averaging, AdaGrad's primal-dual variant, distributed dual averaging). Section 6 of Nesterov's paper is the original statement of its expected-accuracy guarantee.
Setting
Let be a finite-dimensional real vector space with an arbitrary norm and dual space with dual norm . Let be closed and convex, and let be a prox-function of : continuous on and strongly convex on with convexity parameter ,
with prox-center and . For and the point is well defined. The scalar sequence is given by and for .
The stochastic problem is given by a probability space and a cost with well defined on , and a minimizer of over with . Assumption 1: each is convex on . Assumption 2: a stochastic oracle returns a subgradient of at , and for all , .
The method of stochastic simple averages (SSA, (6.3)) fixes , starts from and , and for draws a fresh independent sample and sets
Its deterministic counterpart, the method of simple dual averages (SDA, (2.21)), is the same recursion with arbitrary answers in place of . The quality of the points is measured by the gap . Write and for the expectation over these independent samples.
Formalization targets
Goal: Theorem 7 (p. 27)
For every ,
Milestones
- Theorem 2 (p. 11): for SDA with any answers and , for , and, if some satisfies for all , .
- (6.5) (p. 26): along every sample path, .
- Display after (6.5) (p. 26): for , the gap of a run dominates , which dominates .
- (6.6) (p. 26): .
- p. 27, first display: for .
- p. 27, display before Theorem 7: , together with the averaged identity.
Significance
Theorem 7 gives an explicit, non-asymptotic bound on the expected optimality gap of the averaged SSA point. Since , the bound is , the optimal rate for nonsmooth stochastic convex optimization with bounded subgradients. Two features distinguish it from the classical stochastic subgradient bound: it holds in an arbitrary norm with a non-Euclidean prox-function (for example the entropy on the simplex with the norm), and the feasible set need not be bounded, since only enters. The constant is explicit and can be tuned to .
The result is proved in the paper; to our knowledge no machine-checked proof of Theorem 7 or of the dual averaging bounds of the paper exists. A formalization yields a reusable verified analysis of dual averaging in a general normed space, a template for expectation bounds over i.i.d. sample paths of an adaptive algorithm, and checked constants.
Difficulty
The pathwise part is a consequence of the deterministic Theorem 2, which itself rests on the general Dual Averaging bound (the companion mission's Theorem 1) and on the strong convexity estimates for . The probabilistic part looks routine on paper but is where formal arguments break: the iterate is a function of , and the identity requires measurability of the whole run and a Fubini argument on the product space. Measurability of the run in turn depends on continuity of the argmin map , and integrability of and of the gap is not assumed but must be derived from the uniform bound and the boundedness of the iterates. Pathwise bounds alone do not suffice: (6.5) controls the gap on each sample, not .
Formalization scope
Theorem numbers and page numbers refer to source.pdf, the author's revised CORE discussion-paper version (September 2005) of the Math. Program. article; printed pages equal PDF pages.
- is a finite-dimensional real normed space with an arbitrary norm; is
StrongDual ℝ Ewith the operator norm, and iss x. Strong convexity is Mathlib'sStrongConvexOn Q σ d, which is exactly (1.8); . - is a map
ℝ → StrongDual ℝ E → Ewith the hypothesis that lies in and minimizes for every ; no choice function is used. - is defined by recursion with . The gap is a real supremum over , used only for , where the set is nonempty and bounded.
- Indices start at 0; is the prox-center. The run is a function of a sample path ; is the integral against the product measure on (the law of i.i.d. copies of ), with the finite sample padded to a sequence that the iterates never read.
- and are functions on whose values off are irrelevant; the subgradient inequality is relative to . Integrability of for , and joint measurability of and , are explicit hypotheses. The paper uses measurability tacitly; without it a Lean integral of a non-measurable function is and the goal could become false.
- A statement for a fixed sample path, or with replaced by a pathwise quantity, is a different and weaker theorem and does not count. Integrability of the run is not assumed: it is part of what must be proved.
A complete development needs the deterministic dual averaging analysis (Lemma 1, Lemma 2, Theorem 1 of the paper, restated here only through Theorem 2), measurability and continuity of , and Fubini on finite products of probability spaces. The dual averaging lemmas and the product-measure lemmas for adaptive sample paths are reusable beyond this mission. Proofs of any milestone, and supporting lemmas on measurability of the SSA run, are welcome.
Selected references
- Yu. Nesterov, Primal-dual subgradient methods for convex problems, Mathematical Programming 120 (2009), 221–259. https://doi.org/10.1007/s10107-007-0149-x
- 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
- L. Xiao, Dual averaging methods for regularized stochastic learning and online optimization, J. Mach. Learn. Res. 11 (2010), 2543–2596. https://jmlr.org/papers/v11/xiao10a.html