AI & ML
WASABI: Wasserstein Sampling via Bridge Interpolants
arXiv stat.ML0 views
arXiv:2610.41958v1 Announce Type: new Abstract: Sampling from Boltzmann distributions is a fundamental problem in physics, chemistry, and biology. We introduce WAsserstein SAmpling via Bridge Interpolants (WASABI), a method that trains a diffusion model to generate samples from a target Boltzmann distribution given access only to the energy function. WASABI minimizes the KL divergence between the distribution generated by its diffusion model and the Boltzmann target using Wasserstein gradient flows. This approach builds on a principle underlying classical sampling methods such as Langevin dynamics. However, applying Wasserstein gradient flows to diffusion models is challenging because the updates require the gradient of the model's own log density, which is generally intractable. We derive a practical training algorithm that avoids computing this gradient. Once trained, WASABI can generate new samples without further evaluations of the energy or its gradient. On alanine dipeptide, WASABI outperforms the best reported baselines using an estimated 3.3 times fewer energy evaluations. It also achieves competitive performance on interacting particle system benchmarks.
Source:https://arxiv.org/abs/2610.41958PreviousStatistical Foundations of Reinforcement Learning: A Non-Asymptotic PerspectiveNextRegret-Optimal Wasserstein-Robust Linear Regression
Related posts

Stop Writing the Perfect Prompt. Claude Code's Docs Say Let It Interview You
0dev.to AI
$2 Trillion to $1.5 Trillion: How the 2026 Midterms Turned Anthropic’s IPO Into a Stress Test
0dev.to AI
ViKernel's Action-Permission Boundary: What VIDRAFT's 661-Run Internal Test Actually Tells Engineers
0dev.to AI
Hydra · Coeziune OM–IA v13: new DOI and the complete DOI registry (2026-11-06)
0dev.to AI
Merge PDF Pages in the Browser: Reorder, Rotate and Export Without Uploading
0dev.to WebDev
Tech trends, week of Nov 6, 2026: AI security research, audit skills, Images 2.6
0dev.to Security