Improved sampling via learned diffusions
Not on the map. Why?
Cited by 47 in Diffusion models, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Recently, a series of papers proposed deep learning-based approaches to sample from target distributions using controlled diffusion processes, being trained only on the unnormalized target densities without access to samples. Building on previous work, we identify these approaches as special cases of a generalized Schr"odinger bridge problem, seeking a stochastic evolution between a given prior distribution and the specified target. We further generalize this framework by introducing a variational formulation based on divergences between path space measures of time-reversed diffusion processes. This abstract perspective leads to practical losses that can be optimized by gradient-based algorithms and includes previous objectives as special cases. At the same time, it allows us to consider divergences other than the reverse Kullback-Leibler divergence that is known to suffer from mode collapse. In particular, we propose the so-called log-variance loss, which exhibits favorable numerical properties and leads to significantly improved performance across all considered approaches.
Key facts. Cited by 103 papers in the map (data through 7 Oct 2026). Builds on 33 of its 83 references that are in the map. First posted on arXiv 3 Jul 2023.
Builds on
33 of 83 references.
References · 33
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,277
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2021Variational Diffusion Modelscited by 1,145
- 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,036
- 2021Score-based Generative Modeling in Latent Spacecited by 634
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
- 2022Torsional Diffusion for Molecular Conformer Generationcited by 262
- 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 244
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 216
21 more
- 2021A Variational Perspective on Diffusion-Based Generative Models and Score Matchingcited by 182
- 2021Path Integral Sampler: a stochastic control approach for samplingcited by 175
- 2020Stochastic Normalizing Flowscited by 156
- 2022An optimal control perspective on diffusion-based generative modelingcited by 145
- 2023Denoising Diffusion Samplerscited by 144
- 2022Flow Annealed Importance Sampling Bootstrapcited by 124
- 2022GFlowNets and variational inferencecited by 103
- 2017Sticking the Landing: Simple, Lower-Variance Gradient Estimators for Variational Inferencecited by 97
- 2019Theoretical guarantees for sampling and inference in generative models with latent diffusionscited by 97
- 2021Annealed Flow Transport Monte Carlocited by 90
- 2020Solving high-dimensional Hamilton-Jacobi-Bellman PDEs using neural networks: perspectives from the theory of controlled diffusions and measures on path spacecited by 80
- 2020VarGrad: A Low-Variance Gradient Estimator for Variational Inferencecited by 73
- 2021Bayesian Learning via Neural Schr"odinger-F"ollmer Flowscited by 56
- 2022Score-Based Diffusion meets Annealed Importance Samplingcited by 51
- 2019Wasserstein Proximal Algorithms for the Schr"odinger Bridge Problem: Density Control with Nonlinear Driftcited by 50
- 2022Deep Generalized Schr"odinger Bridgecited by 47
- 2021Resampling Base Distributions of Normalizing Flowscited by 42
- 2019Schr"odinger Bridge Samplerscited by 14
- 2022Robust SDE-Based Variational Formulations for Solving Linear PDEs via Deep Learningcited by 13
- 2022Neural Lagrangian Schr"odinger Bridge: Diffusion Modeling for Population Dynamicscited by 7
- 2022On local entropy, stochastic control and deep neural networkscited by 7
Built on by
Citing papers per year: 2023: 2; 2024: 31; 2025: 43; 2026: 27.
103 papers.
Sort:
By area:
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 113
- 2025Feynman-Kac Correctors in Diffusion: Annealing, Guidance, and Product of Expertscited by 85
- 2025Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matchingcited by 78
- 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 70
- 2024Particle Denoising Diffusion Samplercited by 63
- 2024A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimizationcited by 58
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2024Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Samplingcited by 55
- 2025DIME:Diffusion-Based Maximum Entropy Reinforcement Learningcited by 52
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Sequential Controlled Langevin Diffusionscited by 47
- 2024Target Score Matchingcited by 46
38 more
- 2024DEFT: Efficient Fine-Tuning of Diffusion Models by Learning the Generalised h-transformcited by 44
- 2023Stochastic Optimal Control Matchingcited by 39
- 2025Scalable Equilibrium Sampling with Sequential Boltzmann Generatorscited by 36
- 2025LEAPS: A discrete neural sampler via locally equivariant networkscited by 30
- 2025Scalable Discrete Diffusion Samplers: Combinatorial Optimization and Statistical Physicscited by 28
- 2025Underdamped Diffusion Bridges with Applications to Samplingcited by 28
- 2024Stochastic Localization via Iterative Posterior Samplingcited by 27
- 2025Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densitiescited by 26
- 2025No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplerscited by 25
- 2024Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave Samplingcited by 23
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 22
- 2025DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Modelscited by 21
- 2025Adjoint Schr"odinger Bridge Samplercited by 20
- 2024Zeroth-Order Sampling Methods for Non-Log-Concave Distributions: Alleviating Metastability by Denoising Diffusioncited by 19
- 2024Dynamical Measure Transport and Neural PDE Solvers for Samplingcited by 19
- 2025From discrete-time policies to continuous-time diffusion samplers: Asymptotic equivalences and faster trainingcited by 17
- 2024Stochastic Optimal Control for Diffusion Bridges in Function Spacescited by 16
- 2025Outsourced diffusion sampling: Efficient posterior inference in latent spaces of generative modelscited by 16
- 2025Proximal Diffusion Neural Samplercited by 16
- 2024BNEM: A Boltzmann Sampler Based on Bootstrapped Noised Energy Matchingcited by 14
- 2025Temperature-Annealed Boltzmann Generatorscited by 14
- 2025Accelerated Parallel Tempering via Neural Transportscited by 13
- 2025Discrete Neural Flow Samplers with Locally Equivariant Transformercited by 13
- 2025Inference-Time Scaling of Discrete Diffusion Models via Importance Weighting and Optimal Proposal Designcited by 12
- 2024Harmonic Path Integral Diffusioncited by 11
- 2025Complexity Analysis of Normalizing Constant Estimation: from Jarzynski Equality to Annealed Importance Sampling and beyondcited by 11
- 2025Amortized Sampling with Transferable Normalizing Flowscited by 11
- 2024Transition Path Sampling with Improved Off-Policy Training of Diffusion Path Samplerscited by 10
- 2024Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flowscited by 9
- 2024Learned Reference-based Diffusion Sampling for multi-modal distributionscited by 9
- 2025Improving the evaluation of samplers on multi-modal targetscited by 9
- 2024PQMass: Probabilistic Assessment of the Quality of Generative Models using Probability Mass Estimationcited by 8
- 2024Soft-constrained Schrodinger Bridge: a Stochastic Control Approachcited by 8
- 2025Non-equilibrium Annealed Adjoint Samplercited by 8
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributionscited by 7
- 2024Bridging discrete and continuous state spaces: Exploring the Ehrenfest process in time-continuous diffusion modelscited by 7
- 2024Importance Corrected Neural JKO Samplingcited by 7
- 2025Rethinking Losses for Diffusion Bridge Samplerscited by 7
50 most cited of 103 All 103 (JSON)
- 2026Wasserstein Gradient Flows and Forward-Only Diffusion Are Not Enough for Multimodal Sampling1 Oct 2026
- 2026Neural Transport Nested Sampling24 Sep 2026
- 2026Mode Coverage in Normalizing Flow Boltzmann Generators via Log-Ratio Variation8 Sep 2026
- 2026FrOGS: Discrete Neural Sampler for Independent Alloy Configurations Across Chemical Conditions1 Sep 2026
- 2026Exact Global MCMC with Denoising Diffusion31 Aug 2026
- 2026ATLAS: A Foundation Neural Sampler for Amorphous Materials21 Jul 2026
- 2026Tensor Train Diffusion: Leveraging Low-Rank Structures for High-Dimensional Score-Based Sampling7 Jul 2026
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matching21 Jun 2026
- 2026Constrained Diffusion Models with Primal-Dual Inference15 Jun 2026
- 2026Trust-Region Diffusion Policies for Massively Parallel On-Policy RL13 Jun 2026
- 2026A Stabilized Path-Space Approach to Diffusion-Based Posterior Sampling10 Jun 2026
- 2026Scalable Inference-Time Annealing with Surrogate Likelihood Estimators29 May 2026
38 more
- 2026A Unified Framework for Data-Free One-Step Sampling via Wasserstein Gradient Flows18 May 2026
- 2026Flow Sampling: Learning to Sample from Unnormalized Densities via Denoising Conditional Processes5 May 2026
- 2026FES-FM: Free Energy Surface Sampling via Reduced Flow Matching1 May 2026
- 2026Total Variation Guarantees for Sampling with Stochastic Localization31 Mar 2026
- 2026Adjoint Matching through the Lens of the Stochastic Maximum Principle in Optimal Control28 Mar 2026
- 2026Alternating Diffusion for Proximal Sampling with Zeroth Order Queries20 Mar 2026
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching28 Feb 2026
- 2026Unsupervised Continual Learning for Amortized Bayesian Inference26 Feb 2026
- 2026Annealing in variational inference mitigates mode collapse: A theoretical study on Gaussian mixtures13 Feb 2026
- 2026Inverting Data Transformations via Diffusion Sampling9 Feb 2026
- 2026Discrete Adjoint Schr"odinger Bridge Sampler9 Feb 2026
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spaces5 Feb 2026
- 2026Efficient Training of Boltzmann Generators Using Off-Policy Log-Dispersion Regularization3 Feb 2026
- 2026Neural Predictor-Corrector: Solving Homotopy Problems with Reinforcement Learning3 Feb 2026
- 2026Diffusion Path Samplers via Sequential Monte Carlo29 Jan 2026
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flow4 Dec 2025
- 2025Diffusion-Augmented Markov Decision Processes for Maximum Entropy Reinforcement Learning1 Dec 2025
- 2025Particle Monte Carlo methods for Lattice Field Theory19 Nov 2025
- 2025Fractional Diffusion Bridge Models3 Nov 2025
- 2025Schr"odinger bridge for generative AI: Soft-constrained formulation and convergence analysis13 Oct 2025
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
- 2025Control-Augmented Autoregressive Diffusion for Data Assimilation8 Oct 2025
- 2025Proximal Diffusion Neural Sampler4 Oct 2025
- 2025Data-to-Energy Stochastic Dynamics30 Sep 2025
- 2025DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Models25 Sep 2025
- 2025Error Propagation in Dynamic Programming: From Stochastic Control to American Option Pricing24 Sep 2025
- 2025Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Sampling3 Sep 2025
- 2025Continuously Tempered Diffusion Samplers30 Aug 2025
- 2025Amortized Sampling with Transferable Normalizing Flows25 Aug 2025
- 2025Sampling by averaging: A multiscale approach to score estimation20 Aug 2025
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
- 2025Posterior Inference in Latent Space for Scalable Constrained Black-box Optimization1 Jul 2025
- 2025Adjoint Schr"odinger Bridge Sampler27 Jun 2025
- 2025Non-equilibrium Annealed Adjoint Sampler22 Jun 2025
- 2025Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densities19 Jun 2025
- 2025Rethinking Losses for Diffusion Bridge Samplers12 Jun 2025
- 2025Adaptive Destruction Processes for Diffusion Samplers2 Jun 2025
- 2025Aligning Protein Conformation Ensemble Generation with Physical Feedback30 May 2025
50 newest of 103 All 103 (JSON)
- 2025Feynman-Kac Correctors in Diffusion: Annealing, Guidance, and Product of Expertscited by 85
- 2025Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matchingcited by 78
- 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 70
- 2024Particle Denoising Diffusion Samplercited by 63
- 2024A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimizationcited by 58
- 2025DIME:Diffusion-Based Maximum Entropy Reinforcement Learningcited by 52
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Target Score Matchingcited by 46
- 2024DEFT: Efficient Fine-Tuning of Diffusion Models by Learning the Generalised h-transformcited by 44
- 2025Scalable Discrete Diffusion Samplers: Combinatorial Optimization and Statistical Physicscited by 28
- 2025Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densitiescited by 26
- 2025Adjoint Schr"odinger Bridge Samplercited by 20
12 most cited of 45 All 45 (JSON)
- 2026Exact Global MCMC with Denoising Diffusion31 Aug 2026
- 2026Tensor Train Diffusion: Leveraging Low-Rank Structures for High-Dimensional Score-Based Sampling7 Jul 2026
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matching21 Jun 2026
- 2026Constrained Diffusion Models with Primal-Dual Inference15 Jun 2026
- 2026Trust-Region Diffusion Policies for Massively Parallel On-Policy RL13 Jun 2026
- 2026A Stabilized Path-Space Approach to Diffusion-Based Posterior Sampling10 Jun 2026
- 2026Flow Sampling: Learning to Sample from Unnormalized Densities via Denoising Conditional Processes5 May 2026
- 2026FES-FM: Free Energy Surface Sampling via Reduced Flow Matching1 May 2026
- 2026Total Variation Guarantees for Sampling with Stochastic Localization31 Mar 2026
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching28 Feb 2026
- 2026Inverting Data Transformations via Diffusion Sampling9 Feb 2026
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spaces5 Feb 2026
12 newest of 45 All 45 (JSON)
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 113
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2024Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Samplingcited by 55
- 2024Sequential Controlled Langevin Diffusionscited by 47
- 2023Stochastic Optimal Control Matchingcited by 39
- 2025Scalable Equilibrium Sampling with Sequential Boltzmann Generatorscited by 36
- 2025LEAPS: A discrete neural sampler via locally equivariant networkscited by 30
- 2025Underdamped Diffusion Bridges with Applications to Samplingcited by 28
- 2024Stochastic Localization via Iterative Posterior Samplingcited by 27
- 2025No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplerscited by 25
- 2024Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave Samplingcited by 23
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 22
12 most cited of 56 All 56 (JSON)
- 2026Wasserstein Gradient Flows and Forward-Only Diffusion Are Not Enough for Multimodal Sampling1 Oct 2026
- 2026Neural Transport Nested Sampling24 Sep 2026
- 2026Mode Coverage in Normalizing Flow Boltzmann Generators via Log-Ratio Variation8 Sep 2026
- 2026FrOGS: Discrete Neural Sampler for Independent Alloy Configurations Across Chemical Conditions1 Sep 2026
- 2026ATLAS: A Foundation Neural Sampler for Amorphous Materials21 Jul 2026
- 2026Scalable Inference-Time Annealing with Surrogate Likelihood Estimators29 May 2026
- 2026A Unified Framework for Data-Free One-Step Sampling via Wasserstein Gradient Flows18 May 2026
- 2026Adjoint Matching through the Lens of the Stochastic Maximum Principle in Optimal Control28 Mar 2026
- 2026Alternating Diffusion for Proximal Sampling with Zeroth Order Queries20 Mar 2026
- 2026Unsupervised Continual Learning for Amortized Bayesian Inference26 Feb 2026
- 2026Annealing in variational inference mitigates mode collapse: A theoretical study on Gaussian mixtures13 Feb 2026
- 2026Discrete Adjoint Schr"odinger Bridge Sampler9 Feb 2026
12 newest of 56 All 56 (JSON)