Stochastic First- and Zeroth-Order Methods for Nonconvex Stochastic Programming 2: The Randomized Stochastic Gradient-Free Method with Gaussian Smoothing Bounds (1/L)E‖∇f(x_R)‖² for Nonconvex fResearch Paper
Motivation
Many optimization problems in simulation-based design, reinforcement learning and machine learning can be evaluated but not differentiated. Their objective is an expectation that can only be sampled through noisy function values . Gradients may be unavailable because the model is a black box, a simulator, or a legacy code. Methods for this setting are called zeroth-order or derivative-free stochastic methods. The objective is often nonconvex, so the realistic target is an approximately stationary point, , not a global minimizer.
Ghadimi and Lan (arXiv:1309.5549; SIAM J. Optim. 23(4), 2013) gave the first complexity bounds for stochastic zeroth-order methods on nonconvex problems. Their randomized stochastic gradient free (RSGF) method combines two ingredients. The first is the Gaussian smoothing and random finite-difference estimator of Nesterov and Spokoiny (Found. Comput. Math. 17, 2017; preprint 2011). The second is a randomized stopping rule that outputs an iterate chosen at random along the trajectory. The resulting bound has become the standard reference point for derivative-free nonconvex stochastic optimization.
Timeline:
- 1952: Kiefer and Wolfowitz introduce finite-difference stochastic approximation.
- 2011: Nesterov and Spokoiny analyze Gaussian random gradient-free oracles for convex and deterministic nonconvex problems.
- 2013: Ghadimi and Lan prove the nonconvex stochastic rate of the RSGF method (Theorem 3.2) and its complexity (Corollary 3.3).
Setting
Let be Euclidean space and let be a random vector with distribution on a measurable space . The problem is
The sample functions lie in almost surely: they are differentiable, and their gradients are -Lipschitz. Two assumptions describe the noise. Assumption A1 concerns the first-order oracle : it is unbiased, , and has variance at most . Assumption A3 says the zeroth-order oracle is unbiased, .
For a smoothing parameter and an -dimensional standard Gaussian vector , the Gaussian smoothing of is (3.3). The RSGF method starts from and uses stepsizes . At step it draws a fresh sample and a fresh Gaussian direction , forms the zeroth-order estimator
and sets (3.13). The output is , where the random iteration count has the probability mass function
In the Lean development, is objective P F, is szoGrad (the published RandomGradFree.Shared.oracle applied to ), and is the published RandomGradFree.Shared.smoothing. The smoothing parameter is μs, because μ names the probability measure.
Formalization targets
Goal: Theorem 3.2 a)
Let for , with , and let . Then
The stepsizes are left free, so every stepsize policy specializes the goal.
Companions
- Theorem 3.2 b), (3.17): for convex with a minimizer , the bound on .
- Corollary 3.3, (3.26): for the constant stepsizes (3.24) and ,
Milestones
The milestones follow the paper's proof.
- Properties of the smoothing. These are Theorem 3.1 a)–c): has an -Lipschitz gradient given by (3.4), (3.5), the gradient bound (3.6), and the second-moment bound (3.7). Three of these are already on the platform and are referenced. Two consequences follow, (3.9) and (3.11).
- Unbiasedness, (3.14): the estimator is unbiased for .
- The proof chain of Theorem 3.2. The pathwise descent inequality (3.19), the martingale identity (3.20) and the conditional second-moment bound (3.21) combine into the weighted bound (3.23). The -weighting identity converts (3.23) into (3.16).
Significance
The theorem shows that function values alone suffice to find approximately stationary points of nonconvex stochastic programs. The price is a factor of the dimension : with constant stepsizes the RSGF method needs iterations, against for the first-order RSG method. The randomized output rule and the analysis through are reused in later zeroth-order methods, including two-phase variants, mirror-descent variants and variance-reduced schemes.
The result is proved in the paper. It is not formalized anywhere known to this mission. The platform already holds parts of the Gaussian-smoothing theory from the Nesterov–Spokoiny series, but only for convex problems with exact function values. This mission adds:
- the nonconvex form of the second-moment bound (3.7);
- the gradient-approximation bound (3.6);
- the stochastic sampling model with conditional expectations;
- the first machine-checkable statement of a nonconvex stochastic zeroth-order rate.
Difficulty
The first idea is to treat the RSGF method as the first-order RSG method run on with the unbiased estimator , and to quote the RSG bound. This fails, because the variance has no uniform bound: by (3.7) it grows with at the random iterate. The analysis has to carry that term along and absorb it into the left side, which is where the weights come from. It must also move between and through (3.6) at a cost polynomial in .
On the formal side, three obstacles stand out.
- The Gaussian moment estimates behind (3.6) and (3.7) need bounds on up to for a standard Gaussian vector.
- Conditional expectations with respect to are taken at random iterates that depend on the past, so independence of the fresh samples must be used carefully.
- Integrability must be established for every quantity before expectations can be manipulated.
Formalization scope
- Space. is
EuclideanSpace ℝ (Fin n). The constants , , use the same natural number . The Gaussian is Mathlib'sstdGaussian. - Iterates. They are indexed from , and takes values in .
- Optimal values. is the infimum of (
IsGLB), with bounded below. The paper's (3.10) is read as an infimum, since the minimum need not be attained. - Smoothness. is required for -almost every , as on the page.
- Assumption A1 is stated at every point for .
- Assumption A3 is the definition of , with integrable.
The following are explicit additions; the paper leaves them implicit.
- Sampling. The samples and are measurable and mutually independent, and is independent of all of them.
- Measurability and integrability. is jointly measurable, and is integrable.
- Stepsizes are positive.
- Corollary 3.3 assumes .
Every conclusion about an expectation also asserts the integrability of its integrand. A non-integrable function has Lean integral , so a bound without this conjunct could hold for the wrong reason.
The goal is stated about the objects of the method only: , , , the run (3.12)–(3.13), and the constants. It does not mention , or the filtration, so it cannot be discharged by choosing an auxiliary object.
A complete development needs:
- Gaussian moment bounds and differentiation under the Gaussian integral;
- conditional expectation with respect to the natural filtration of an independent sequence;
- the descent lemma for functions.
The Gaussian-smoothing lemmas (3.6), (3.7) and (3.9) are reusable by any zeroth-order mission. Contributions of proofs of individual milestones are welcome, as are alternative proofs of (3.7) without Nesterov's bound.
Selected references
- S. Ghadimi and G. Lan, Stochastic first- and zeroth-order methods for nonconvex stochastic programming, SIAM J. Optim. 23(4):2341–2368, 2013. Preprint arXiv:1309.5549v1, the version cited here. https://arxiv.org/abs/1309.5549
- Y. Nesterov and V. Spokoiny, Random gradient-free minimization of convex functions, Found. Comput. Math. 17:527–566, 2017. https://doi.org/10.1007/s10208-015-9296-2
- J. Kiefer and J. Wolfowitz, Stochastic estimation of the maximum of a regression function, Ann. Math. Statist. 23(3):462–466, 1952. https://doi.org/10.1214/aoms/1177729392