Conditional Diffusion Sampling
Francisco M Castro-Macías, Pablo Morales-Alvarez, Saifuddin Syed, Daniel Hernández-Lobato, Rafael Molina, Jose Miguel Hernandez-Lobato
Abstract
Sampling from unnormalized multimodal distributions with limited density evaluations remains a fundamental challenge in machine learning and natural sciences. Successful approaches construct a bridge between a tractable reference and the target distribution. Parallel Tempering (PT) serves as the gold standard, while recent diffusion-based approaches offer a continuous alternative at the cost of neural training. In this work, we introduce Conditional Diffusion Sampling (CDS), a framework that combines these two paradigms. To this end, we derive Conditional Interpolants, a class of stochastic processes whose transport dynamics are governed by an exact, closed-form stochastic differential equation (SDE), requiring no neural approximation. Although these dynamics require sampling from a non-trivial initialization distribution, we show both theoretically and empirically that the cost of this initialization diminishes for sufficiently short diffusion times. CDS leverages this by a two-stage procedure: (1) PT is used to efficiently sample the initial distribution, and then (2) samples are transported via the transport SDE. This combination couples the robust global exploration of PT with efficient local transport. Experiments suggest that CDS has the potential to achieve a superior trade-off between sample quality and density evaluation cost compared to state-of-the-art samplers.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- C^2FG: Control Classifier-Free Guidance via Score Discrepancy AnalysisJiayang Gao, Tianyi Zheng, Jiayang Zou, Fengxiang Yang et al.CVPR 2026
- Rethinking Visual Counterfactual Explanations Through Region ConstraintBartlomiej Sobieski, Jakub Grzywaczewski, Bartlomiej Sadlej, Matthew Tivnan et al.ICLR 2025
Builds on18
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- What Are Bayesian Neural Network Posteriors Really Like?Pavel Izmailov, Sharad Vikram, Matthew D. Hoffman, Andrew Gordon WilsonICML 2021 · 458 citations
- Equivariant flow matchingLeon Klein, Andreas Krämer, Frank NoéNeurIPS 2023 · 169 citations
- Iterated Denoising Energy Matching for Sampling from Boltzmann DensitiesTara Akhound-Sadegh, Jarrid Rector-Brooks, Avishek Joey Bose, Sarthak Mittal et al.ICML 2024 · 109 citations
- Annealed Flow Transport Monte CarloMichael Arbel, Alexander G. de G. Matthews, Arnaud DoucetICML 2021 · 99 citations
Related papers
- Progressive Tempering Sampler with DiffusionSeveri Rissanen, Ruikang Ouyang, Jiajun He, Wenlin Chen et al.ICML 2025
- Accelerated Parallel Tempering via Neural TransportsLeo Zhang, Peter Potaptchik, Jiajun He, Yuanqi Du et al.ICLR 2026 · 14 citations
- Sequential Controlled Langevin DiffusionsJunhua Chen, Lorenz Richter, Julius Berner, Denis Blessing et al.ICLR 2025
- Test-Time Guidance for Flow-Based Generative Models via Parallel Tempering on Source DistributionsShih-Hsin Wang, Joel Keller, Taos Transue, Drake Brown et al.ICML 2026
- Proximal Diffusion Neural SamplerWei Guo, Jaemoo Choi, Yuchen Zhu, Molei Tao et al.ICLR 2026 · 19 citations
