Processing math: 100%
Skip to yearly menu bar Skip to main content


Poster

Neural Sampling from Boltzmann Densities: Fisher-Rao Curves in the Wasserstein Geometry

Jannis Chemseddine · Christian Wald · Richard Duong · Gabriele Steidl

Hall 3 + Hall 2B #367
[ ]
Fri 25 Apr 7 p.m. PDT — 9:30 p.m. PDT

Abstract: We deal with the task of sampling from an unnormalized Boltzmann density ρDby learning a Boltzmann curve given by energies ft starting in a simple density ρZ.First, we examine conditions under which Fisher-Rao flows are absolutely continuous in the Wasserstein geometry.Second, we address specific interpolations ft and the learning of the related density/velocity pairs (ρt,vt).It was numerically observed that the linear interpolation, which requires only a parametrization of the velocity field vt,suffers from a "teleportation-of-mass" issue.Using tools from the Wasserstein geometry,we give an analytical example,where we can precisely measure the explosion of the velocity field.Inspired by Máté and Fleuret, who parametrize both ft and vt, we propose aninterpolation which parametrizes only ft and fixes an appropriate vt. This corresponds tothe Wasserstein gradient flow of the Kullback-Leibler divergence related to Langevin dynamics. We demonstrate by numerical examples that our model provides a well-behaved flow field which successfully solves the above sampling task.

Live content is unavailable. Log in and register to view live content