stat.MLOct 31, 2026

Lower Bounds on Inference Complexity for Polyak-Ruppert Averaged Stochastic Gradient Descent

Authors: Denis Belomestny, Alexey Naumov, Sergey Samsonov

Organizations: University of Duisburg-Essen, Germany · HSE University, Moscow, Russia

Abstract

In this paper, we establish sharp lower bounds for Gaussian approximation of Polyak-Ruppert averaged stochastic gradient descent with polynomially decreasing step sizes. For every fixed exponent γ∈(1/2,1)γ\in(1/2,1), a single one-dimensional smooth, strongly convex example has approximation error with the limiting variance of order n1/2−γ+nγ−1n^{1/2-γ}+n^{γ-1} in Kolmogorov and convex distances. This matches the existing upper bound throughout the range of γγ. The nonlinear bias contributes n1/2−γn^{1/2-γ} even when the Gaussian reference uses the finite linearized variance. We also derive a lower bound of order n−(1−γ)/2n^{-(1-γ)/2} for the relative mean error of estimating the limiting variance from one SGD trajectory, uniformly over a scalar quadratic family. With the limiting variance known, achieving approximation error ε\varepsilon on this example takes order ε−4\varepsilon^{-4} iterations for the best choice γ=3/4γ=3/4. Achieving both Gaussian approximation and relative mean variance-estimation errors at most ε\varepsilon uniformly over the model class requires at least order ε−6\varepsilon^{-6} iterations for every fixed γγ.

Explore similar work

Sep 30, 2026

Sharp Stationary Gaussian Approximation for Constant-Stepsize SGD

We prove a sharp Gaussian approximation for the invariant law of constant-stepsize SGD with bounded additive noise generated by an exogenous uniformly ergodic Markov chain. For a smooth, strongly convex objective with a Lipschitz Hessian and nondegenerate long-run noise covariance, the centered iterate normalized by the square root of the stepsize is O(α)O(\sqrtα)-close in 1-Wasserstein distance to its limiting Gaussian. The proof combines blockwise Gaussian comparison with long-run contraction. A four-state example gives a matching lower bound although the one-time noise marginal is symmetric and every nonzero-lag autocovariance vanishes. In this example, an adjacent third-order mixed moment produces the leading correction.
Apr 12, 2026

Online Covariance Estimation in Averaged SGD: Improved Batch-Mean Rates and Minimax Optimality via Trajectory Regression

We study online covariance matrix estimation for Polyak--Ruppert averaged stochastic gradient descent (SGD). The online batch-means estimator of Zhu, Chen and Wu (2023) achieves an operator-norm convergence rate of O(n−(1−α)/4)O(n^{-(1-α)/4}), which yields O(n−1/8)O(n^{-1/8}) at the optimal learning-rate exponent α→1/2+α\rightarrow 1/2^+. A rigorous per-block bias analysis reveals that re-tuning the block-growth parameter improves the batch-means rate to O(n−(1−α)/3)O(n^{-(1-α)/3}), achieving O(n−1/6)O(n^{-1/6}). The modified estimator requires no Hessian access and preserves O(d2)O(d^2) memory. We provide a complete error decomposition into variance, stationarity bias, and nonlinearity bias components. A weighted-averaging variant that avoids hard truncation is also discussed. We establish the minimax rate Θ(n−(1−α)/2)Θ(n^{-(1-α)/2}) for Hessian-free covariance estimation from the SGD trajectory: a Le Cam lower bound gives Ω(n−(1−α)/2)Ω(n^{-(1-α)/2}), and a trajectory-regression estimator--which estimates the Hessian by regressing SGD increments on iterates--achieves O(n−(1−α)/2)O(n^{-(1-α)/2}), matching the lower bound. The construction reveals that the bottleneck is the sublinear accumulation of information about the Hessian from the SGD drift.
Oct 20, 2026

When Momentum Helps Polyak--Ruppert Averaging

Polyak--Ruppert averaging gives SGD and momentum methods the same asymptotic covariance. We characterize how fast finite runs reach it and how far they can be parallelized. For strongly convex quadratics with additive noise, we prove an online--offline bridge: the empirical risk minimizer on the samples of a tail average is one Newton step from the tail-averaged query point of any online procedure. For SGD, heavy ball, and Nesterov's method with constant momentum, this step is a boundary term at the two ends of the tail and splits the gap to the offline estimator into center, covariance, and shape errors. Necessary and sufficient conditions show that, on the least curved direction, the center error sets the pace: SGD needs m≫κ2m\ggκ^2 rounds but momentum only m≫κm\ggκ, where κκ is the condition number, so at sample size NN the critical batch size grows from Θ(N/κ)Θ(\sqrt{N}/κ) to Θ(N/κ)Θ(\sqrt{N/κ}). The covariance error separates the momentum methods: heavy ball has relative variance excess Θ(κ3/2/m)Θ(κ^{3/2}/m) along the stiffest direction, whereas Nesterov has relative error O(κ/m)O(\sqrtκ/m) in every direction. The shape error is method-independent by a Berry--Esseen bound uniform in κκ and batch size, so confidence intervals are valid when the other two errors vanish. We further examine how the conclusions change under heavy-tailed or state-dependent noise and clipping, and establish a first-order comparison with ERM for nonlinear Nesterov averaging under explicit conditions.