arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1802.04791stat.MLcs.LGstat.CO
收藏

Stochastic Variance-Reduced Hamilton Monte Carlo Methods

University of California, Los Angeles(加州大学洛杉矶分校)

Difan Zou, Pan Xu, Quanquan Gu

英文摘要:

We propose a fast stochastic Hamilton Monte Carlo (HMC) method, for sampling from a smooth and strongly log-concave distribution. At the core of our proposed method is a variance reduction technique inspired by the recent advance in stochastic optimization. We show that, to achieve $ε$ accuracy in 2-Wasserstein distance, our algorithm achieves $\tilde O(n+κ^{2}d^{1/2}/ε+κ^{4/3}d^{1/3}n^{2/3}/ε^{2/3})$ gradient complexity (i.e., number of component gradient evaluations), which outperforms the state-of-the-art HMC and stochastic gradient HMC methods in a wide regime. We also extend our algorithm for sampling from smooth and general log-concave distributions, and prove the corresponding gradient complexity as well. Experiments on both synthetic and real data demonstrate the superior performance of our algorithm.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑