cs.LGApr 12, 2026

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

Authors: Yijin Ni, Xiaoming Huo

Organizations: H. Milton Stewart School of Industrial and Systems Engineering, Georgia Institute of Technology

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. Refining Covariance Matrix Estimation in Stochastic Gradient Descent Through Bias Reduction

    Apr 23, 2026Ziyang Wei, Wanrong Zhu, Jingyang Lyu +1Covariance EstimationStochastic Gradient Descent

  2. Mini-Batch Covariance, Diffusion Limits, and Oracle Complexity in Stochastic Gradient Descent: A Sampling-Design Perspective

    Mar 2, 2026Daniel Zantedeschi, Kumar MuthuramanStochastic Gradient DescentStochastic Optimization Convergence