8 papers
CMAD: Cooperative Multi-Agent Diffusion via Stochastic Optimal Control
Riccardo Barbano, Alexander Denker, Zeljko Kereta +2
Continuous-time generative models have achieved remarkable success in image restoration and synthesis. However, controlling the composition of multiple pre-trained models remains a…
RNE: plug-and-play diffusion inference-time control and energy-based training
Jiajun He, José Miguel Hernández-Lobato, Yuanqi Du +1
Diffusion models generate data by removing noise gradually, which corresponds to the time-reversal of a noising process. However, access to only the denoising kernels is often insu…
Accelerated Parallel Tempering via Neural Transports
Leo Zhang, Peter Potaptchik, Jiajun He +5
Markov Chain Monte Carlo (MCMC) algorithms are essential tools in computational statistics for sampling from unnormalised probability distributions, but can be fragile when targeti…
CREPE: Controlling Diffusion with Replica Exchange
Jiajun He, Paul Jeha, Peter Potaptchik +5
Inference-time control of diffusion models aims to steer model outputs to satisfy new constraints without retraining. Previous approaches have mostly relied on heuristic guidance o…
Mirror Bridges Between Probability Measures
Leticia Mattos Da Silva, Silvia Sellán, Francisco Vargas +1
Resampling from a target measure whose density is unknown is a fundamental problem in mathematical statistics and machine learning. A setting that dominates the machine learning li…
Sampling from Energy distributions with Target Concrete Score Identity
Sergei Kholkin, Francisco Vargas, Alexander Korotin
We introduce the Target Concrete Score Identity Sampler (TCSIS), a method for sampling from unnormalized densities on discrete state spaces by learning the reverse dynamics of a Co…