Wasserstein Sampling via Bridge Interpolants

新的 WASABI 方法使用扩散模型进行玻尔兹曼分布采样

研究人员推出了一种新颖的从玻尔兹曼分布生成样本的方法，称为 WASABI（Wasserstein Sampling via Bridge Interpolants）。这种基于扩散模型的方法通过最小化其生成分布与目标玻尔兹曼分布之间的 KL 散度进行训练，利用 Wasserstein 梯度流。WASABI 提供了一种实用的训练算法，可以绕过对难以处理的对数密度梯度的需求，并在训练后高效生成新样本，在诸如丙氨酸二肽等系统的能量评估方面优于现有基线。