English

On the Saturation Phenomenon of Stochastic Gradient Descent for Linear Inverse Problems

Optimization and Control 2021-08-10 v2 Numerical Analysis Numerical Analysis

Abstract

Stochastic gradient descent (SGD) is a promising method for solving large-scale inverse problems, due to its excellent scalability with respect to data size. The current mathematical theory in the lens of regularization theory predicts that SGD with a polynomially decaying stepsize schedule may suffer from an undesirable saturation phenomenon, i.e., the convergence rate does not further improve with the solution regularity index when it is beyond a certain range. In this work, we present a refined convergence rate analysis of SGD, and prove that saturation actually does not occur if the initial stepsize of the schedule is sufficiently small. Several numerical experiments are provided to complement the analysis.

Keywords

Cite

@article{arxiv.2010.10916,
  title  = {On the Saturation Phenomenon of Stochastic Gradient Descent for Linear Inverse Problems},
  author = {Bangti Jin and Zehui Zhou and Jun Zou},
  journal= {arXiv preprint arXiv:2010.10916},
  year   = {2021}
}

Comments

to appear at SIAM/ASA J. Uncertainty Quantification, with error corrected