English

Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave Sampling

Machine Learning 2025-02-18 v2 Machine Learning Statistics Theory Computation Statistics Theory

Abstract

We consider the outstanding problem of sampling from an unnormalized density that may be non-log-concave and multimodal. To enhance the performance of simple Markov chain Monte Carlo (MCMC) methods, techniques of annealing type have been widely used. However, quantitative theoretical guarantees of these techniques are under-explored. This study takes a first step toward providing a non-asymptotic analysis of annealed MCMC. Specifically, we establish, for the first time, an oracle complexity of O~(dβ2A2ε6)\widetilde{O}\left(\frac{d\beta^2{\cal A}^2}{\varepsilon^6}\right) for the simple annealed Langevin Monte Carlo algorithm to achieve ε2\varepsilon^2 accuracy in Kullback-Leibler divergence to the target distribution πeV\pi\propto{\rm e}^{-V} on Rd\mathbb{R}^d with β\beta-smooth potential VV. Here, A{\cal A} represents the action of a curve of probability measures interpolating the target distribution π\pi and a readily sampleable distribution.

Keywords

Cite

@article{arxiv.2407.16936,
  title  = {Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave Sampling},
  author = {Wei Guo and Molei Tao and Yongxin Chen},
  journal= {arXiv preprint arXiv:2407.16936},
  year   = {2025}
}

Comments

ICLR 2025 camera ready https://openreview.net/forum?id=P6IVIoGRRg