cs.LGFeb 15, 2026

Constant-Stepsize Stochastic Approximation: Finite-Time Convergence, Gaussian Approximation, and Tail Bounds

Authors: Zedong WangYuyang WangIjay NarangFelix WangYuzhou WangSiva Theja Maguluri

Abstract

Constant-stepsize stochastic approximation (SA) is widely used in learning for computational efficiency, yet the distribution of the iterates is typically intractable. Classical asymptotics results give Xk(α)X(α)x+αYX_k^{(α)} \approx X^{(α)} \approx x^\star+\sqrtαY, where X(α)X^{(α)} is the steady state and YY is an appropriate Gaussian limit, by progressively taking the time kk\uparrow\infty and stepsize α0α\downarrow0. Such limit results, however, do not quantify finite-time, finite-stepsize errors. We develop an explicit pre-limit characterization for SA with i.i.d.\ and Markovian noise. We establish existence and uniqueness of the stationary law, a geometric Wasserstein convergence to stationarity, and almost-sure and L3L^3 convergence of the steady state to the root xx^\star, identifying the scale α\sqrtα as first-order fluctuation. At this scale, we derive a higher-order quantitative Gaussian approximation with a Wasserstein error, using Stein's method and Poisson equation techniques. We further obtain non-uniform Berry--Esseen-type tail bounds, incorporating both steady-state approximation and finite-time convergence errors. We instantiate the theory for strongly convex smooth SGD, linear SA, and nonlinear contractive SA. Beyond strong convexity, for general convex SGD, we identify a Gibbs limiting law and prove a pre-limit Wasserstein approximation error under stability and Stein-equation hypothesis, which are validated numerically.

Explore similar work

Sep 14, 2026stat.ML

Steady-State Convergence of Stochastic Approximation

For constant-stepsize stochastic approximation (SA), the iterates converge in distribution to a stationary law that depends on the stepsize α.α. Steady-state convergence (SSC) concerns the limit of the scaled stationary distribution as α0.α\downarrow 0. Existing SSC theory requires i.i.d. or additive noise and global differentiability of the mean operator, and yields suboptimal rates. We develop a unified SSC theory for constant-stepsize contractive SA driven by Markovian, multiplicative noise, covering both locally differentiable and locally nondifferentiable mean operators. A key methodological contribution is a multi-step universality framework that progressively reduces the original stochastic recursion to tractable auxiliary dynamics while preserving its steady-state limit. Under local quadratic linearization at the fixed point, we obtain a Gaussian approximation of the scaled steady state at the optimal rate O(α)O(\sqrtα) in Wasserstein-2 distance, which further gives finite-time Gaussian approximations for the raw iterates. In the locally nondifferentiable regime, we establish a general SSC result and show that the leading-order asymptotic bias can be of order α\sqrtα, in contrast to the αα-order bias in the smooth regime. We apply the theory to Markovian linear SA and asynchronous Q-learning, neither of which is covered by prior results. We further propose a bias-reduction scheme for Q-learning that requires no knowledge of the local smoothness regime, validated by numerical experiments.
Yixuan Zhang, Qiaomin Xie
May 20, 2026math.PR

Concentration of General Stochastic Approximation Under Heavy-Tailed Markovian Noise

We establish maximal concentration bounds for the iterates generated by stochastic approximation algorithms with general step sizes, where the noise has a finite-state Markovian component plus a Martingale-difference component. When the Martingale-difference noise is bounded, we show that the tail of the error can be sub-Gaussian, sub-Weibull, or something lighter than any Pareto but heavier than any Weibull, depending on the step size sequence and on whether the random operator is almost surely contractive, almost surely non-expansive, or expansive with positive probability. Our analysis relies on a novel Lyapunov function involving the moment-generating function of the solution to a Poisson equation, together with an auxiliary projected algorithm. We complement the upper bounds with worst-case examples showing that qualitatively sharper bounds are impossible. We further study the case of unbounded Martingale-difference noise when the average operator is contractive, and the step sizes are of order 1/k1/k. In this setting, we show that if the random operator is almost surely non-expansive, then the error tail is at most three times heavier than the noise tail, whereas if the random operator is expansive with positive probability, then the error may have substantially heavier tails. These results are obtained through a novel black-box truncation argument that reduces the unbounded-noise setting to the bounded-noise case.
Shubhada Agrawal, Siva Theja Maguluri, Martin Zubeldia
Jul 17, 2026cs.LG

Scaling Limits of Constant-Stepsize SGD at Flat Minima

For stochastic gradient descent (SGD) with a constant stepsize αα, the invariant law of the iterates, centered at a minimizer, describes the behavior of the algorithm over long time horizons. In the strongly convex case, this invariant law has the familiar α\sqrtα scaling and a Gaussian limit as α0α\downarrow 0. We show that this behavior changes fundamentally for convex objectives HH with flat minima and (sub)quadratic tails. More specifically, we study SGD with Markovian noise generated by a contractive driving chain. For every sufficiently small constant stepsize αα, we prove existence, uniqueness, and geometric convergence to an augmented invariant law in a Wasserstein distance induced by an αα-dependent metric. When the minimizer xx_\star has local flatness exponent m2m\ge2, meaning that 2H(x)xxm2Id\nabla^2 H(x)\asymp \lVert x-x_\star\rVert^{m-2} I_d as xxx\to x_\star, we obtain a contraction bound with factor 1cαm11-cα^{m-1}, where c>0c>0 is a constant. This recovers the factor 1cα1-cα in the quadratic case m=2m=2. We then analyze the small-stepsize scaling limit. We show that the invariant law concentrates on the scale α1/mα^{1/m} and that the rescaled iterates converge weakly to the stationary distribution of the stochastic differential equation dYt=h0(Yt)dt+Σ1/2dBt,dY_t=-h_0(Y_t)\,dt+Σ^{1/2}\,dB_t , where h0h_0 is the limiting drift at the minimizer and ΣΣ denotes the asymptotic covariance. This recovers the Gaussian limit when m=2m=2 and gives generally non-Gaussian stationary limits in the flat case m>2m>2. Finally, we give corresponding results for coordinate-separable objectives with unequal flatness exponents.
Jingyi Zhang, Cheng Mao, Debankur Mukherjee