Online Covariance Matrix Estimation in Stochastic Gradient Descent

• 成果类型: Article
• 署名作者: Zhu, Wanrong; Chen, Xi; Wu, Wei Biao
• 署名单位: University of Chicago; New York University; University of Chicago
• 刊物名称: JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION
• ISSN/ISSBN: 0162-1459
• DOI: 10.1080/01621459.2021.1933498
• 发表日期: 2023
• 页码: 393-404
• 关键词: empirical likelihood; statistical-inference; OUTPUT ANALYSIS; variance; approximation

摘要

The stochastic gradient descent (SGD) algorithm is widely used for parameter estimation, especially for huge datasets and online learning. While this recursive algorithm is popular for computation and memory efficiency, quantifying variability and randomness of the solutions has been rarely studied. This article aims at conducting statistical inference of SGD-based estimates in an online setting. In particular, we propose a fully online estimator for the covariance matrix of averaged SGD (ASGD) iterates only using the iterates from SGD. We formally establish our online estimator's consistency and show that the convergence rate is comparable to offline counterparts. Based on the classic asymptotic normality results of ASGD, we construct asymptotically valid confidence intervals for model parameters. Upon receiving new observations, we can quickly update the covariance matrix estimate and the confidence intervals. This approach fits in an online setting and takes full advantage of SGD: efficiency in computation and memory.