REVIEW 7 cited by
Iterated Denoising Energy Matching for Sampling from Boltzmann Densities
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
abstract
Efficiently generating statistically independent samples from an unnormalized probability distribution, such as equilibrium samples of many-body systems, is a foundational problem in science. In this paper, we propose Iterated Denoising Energy Matching (iDEM), an iterative algorithm that uses a novel stochastic score matching objective leveraging solely the energy function and its gradient -- and no data samples -- to train a diffusion-based sampler. Specifically, iDEM alternates between (I) sampling regions of high model density from a diffusion-based sampler and (II) using these samples in our stochastic matching objective to further improve the sampler. iDEM is scalable to high dimensions as the inner matching objective, is simulation-free, and requires no MCMC samples. Moreover, by leveraging the fast mode mixing behavior of diffusion, iDEM smooths out the energy landscape enabling efficient exploration and learning of an amortized sampler. We evaluate iDEM on a suite of tasks ranging from standard synthetic energy functions to invariant $n$-body particle systems. We show that the proposed approach achieves state-of-the-art performance on all metrics and trains $2-5\times$ faster, which allows it to be the first method to train using energy on the challenging $55$-particle Lennard-Jones system.
Forward citations
Cited by 7 Pith papers
-
GeMPO: Generalized Measure Matching for Online Diffusion Reinforcement Learning
GeMPO unifies diffusion RL reweighting as measure matching to a regularized target, enabling flexible and negative weights that improve exploration and performance.
-
FES-FM: Free Energy Surface Sampling via Reduced Flow Matching
FES-FM learns a reduced flow-matching transport in collective-variable space to sample free energy surfaces, cutting per-sample generation cost while leaving full-space training cost unchanged.
-
Self-Refining Training for Amortized Density Functional Theory
A self-refining training loop, where a neural network samples molecular conformations from its own predicted energy and trains on them, reduces the need for large labeled DFT datasets in amortized density functional theory.
-
Towards Adaptive External Communication in Autonomous Vehicles: A Conceptual Design Framework
A three-layer framework (input, processing, output) for adaptive external human-machine interfaces in autonomous vehicles is introduced to systematize design and analysis.
-
Torsional-GFN: a conditional conformation generator for small molecules
Torsional-GFN, a conditional GFlowNet with a new graph network, samples torsion angles of small molecules to approximate the Boltzmann distribution, with partial generalization to unseen local structures.
-
Importance Weighted Score Matching for Diffusion Samplers with Enhanced Mode Coverage
Importance Weighted Score Matching trains diffusion samplers by reweighting score matching with self-normalized importance sampling to approximate the forward KL and improve mode coverage.
-
Neural Flow Samplers with Shortcut Models
Neural Flow Shortcut Sampler (NFS2) estimates the partition-function derivative with velocity-driven SMC and Stein control variates, and adds a generalized shortcut consistency loss for few-step sampling.
Discussion (0). Continue with ORCID to comment.