A Single Timescale Stochastic Approximation Method for Nested Stochastic Optimization: NASA with Stepsizes 1/√N Drives the Expected Stationarity Measure E[V(x_R, z_R)] Below C/(√N − 1)Research Paper
Motivation
Many learning and risk-averse decision problems minimize a function of an expectation that is itself passed through another function: with . Examples are mean–deviation risk measures, stochastic variational inequalities rewritten as minimization, and policy evaluation in reinforcement learning. Only noisy samples of , of its Jacobian and of are available, and the plug-in estimate of is biased because is nonlinear. Classical stochastic gradient analysis does not apply directly.
Timeline of sample complexities for smooth nonconvex two-level problems, measured by the number of samples needed to reach an -stationary point:
- 2017 — Wang, Fang and Liu (Math. Program.) introduce stochastic compositional gradient descent with two time scales; the unconstrained nonconvex rate is in their measure.
- 2017 — Wang, Liu and Fang (JMLR) accelerate it to for smooth unconstrained problems, still with two time scales.
- 2019–2020 — Ghadimi, Ruszczyński and Wang (arXiv:1812.01094, SIAM J. Optim. 2020) propose NASA, a single-time-scale method with one sample per iteration, and prove for constrained problems. This mission formalizes that result.
Setting
Problem (1.1) is
where and are continuously differentiable, is closed and convex, and nothing is convex. Assumption 2 asks that , , , be Lipschitz with constants ; then is Lipschitz with . is the infimum of over .
For the subproblem solution is , with the Euclidean projection. A pair is judged by the optimality measure
which vanishes exactly when and lies in the normal cone to at .
The NASA method (Algorithm 1) keeps three sequences: iterates , gradient averages and value averages . With parameters , regularization and stepsizes :
Assumption 1 governs the oracle. Given , the outputs , and are conditionally unbiased for , and . Their second moments are bounded by , and , are conditionally independent. The analysis rests on the merit function , where is the optimal value of the subproblem.
Formalization targets
Goal: Theorem 2(c), (3.37)
With , for some , , for , , and uniform on :
The constants (3.32) and are explicit in the problem data. This is the paper's headline rate, and every other statement of the mission feeds it.
Milestones
- Lemma 1: .
- Lemma 2: is -Lipschitz.
- Lemma 3: has an -Lipschitz gradient in .
- (3.21): for the subproblem step .
- Lemma 4: the pathwise descent inequality .
- Proposition 1(a): moment bounds on and .
- Proposition 1(b): .
- (3.29): .
- Theorem 2(a): the weighted gradient-tracking error.
- Theorem 2(b): the bound on for general admissible stepsizes.
Significance
The result. Theorem 2(c) gives after iterations with one oracle call each. That matches the complexity of stochastic gradient descent on single-level smooth nonconvex problems, so the nested structure costs only constants. The bound holds with constraints, and comes with a certified estimate of . The same merit-function technique underlies later single-time-scale methods for multi-level and conditional stochastic optimization.
Formalizing it. The result is proved on paper, not formalized. A machine-checked proof requires the whole chain, Lemma 4 through Theorem 2, to be stated with constants that the argument actually supports. Several printed constants do not meet that bar (see Formalization scope), so the formalization also certifies a corrected version of the theorem. The deterministic milestones (Lemmas 1–3, (3.21), (3.29)) are general facts about Euclidean projections and value functions of strongly convex subproblems, and other projected-gradient developments can reuse them.
Difficulty
The obvious approach fails. Analyzing alone does not work because the step uses , a biased and correlated estimate of , and its error does not vanish at a single time scale. The analysis instead tracks the coupled potential . Each of its three terms needs a different tool: the descent lemma for , the smoothness of the value function (Lemma 3, which needs differentiability of a parametric minimum), and a recursion for . These pieces must combine under the single coupling condition (3.16). The probabilistic part is delicate as well. The cross terms vanish only in expectation, they require conditional independence of and , and their integrability is not automatic. Theorem 2(a) adds a weighted unrolling of the -recursion through the products and the summation condition (3.30).
Formalization scope
Points live in EuclideanSpace ℝ (Fin n) and EuclideanSpace ℝ (Fin m). Jacobians are continuous linear maps with the operator norm, and is the adjoint. The projection is a function with the published predicate SpectralProjGrad.Shared.IsProjOnto X P, and . is the subproblem objective evaluated at , and Lemma 3 places in the product WithLp 2 (ℝⁿ × ℝⁿ). is sInf over , and is assumed bounded below on . The regularization and the stepsizes are deterministic.
The randomness lives on a standard Borel probability space. Assumption 1 is stated with respect to any filtration to which are adapted, which includes the paper's natural filtration. It requires square-integrable oracle outputs, so that no conditional-expectation hypothesis is vacuous. Statements printed with conditional expectations on the left and constants on the right (Proposition 1, Theorem 2(a)) are stated in total expectation, which is what the proofs establish. Every conclusion about an expectation includes the integrability of the integrand, so a bound cannot hold because a non-integrable integral defaults to . is the -weighted average of over (3.34). Lemma 4 is deterministic: it holds along every path of the recursions, whatever the oracle outputs. The recursions tie to the projection and to the oracle outputs, so the goal cannot be satisfied by arbitrary sequences or by a vacuous oracle.
Corrections to printed constants, each stated in the item's Formalization Note:
- (3.18): the noise term of carries , not . The printed Lemma 4 has a counterexample.
- (3.26): uses , (the term) and .
- (3.33): the coefficient of is , not . The two agree when .
- (3.37): the denominator is , not , since the proof sets . The proof also fixes in and .
- Proposition 1(a): the bound is false conditionally on and is stated in expectation.
- Hypotheses the proofs use but the statements omit: , , and , (Theorem 2(a),(b)), , and .
The second claim (3.38) of Theorem 2(c) is excluded: its printed constant is smaller than what the proof gives. Theorem 3 (asymptotics) and the ASA variant of §4 are out of scope. Contributions are welcome at every milestone. Solvers will need Euclidean projection characterizations, differentiability of the squared distance to a convex set, conditional-independence product rules for conditional expectations, and Abel-type summation for the weights.
Selected references
- S. Ghadimi, A. Ruszczyński, M. Wang, A Single Time-Scale Stochastic Approximation Method for Nested Stochastic Optimization, arXiv:1812.01094v2, 2019; SIAM J. Optim. 30(1), 2020. https://arxiv.org/abs/1812.01094
- M. Wang, E. X. Fang, H. Liu, Stochastic compositional gradient descent: algorithms for minimizing compositions of expected-value functions, Math. Program. 161, 2017. https://doi.org/10.1007/s10107-016-1017-3
- M. Wang, J. Liu, E. X. Fang, Accelerating stochastic composition optimization, JMLR 18(105), 2017. https://jmlr.org/papers/v18/16-504.html
- S. Ghadimi, G. Lan, Stochastic first- and zeroth-order methods for nonconvex stochastic programming, SIAM J. Optim. 23(4), 2013. https://doi.org/10.1137/120880811
- A. Ruszczyński, Nonlinear Optimization, Princeton University Press, 2006.