Saved in:
Bibliographic Details
Main Authors: Chen, Junhua, Richter, Lorenz, Berner, Julius, Blessing, Denis, Neumann, Gerhard, Anandkumar, Anima
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2412.07081
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915483560706048
author Chen, Junhua
Richter, Lorenz
Berner, Julius
Blessing, Denis
Neumann, Gerhard
Anandkumar, Anima
author_facet Chen, Junhua
Richter, Lorenz
Berner, Julius
Blessing, Denis
Neumann, Gerhard
Anandkumar, Anima
contents An effective approach for sampling from unnormalized densities is based on the idea of gradually transporting samples from an easy prior to the complicated target distribution. Two popular methods are (1) Sequential Monte Carlo (SMC), where the transport is performed through successive annealed densities via prescribed Markov chains and resampling steps, and (2) recently developed diffusion-based sampling methods, where a learned dynamical transport is used. Despite the common goal, both approaches have different, often complementary, advantages and drawbacks. The resampling steps in SMC allow focusing on promising regions of the space, often leading to robust performance. While the algorithm enjoys asymptotic guarantees, the lack of flexible, learnable transitions can lead to slow convergence. On the other hand, diffusion-based samplers are learned and can potentially better adapt themselves to the target at hand, yet often suffer from training instabilities. In this work, we present a principled framework for combining SMC with diffusion-based samplers by viewing both methods in continuous time and considering measures on path space. This culminates in the new Sequential Controlled Langevin Diffusion (SCLD) sampling method, which is able to utilize the benefits of both methods and reaches improved performance on multiple benchmark problems, in many cases using only 10% of the training budget of previous diffusion-based samplers.
format Preprint
id arxiv_https___arxiv_org_abs_2412_07081
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Sequential Controlled Langevin Diffusions
Chen, Junhua
Richter, Lorenz
Berner, Julius
Blessing, Denis
Neumann, Gerhard
Anandkumar, Anima
Machine Learning
Artificial Intelligence
An effective approach for sampling from unnormalized densities is based on the idea of gradually transporting samples from an easy prior to the complicated target distribution. Two popular methods are (1) Sequential Monte Carlo (SMC), where the transport is performed through successive annealed densities via prescribed Markov chains and resampling steps, and (2) recently developed diffusion-based sampling methods, where a learned dynamical transport is used. Despite the common goal, both approaches have different, often complementary, advantages and drawbacks. The resampling steps in SMC allow focusing on promising regions of the space, often leading to robust performance. While the algorithm enjoys asymptotic guarantees, the lack of flexible, learnable transitions can lead to slow convergence. On the other hand, diffusion-based samplers are learned and can potentially better adapt themselves to the target at hand, yet often suffer from training instabilities. In this work, we present a principled framework for combining SMC with diffusion-based samplers by viewing both methods in continuous time and considering measures on path space. This culminates in the new Sequential Controlled Langevin Diffusion (SCLD) sampling method, which is able to utilize the benefits of both methods and reaches improved performance on multiple benchmark problems, in many cases using only 10% of the training budget of previous diffusion-based samplers.
title Sequential Controlled Langevin Diffusions
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2412.07081