HMC with Normalizing Flows

INSPIRE: 1983469
arXiv: 2112.01586
DOI: 10.22323/1.396.0073

Authors: Foreman, Sam, Izubuchi, Taku, Jin, Luchang, Jin, Xiao-Yong, Osborn, James C., Tomiya, Akio

Submitted: 2 December 2021

Subjects:

• cs.LG
• hep-lat
• Computing
• Lattice

Journal reference: PoS LATTICE2021 073 (2022)

Citations: 34

Abstract

We propose using Normalizing Flows as a trainable kernel within the molecular dynamics update of Hamiltonian Monte Carlo (HMC). By learning (invertible) transformations that simplify our dynamics, we can outperform traditional methods at generating independent configurations. We show that, using a carefully constructed network architecture, our approach can be easily scaled to large lattice volumes with minimal retraining effort. The source code for our implementation is publicly available online at https://www.github.com/nftqcd/fthmc.