Online Covariance Estimation in Averaged SGD: Improved Batch-Mean Rates and Minimax Optimality via Trajectory Regression
Not on the map. Why?
Abstract (arXiv)
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), which yields O(n-1/8) at the optimal learning-rate exponent α 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), achieving O(n-1/6). The modified estimator requires no Hessian access and preserves O(d2) 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 Theta(n-(1-α)/2) for Hessian-free covariance estimation from the SGD trajectory: a Le Cam lower bound gives Omega(n-(1-α)/2), and a trajectory-regression estimator--which estimates the Hessian by regressing SGD increments on iterates--achieves 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.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 10 of its 23 references that are in the map. First posted on arXiv 12 Apr 2026.
Builds on
10 of 23 references.
References · 10
- 2015An Introduction to Matrix Concentration Inequalitiescited by 397
- 2016Statistical Inference for Model Parameters in Stochastic Gradient Descentcited by 71
- 2020Online Covariance Matrix Estimation in Stochastic Gradient Descentcited by 37
- 2021Fast and Robust Online Inference with Stochastic Gradient Descent via Random Scalingcited by 28
- 2024Gaussian Approximation and Multiplier Bootstrap for Polyak-Ruppert Averaged Linear Stochastic Approximation with Applications to TD Learningcited by 19
- 2018HiGrad: Uncertainty Quantification for Online Learning and Stochastic Approximationcited by 13
- 2023Online covariance estimation for stochastic gradient descent under Markovian samplingcited by 7
- 2022Covariance Estimators for the ROOT-SGD Algorithm in Online Learningcited by 5
- 2025Online Covariance Estimation in Nonsmooth Stochastic Approximationcited by 4
- 2023On the Utility of Equal Batch Sizes for Inference in Stochastic Gradient Descentcited by 1
Built on by
Not cited yet.