9
64

Optimal Convergence Rate of Hamiltonian Monte Carlo for Strongly Logconcave Distributions

Abstract

We study Hamiltonian Monte Carlo (HMC) for sampling from a strongly logconcave density proportional to efe^{-f} where f:RdRf:\mathbb{R}^d \to \mathbb{R} is μ\mu-strongly convex and LL-smooth (the condition number is κ=L/μ\kappa = L/\mu). We show that the relaxation time (inverse of the spectral gap) of ideal HMC is O(κ)O(\kappa), improving on the previous best bound of O(κ1.5)O(\kappa^{1.5}); we complement this with an example where the relaxation time is Ω(κ)\Omega(\kappa). When implemented using a nearly optimal ODE solver, HMC returns an ε\varepsilon-approximate point in 22-Wasserstein distance using O~((κd)0.5ε1)\widetilde{O}((\kappa d)^{0.5} \varepsilon^{-1}) gradient evaluations per step and O~((κd)1.5ε1)\widetilde{O}((\kappa d)^{1.5}\varepsilon^{-1}) total time.

View on arXiv
Comments on this paper