472

Learning Deformation Trajectories of Boltzmann Densities

Abstract

We introduce a training objective for continuous normalizing flows that can be used in the absence of samples but in the presence of an energy function. Our method relies on either a prescribed or a learnt interpolation ftf_t of energy functions between the target energy f1f_1 and the energy function of a generalized Gaussian f0(x)=x/σpf_0(x) = |x/\sigma|^p. This, in turn, induces an interpolation of Boltzmann densities pteftp_t \propto e^{-f_t} and we aim to find a time-dependent vector field VtV_t that transports samples along this family of densities. Concretely, this condition can be translated to a PDE between VtV_t and ftf_t and we minimize the amount by which this PDE fails to hold. We compare this objective to the reverse KL-divergence on Gaussian mixtures and on the ϕ4\phi^4 lattice field theory on a circle.

View on arXiv
Comments on this paper