English

Refining Covariance Matrix Estimation in Stochastic Gradient Descent Through Bias Reduction

Machine Learning 2026-04-24 v1 Machine Learning

Abstract

We study online inference and asymptotic covariance estimation for the stochastic gradient descent (SGD) algorithm. While classical methods (such as plug-in and batch-means estimators) are available, they either require inaccessible second-order (Hessian) information or suffer from slow convergence. To address these challenges, we propose a novel, fully online de-biased covariance estimator that eliminates the need for second-order derivatives while significantly improving estimation accuracy. Our method employs a bias-reduction technique to achieve a convergence rate of n(α1)/2lognn^{(\alpha-1)/2} \sqrt{\log n}, outperforming existing Hessian-free alternatives.

Keywords

Cite

@article{arxiv.2604.21203,
  title  = {Refining Covariance Matrix Estimation in Stochastic Gradient Descent Through Bias Reduction},
  author = {Ziyang Wei and Wanrong Zhu and Jingyang Lyu and Wei Biao Wu},
  journal= {arXiv preprint arXiv:2604.21203},
  year   = {2026}
}
R2 v1 2026-07-01T12:31:45.363Z