arXiv · cs.LG· Sebastian Salazar, Dilum Fernando, Michal Kucer, Emily Casleton, Luhuan Wu, David M. Blei·· 3 小时前
WASABI:用桥插值实现 Wasserstein 采样的扩散模型方法
WASABI: Wasserstein Sampling via Bridge Interpolants
AI 导读
WASABI 通过 Wasserstein 梯度流训练扩散模型,仅凭能量函数即可从目标 Boltzmann 分布采样,且训练算法无需计算模型自身对数密度的梯度。在 alanine dipeptide 上,它比最佳已报告基线少用约 3.3 倍能量评估,并在相互作用粒子系统基准上取得有竞争力的表现。训练完成后生成新样本无需再评估能量或其梯度。
来源:arXiv · cs.LG · arxiv.org