Online inference with debiased stochastic gradient descent

被引:6
|
作者
Han, Ruijian [1 ]
Luo, Lan [2 ]
Lin, Yuanyuan [3 ]
Huang, Jian [1 ]
机构
[1] Hong Kong Polytech Univ, Dept Appl Math, Kowloon, Hung Hom, 11 Yuk Choi Rd, Hong Kong 999077, Peoples R China
[2] Rutgers Sch Publ Hlth, Dept Biostat & Epidemiol, 683 Hoes Lane West, Piscataway, NJ 08854 USA
[3] Chinese Univ Hong Kong, Dept Stat, Shatin, Cent Ave, Hong Kong 999077, Peoples R China
基金
中国国家自然科学基金;
关键词
Confidence interval; High-dimensional statistics; Online learning; Stochastic gradient descent; CONFIDENCE-INTERVALS; APPROXIMATION; PARAMETERS;
D O I
10.1093/biomet/asad046
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
We propose a debiased stochastic gradient descent algorithm for online statistical inference with high-dimensional data. Our approach combines the debiasing technique developed in high-dimensional statistics with the stochastic gradient descent algorithm. It can be used to construct confidence intervals efficiently in an online fashion. Our proposed algorithm has several appealing aspects: as a one-pass algorithm, it reduces the time complexity; in addition, each update step requires only the current data together with the previous estimate, which reduces the space complexity. We establish the asymptotic normality of the proposed estimator under mild conditions on the sparsity level of the parameter and the data distribution. Numerical experiments demonstrate that the proposed debiased stochastic gradient descent algorithm attains nominal coverage probability. Furthermore, we illustrate our method with analysis of a high-dimensional text dataset.
引用
收藏
页码:93 / 108
页数:16
相关论文
共 50 条
  • [1] Fast and Robust Online Inference with Stochastic Gradient Descent via Random Scaling
    Lee, Sokbae
    Liao, Yuan
    Seo, Myung Hwan
    Shin, Youngki
    THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 7381 - 7389
  • [3] Stochastic Gradient Descent as Approximate Bayesian Inference
    Mandt, Stephan
    Hoffman, Matthew D.
    Blei, David M.
    JOURNAL OF MACHINE LEARNING RESEARCH, 2017, 18
  • [4] STATISTICAL INFERENCE FOR MODEL PARAMETERS IN STOCHASTIC GRADIENT DESCENT
    Chen, Xi
    Lee, Jason D.
    Tong, Xin T.
    Zhang, Yichen
    ANNALS OF STATISTICS, 2020, 48 (01): : 251 - 273
  • [5] Online stochastic gradient descent on non-convex losses from high-dimensional inference
    Ben Arous, Gerard
    Gheissari, Reza
    Jagannath, Aukosh
    JOURNAL OF MACHINE LEARNING RESEARCH, 2021, 22
  • [6] Online Covariance Matrix Estimation in Stochastic Gradient Descent
    Zhu, Wanrong
    Chen, Xi
    Wu, Wei Biao
    JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 2023, 118 (541) : 393 - 404
  • [7] Online stochastic gradient descent on non-convex losses from high-dimensional inference
    Arous, Gerard Ben
    Gheissari, Reza
    Jagannath, Aukosh
    Journal of Machine Learning Research, 2021, 22
  • [8] Statistical Inference for Online Decision Making via Stochastic Gradient Descent (Nov, 10.1080/01621459.2020.1826325, 2020)
    Chen, Haoyu
    Lu, Wenbin
    Song, Rui
    JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 2021,
  • [9] Scalable statistical inference for averaged implicit stochastic gradient descent
    Fang, Yixin
    SCANDINAVIAN JOURNAL OF STATISTICS, 2019, 46 (04) : 987 - 1002
  • [10] Statistical Inference for Online Decision Making via Stochastic Gradient Descent (vol 116, pg 708, 2021)
    Chen, Haoyu
    Lu, Wenbin
    Song, Rui
    JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 2021, 116 (534) : 1039 - 1039