In this paper, we study the bias and high-order error bounds of the Linear Stochastic Approximation (LSA) algorithm with Polyak-Ruppert (PR) averaging under Markovian noise. We focus on the version of the algorithm with constant step size α and propose a novel decomposition of the bias via a linearization technique. We analyze the structure of the bias and show that the leading-order term is linear in α and cannot be eliminated by PR averaging. To address this, we apply the Richardson-Romberg (RR) extrapolation procedure, which effectively cancels the leading bias term. We derive high-order moment bounds for the RR iterates and show that the leading error term aligns with the asymptotically optimal covariance matrix of the vanilla averaged LSA iterates.
Figures & tables
Figure 1: In subfigures (a), (b) and (c), we consider the step size α=n−β . Subfigure (a): error for RR iterations ( 48 ). Subfigure (b): error for PR iterations ( 2 ). Subfigure (c): error for RR, multiplied by a factor n2−β , which corresponds to the leading term of ( 59 ) after substituting α . Subfigure (d): MSE of Polyak-Ruppert and Richardson-Romberg iterations for different step sizes α .
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Variable
Description
Reference
Q
Solution of Lyapunov equation for Aˉ
Proposition 10
κQ
λmin−1(Q)λmax(Q)
Proposition 10
a
Real part of minimum eigenvalue of Aˉ
Proposition 10
Γm:n(α)
Product of random matrices with step size α
( 23 )
ε(Zn)
Noise in LSA procedure
( 9 )
θ~n(tr),θ~n(fl)
Transient and fluctuation terms of LSA error
( 25 )
Appendix
Table 1: Constants, definitions, notations
Figure 2: Comparison of remainder errors for Polyak-Ruppert averaged and Richardson-Romberg iterates in two regimes. In the first regime (a), the error attains its optimum for PR averaging, whereas for RR iterates it decays as predicted by Theorem 2 . Conversely, in the second regime (b), the optimum is achieved for RR iterates, which is also consistent with the theory.
Figure 3: Subfigure (a): the MSE remainder term for the Richardson-Romberg iterates is well approximated by rc−1 for some constant r>0 , matching the leading term (αn)−1n−1 in ( 59 ). Subfigure (b): rescaled by n4/3 MSE remainder trajectories for varying step sizes α . These plots cease decaying and stabilize, confirming the predicted order of the remainder term.
Constant-step stochastic approximation generally has a nonzero stationary mean error that persists under time averaging. This paper studies that error for nonlinear two-timescale recursions driven by an exogenous finite-state Markov chain. Under stated smoothness assumptions and conditions on the stationary distribution, we derive a first-order bias expansion whose error bound remains uniform as the slow step size becomes much smaller than the fast step size. Fast-manifold coordinates keep the associated covariance equation regular in this limit. For fast step η and slow step ε, the expansion reveals a mixed contribution ε2/η alongside terms linear in each step size. This dependence matters for bias reduction: along power-law step-size paths, the bias exponents need not be integers, so Richardson--Romberg extrapolation requires weights matched to the path. An exactly solvable nonlinear Markov example verifies the coefficients. We verify localization for temporal-difference learning and compare finite-run extrapolation at equal update budgets. For finite runs, we bound the initialization error of tail averages on both timescales under an additional coupling assumption. In the special case of additive independent noise, signed third-moment cancellation yields a sharper remainder.
A. Ch. Madhusudanarao, Rahul Singh
Department of Computer Science and Automation Indian Institute of Science, Bengaluru, India · Laboratoire de Recherche de l’EPITA, Paris, France
In this paper, we establish Berry-Esseen-type bounds for federated linear stochastic approximation (LSA). Our results provide the first federated Gaussian approximations for LSA that explicitly capture communication-computation trade-offs and heterogeneity-aware error terms, quantifying the effects of local step size, number of local updates, and heterogeneity on convergence rates. We present results for both (i) constant step size regime and (ii) decreasing step size with an increasing number of local iterations, recovering the recent rates of Bonnerjee et al. [2025] as a special case. As a primary application of our results, we develop an online multiplier bootstrap procedure for inference on the last iterate, which avoids explicit estimation of the asymptotic covariance matrix, and obtain non-asymptotic validity guarantees for this procedure.
Ilya Levin, Maksim Shuklin, Eric Moulines +2
1HSE University, Russia · 2MBZUAI · 3CMAP, CNRS, École Polytechnique, Institut Polytechnique de Paris, 91120 Palaiseau, France
We establish near-linear accuracy bounds for the classical Moreau--Yosida unadjusted Langevin algorithm (MYULA). The target is π∝e−f−g, where f∈C2(Rd) is m-strongly convex with Lipschitz gradient and g is convex and globally Lipschitz. Under an explicit parameter-dependent step-size condition, we bound the invariant-measure bias relative to the Moreau-smoothed target by O(h), with only logarithmic dependence on the inverse smoothing parameter in the error coefficient. Combining this estimate with the Moreau approximation bias and Wasserstein contraction gives O(ε−1) iterations to make the Nth-iterate law μN satisfy mW2(μN,π)≤ε, for fixed model parameters and initialization. We bound the stationary error directly, without assuming third derivatives or a Lipschitz Hessian. Each iteration uses one gradient evaluation and one exact proximal evaluation. The key idea in our analysis is to convert a second-order stationary residual into a Wasserstein bound using a Poisson-based estimate.
Yuchen Xin, Zhihua Zhang
School of Mathematical Sciences, Peking University