Sampling in High-dimensions Using Stochastic Interpolants and Forward -backward Stochastic Differential Equations
We present a class of diffusion-based algorithms to draw samples from high-dimensional probability distributions given their unnormalized densities. Ideally, our methods can transport samples from a Gaussian distribution to a specified target distribution in finite time. Our approach relies on the stochastic interpolants framework to define a time-indexed collection of probability densities that bridge a Gaussian distribution to the target distribution. Subsequently, we derive a diffusion process that obeys the aforementioned probability density at each time instant. Obtaining such a diffusion process involves solving certain Hamilton-JacobiBellman PDEs. We solve these PDEs using the theory of forward-backward stochastic differential equations (FBSDE) together with machine learning-based methods. Through numerical experiments, we demonstrate that our algorithm can effectively draw samples from distributions that conventional methods struggle to handle.
WOS:001593416700332
École Polytechnique Fédérale de Lausanne
École Polytechnique Fédérale de Lausanne
2025-01-01
San Diego
Proceedings of Machine Learning Research; 258
2640-3498
REVIEWED
EPFL
| Event name | Event acronym | Event place | Event date |
AISTATS 2025 | Mai Khao, Thailand | 2025-05-03 - 2025-05-05 | |