mlsanityData to 7 Oct 2026

Unbiased least squares regression via averaged stochastic gradient descent

arXiv 2406.18623 ↗First posted 26 Jun 2024stat.ML cs.LG stat.MEcited by 1

Not on the map. Why?

Abstract (arXiv)

We consider an on-line least squares regression problem with optimal solution θ^* and Hessian matrix H, and study a time-average stochastic gradient descent estimator of θ^*. For k≥2, we provide an unbiased estimator of θ^* that is a modification of the time-average estimator, runs with an expected number of time-steps of order k, with O(1/k) expected excess risk. The constant behind the O notation depends on parameters of the regression and is a poly-logarithmic function of the smallest eigenvalue of H. We provide both a biased and unbiased estimator of the expected excess risk of the time-average estimator and of its unbiased counterpart, without requiring knowledge of either H or θ^*. We describe an "average-start" version of our estimators with similar properties. Our approach is based on randomized multilevel Monte Carlo. Our numerical experiments confirm our theoretical findings.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 11 of its 32 references that are in the map. First posted on arXiv 26 Jun 2024.

Builds on

11 of 32 references.

References · 11

  1. 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
  2. 2013Stochastic Gradient Descent, Weighted Sampling, and the Randomized Kaczmarz algorithmcited by 132
  3. 2013Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n)cited by 110
  4. 2020Variance-Reduced Methods for Machine Learningcited by 79
  5. 2016Harder, Better, Faster, Stronger Convergence Rates for Least-Squares Regressioncited by 64
  6. 2019Exact minimax risk for linear least squares, and the lower tail of sample covariance matricescited by 34
  7. 2021Stochastic Bias-Reduced Gradient Methodscited by 25
  8. 2019Unbiased Multilevel Monte Carlo: Stochastic Optimization, Steady-state Simulation, Quantiles, and Other Applicationscited by 17
  9. 2021Distributed stochastic optimization with large delayscited by 3
  10. 2014Unbiased Monte Carlo: posterior estimation for intractable/infinite-dimensional modelscited by 1
  11. 2018Unbiased inference for discretely observed hidden Markov model diffusionscited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.