We develop Microcanonical Hamiltonian Monte Carlo (MCHMC), a class of models which follow a fixed energy Hamiltonian dynamics, in contrast to Hamiltonian Monte Carlo (HMC), which follows canonical distribution with different energy levels. MCHMC tunes the Hamiltonian function such that the marginal of the uniform distribution on the constant-energy-surface over the momentum variables gives the desired target distribution. We show that MCHMC requires occasional energy conserving billiard-like momentum bounces for ergodicity, analogous to momentum resampling in HMC. We generalize the concept of bounces to a continuous version with partial direction preserving bounces at every step, which gives an energy conserving underdamped Langevin-like dynamics with non-Gaussian noise (MCLMC). MCHMC and MCLMC exhibit favorable scalings with condition number and dimensionality. We develop an efficient hyperparameter tuning scheme that achieves high performance and consistently outperforms NUTS HMC on several standard benchmark problems, in some cases by more than an order of magnitude.
翻译:我们提出了微正则哈密顿蒙特卡罗(MCHMC),这是一类遵循固定能量哈密顿动力学的模型,与遵循不同能量水平正则分布的哈密顿蒙特卡罗(HMC)形成对比。MCHMC调整哈密顿函数,使得动量变量在等能曲面上的均匀分布的边缘分布能够给出期望的目标分布。我们指出,MCHMC需要偶尔的、类似台球碰撞的保能量动量反弹来实现遍历性,这类似于HMC中的动量重采样。我们将反弹的概念推广为一种连续版本,在每个步骤中进行部分方向保持的反弹,从而得到一种具有非高斯噪声的保能量欠阻尼朗之万动力学(MCLMC)。MCHMC和MCLMC在条件数和维度上展现出有利的缩放特性。我们开发了一种高效的超参数调优方案,在多个标准基准问题上实现了高性能,并始终优于NUTS HMC,在某些情况下性能提升超过一个数量级。