English

Non-Asymptotic Analysis of Online Multiplicative Stochastic Gradient Descent

Machine Learning 2023-03-02 v9 Machine Learning Statistics Theory Statistics Theory

Abstract

Past research has indicated that the covariance of the Stochastic Gradient Descent (SGD) error done via minibatching plays a critical role in determining its regularization and escape from low potential points. Motivated by some new research in this area, we prove universality results by showing that noise classes that have the same mean and covariance structure of SGD via minibatching have similar properties. We mainly consider the Multiplicative Stochastic Gradient Descent (M-SGD) algorithm as introduced in previous work, which has a much more general noise class than the SGD algorithm done via minibatching. We establish non asymptotic bounds for the M-SGD algorithm in the Wasserstein distance. We also show that the M-SGD error is approximately a scaled Gaussian distribution with mean 00 at any fixed point of the M-SGD algorithm.

Keywords

Cite

@article{arxiv.2112.07110,
  title  = {Non-Asymptotic Analysis of Online Multiplicative Stochastic Gradient Descent},
  author = {Riddhiman Bhattacharya and Tiefeng Jiang},
  journal= {arXiv preprint arXiv:2112.07110},
  year   = {2023}
}