Stochastic Gradient and Langevin Processes
Machine Learning
2020-11-20 v7 Machine Learning
Abstract
We prove quantitative convergence rates at which discrete Langevin-like processes converge to the invariant distribution of a related stochastic differential equation. We study the setup where the additive noise can be non-Gaussian and state-dependent and the potential function can be non-convex. We show that the key properties of these processes depend on the potential function and the second moment of the additive noise. We apply our theoretical findings to studying the convergence of Stochastic Gradient Descent (SGD) for non-convex problems and corroborate them with experiments using SGD to train deep neural networks on the CIFAR-10 dataset.
Cite
@article{arxiv.1907.03215,
title = {Stochastic Gradient and Langevin Processes},
author = {Xiang Cheng and Dong Yin and Peter L. Bartlett and Michael I. Jordan},
journal= {arXiv preprint arXiv:1907.03215},
year = {2020}
}
Comments
ICML 2020, code available at https://github.com/dongyin92/noise_covariance