Non-Strongly-Convex Smooth Stochastic Approximation with Convergence Rate O(1/n): Averaged Constant-Step-Size LMS Has Expected Excess Risk at Most (1/2n)[σ√d/(1−√(γR²)) + R‖θ₀−θ*‖/√(γR²)]²Research Paper
Motivation
Least-squares regression fitted by stochastic gradient descent — the least-mean-square (LMS) algorithm — is the basic large-scale learning procedure: each observation is touched once, at a cost linear in the dimension. Classical analyses of stochastic approximation give the rate for non-strongly-convex objectives, and when the objective is -strongly convex. For least squares, is the smallest eigenvalue of the input covariance, which in high-dimensional problems is close to zero, so the strongly convex rate is often worse than the non-strongly-convex one.
F. Bach and E. Moulines (arXiv:1306.2119, NeurIPS 2013) showed that for the square loss this dichotomy disappears: averaged LMS with a constant step size reaches the rate with no strong-convexity assumption, and with a constant that does not involve the smallest eigenvalue. Averaging of stochastic approximation iterates goes back to Polyak and Juditsky (SIAM J. Control Optim. 1992), whose guarantees are asymptotic and use decreasing step sizes. The proof technique for the expansion of the noise process is adapted from Aguech, Moulines and Priouret (SIAM J. Control Optim. 2000). This mission formalizes the non-asymptotic bound in expectation (Theorem 1 of the paper) and the chain of lemmas of its Appendix A.
Setting
Let with , inner product and norm . For , is the operator . For self-adjoint operators, means that is positive semi-definite.
The data are independent and identically distributed pairs , , with finite second moments. The covariance operator is , assumed invertible (its eigenvalues may be arbitrarily small). The least-squares objective
attains its global minimum at , and is the residual. The model need not be well specified: need not vanish. Two constants satisfy
These are assumptions (A1)–(A6) of §2.1. The LMS recursion with constant step size , started at , is
and its average is .
Formalization targets
Goal: Theorem 1, Eq. (2)
For every step size and every ,
The constants are the paper's. A companion item states the case , where the bound reads .
Milestones (Appendix A)
- The excess risk is a quadratic form: .
- Consequences of (A6): , , , and for .
- Lemma 1: for a recursion with martingale-difference noise and ,
- Lemma 3: for and .
- Lemma 2: for with , the second-moment bound (13) and
- The pathwise decomposition (A.2).
- The initial-condition bound for the noise-free process (A.3).
- The expansion of the noise process (A.4): the remainder recursion (16), the covariance bound (17) , the order- bound , the remainder bound , and the noise bound
Significance
The result. Theorem 1 gives a finite-sample, dimension-explicit bound with two terms: a variance term , which matches the minimax rate for least-squares regression, and a bias term . Neither involves the smallest eigenvalue of , so the guarantee survives ill-conditioning, which is the regime of high-dimensional learning. The bound is the basis for the paper's later results: the high-probability bound (Theorem 2) and the constant-step algorithm for logistic regression (Theorem 3), whose analysis invokes Theorem 1 for the quadratic approximations.
Formalizing it. The result is proved in the paper; nothing here is open. To our knowledge no part of it has been machine-checked. A formalization yields a reusable layer for linear stochastic approximation in finite dimension: martingale-difference noise in , second-moment bounds for linear recursions driven by random operators, Loewner-order arguments, and averaging. The lemmas are stated for an abstract filtration and an abstract operator , so they apply beyond this model. The mission also records the corrections the appendix needs (an "" that should be "" in (13), an index in (16), and the exponent of in A.5).
Difficulty
Two steps resist the naive approach. First, the obvious one-step analysis — expand and take expectations — yields the bias part and Lemma 1, but on the noise it gives only , which does not decrease with . The rate requires averaging to cancel the noise, and this cancellation is visible only for the recursion with replaced by its mean . The random recursion is therefore expanded in powers of around the mean recursion, and each term needs its own covariance bound, by induction on , using the independence of from . Second, the induction relies on Loewner-order bookkeeping: sums of must be bounded uniformly in without dividing by small eigenvalues.
Formalization scope
The space is EuclideanSpace ℝ (Fin d). Operators are continuous linear maps, and is an explicit two-sided inverse. Observations are indexed from . Averages are , with in every statement that uses them. The covariance operator is defined by its bilinear form, . Every Loewner inequality whose sides are expectations is an inequality of quadratic forms (for example for all ), which is the same order for self-adjoint operators.
Lean's Bochner integral is on non-integrable functions, so every moment assumption carries the integrability of its integrand, and every bounded expectation in a conclusion is paired with an integrability conjunct. Without these, a heavy-tailed would satisfy (A6) vacuously and a conclusion could hold through the value ; neither formalization is acceptable. Independence is of the pairs , not of and separately. (A4) is attainment of the minimum, not a gradient condition.
The following hypotheses are added to the page and disclosed in each item:
- (a step size, and is a denominator);
- ;
- the positivity and self-adjointness of in Lemma 2;
- instead of in the remainder bound, which divides by .
A complete development needs:
- conditional expectations of -valued martingale differences, and the orthogonality of their sums;
- independence of a fresh observation from the past iterates;
- spectral calculus for ;
- Minkowski's inequality in .
All of these are reusable for other stochastic-approximation missions. Proofs of any milestone are welcome, as are alternative arguments for the noise bound that avoid the expansion.
Selected references
- F. Bach and E. Moulines, Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n), Advances in Neural Information Processing Systems 26, 2013. https://arxiv.org/abs/1306.2119
- B. T. Polyak and A. B. Juditsky, Acceleration of stochastic approximation by averaging, SIAM Journal on Control and Optimization 30(4), 1992. https://doi.org/10.1137/0330046
- R. Aguech, E. Moulines and P. Priouret, On a perturbation approach for the analysis of stochastic tracking algorithms, SIAM Journal on Control and Optimization 39(3), 2000. https://doi.org/10.1137/S0363012997331639
- F. Bach and E. Moulines, Non-asymptotic analysis of stochastic approximation algorithms for machine learning, Advances in Neural Information Processing Systems 24, 2011. https://hal.science/hal-00608041