5.5MLOct 22, 2024
Error estimates between SGD with momentum and underdamped Langevin diffusionArnaud Guillin, Yu Wang, Lihu Xu et al.
Stochastic gradient descent with momentum is a popular variant of stochastic gradient descent, which has recently been reported to have a close relationship with the underdamped Langevin diffusion. In this paper, we establish a quantitative error estimate between them in the 1-Wasserstein and total variation distances.