Acceleration of Stochastic Approximation by Averaging: Almost-Sure Convergence and Asymptotic Normality of the Averaged IterateResearch Paper
Motivation
Stochastic approximation finds a root of an unknown map from noisy evaluations , by the Robbins–Monro recursion . It underlies stochastic gradient descent, recursive estimation in statistics, adaptive control and simulation-based optimization. The classical theory (Sacks 1958) shows that the fastest attainable rate, with and the noise covariance, is achieved by the matrix step , which requires knowing .
Polyak and Juditsky (SIAM J. Control Optim. 30 (1992) 838–855) proved that the same optimal covariance is attained without any knowledge of : run the recursion with scalar steps that decrease more slowly than and output the running average of the iterates. Ruppert (Cornell ORIE technical report, 1988) obtained the one-dimensional case independently. The method, known as Polyak–Ruppert averaging, is the standard device for variance reduction in stochastic approximation.
Timeline:
- 1951, Robbins and Monro: the recursion and its convergence in probability.
- 1958, Sacks: asymptotic normality of for .
- 1988, Ruppert: averaging in one dimension, i.i.d.-type noise.
- 1990–1992, Polyak; Polyak and Juditsky: averaging in for linear problems with martingale-difference noise (Theorem 1) and nonlinear problems (Theorem 2).
Setting
Let be a filtered probability space and an adapted -valued noise process. Given a nonrandom and step sizes , algorithm (7) is
The error is and the estimation error is .
The hypotheses are:
- Assumption 3.1: a Lyapunov function with , Lipschitz gradient, , for , and near .
- Assumption 3.2: near , with and every eigenvalue of having positive real part.
- Assumption 3.3: is a martingale difference with . It splits as , where is a martingale difference whose conditional covariance tends to in probability and whose conditional second moments are uniformly integrable, and with as .
- Assumption 3.4: , , and .
The linear case, algorithm (2), is with every eigenvalue of having positive real part.
Formalization targets
Goal: Theorem 2
Under Assumptions 3.1–3.4,
Milestones
- Lemma 1, Part 2: under condition (4) on the steps, .
- Lemma 1: the matrices are uniformly bounded, and .
- Lemma 2: the representation (A9) of for the linear error recursion.
- Theorem 1(a): the linear case, .
- Proof of Theorem 2, Part 1: converges almost surely to a finite limit.
- Proof of Theorem 2, p. 850: almost surely.
- Proof of Theorem 2, Part 4: the average of the linearised process satisfies almost surely.
Significance
Theorem 2 shows that averaging turns a robust, slowly-stepped recursion into an asymptotically efficient estimator. The covariance is the lower bound for this class of problems: for linear recursive estimates with independent noise it is the bound of [26] in the paper. Downstream, the result is what is invoked for the asymptotic efficiency of averaged stochastic gradient descent (Theorem 3 of the paper) and of recursive M-estimators in regression (Theorem 4).
The result is proved, with a published proof, but has no machine-checked version. As far as a search of the platform shows, no statement of Theorem 1 or Theorem 2 exists on Prove2Me. The platform does have a scalar martingale central limit theorem (Martingale.clt_of_mds, proved, with unconditional Lindeberg condition), which is usable through the Cramér–Wold device. Formalizing Theorem 2 also requires the Robbins–Siegmund almost-supermartingale theorem, a multivariate CLT for martingale differences under conditional Lindeberg and conditional covariance conditions, and the Kronecker lemma. Mathlib has none of these three in the required form, and each is reusable well beyond this mission. Non-asymptotic SGD rates already on the platform (the Bottou–Curtis–Nocedal and Lan missions) are different results.
Difficulty
The obvious approach analyses directly. It fails: with steps decreasing more slowly than , diverges, and only the average has the rate. The average must be compared with the averaged noise through the matrix sums of Lemma 1, whose bounds are uniform in both indices. Those bounds rely on the step condition in a quantitative way.
The nonlinear case adds a second difficulty. The iterates are first shown to converge almost surely, by a Lyapunov argument. The nonlinear error is then transferred to a linearised process at the scale, which needs a summability estimate on obtained through stopping times. A central limit theorem for the linear process alone does not give the result, because the linearisation error must vanish after multiplication by .
Formalization scope
Points are in EuclideanSpace ℝ (Fin N) and matrices are Matrix (Fin N) (Fin N) ℝ, acting through Matrix.toEuclideanLin. Matrix norms are operator norms. Conditional expectations are MeasureTheory.condExp on a Filtration ℕ. "Given " is written with shifted indices ( given ). The algorithm is a recursive definition from , with unused and averaging . Convergence in distribution is TendstoInDistribution to multivariateGaussian 0 V. Convergence of conditional covariances in probability is entrywise TendstoInMeasure. A limsup or supremum "tending to 0 in probability" is unfolded into its – definition.
Corrections of the printed text, each used by the paper's own proof:
- Assumption 3.1 prints and . Stated as and (as printed they force ). The drift constant is renamed , since the paper uses also in Assumption 3.2.
- Eq. (10) is garbled as printed. It is stated as , the form of Assumptions 4.7 and 5.6 and of p. 851.
- Assumption 3.3's is stated as .
- and are added to Assumption 3.4. The proof uses them (p. 849), and they do not follow from it.
- is assumed continuous. The paper states no regularity of , but its proof of almost sure convergence (pp. 849–850) needs bounded away from on annuli around , which continuity and Assumption 3.1 provide.
- Lemma 1 and Theorem 1(a) are stated under condition (4) only. The constant-step condition (3) is false as printed (, ), and Theorem 2 does not use it.
- (A3) is stated with the norm inside, as its proof establishes.
- (A9) and the linearised process of Part 4 are stated with noise signs, and with . The printed signs contradict (A8) at .
Several formalizations would make the goal trivial, and all are ruled out:
- conditional expectations of non-integrable functions, which are in Lean (every noise process is required to be in );
- a real supremum for the uniform integrability in Assumption 3.3, which is on unbounded families;
- an arbitrary process with a property in place of the recursion (7);
- a degenerate Dirac target (the covariance is positive definite under the hypotheses).
Welcome contributions: the Robbins–Siegmund theorem, a vector martingale CLT under conditional Lindeberg conditions, the Kronecker lemma, and the matrix estimates of Lemma 1.
Selected references
- B. T. Polyak, A. B. Juditsky, Acceleration of stochastic approximation by averaging, SIAM J. Control Optim. 30(4), 838–855, 1992. https://doi.org/10.1137/0330046
- H. Robbins, S. Monro, A stochastic approximation method, Ann. Math. Statist. 22, 400–407, 1951. https://doi.org/10.1214/aoms/1177729586
- J. Sacks, Asymptotic distribution of stochastic approximation procedures, Ann. Math. Statist. 29, 373–405, 1958. https://doi.org/10.1214/aoms/1177706619
- D. Ruppert, Efficient estimations from a slowly convergent Robbins–Monro process, Cornell University ORIE Technical Report 781, 1988 (no stable online link located).
- H. Robbins, D. Siegmund, A convergence theorem for non negative almost supermartingales and some applications, in Optimizing Methods in Statistics, Academic Press, 233–257, 1971. https://doi.org/10.1016/B978-0-12-604550-5.50015-8