arXiv:2504.11713v2 通知类型: 替换-交叉 摘要: 我们引入了伴随采样算法，这是一种高度可扩展且高效的算法，用于学习从非标准化密度或能量函数中采样的扩散过程。这是第一个允许显著多于能量评估和模型样本的梯度更新次数的方法，使我们能够扩展到比以前的方法探索的更大规模的问题设置中。我们的框架在随机最优控制方面具有坚实的理论基础，并与伴随匹配具有相同理论保证，能够进行无需采取措施将样本推向目标分布的训练。我们展示了如何在笛卡尔坐标和扭转坐标中结合关键对称性和周期边界条件，以建模分子。通过在经典能量函数上的大量实验，我们展示了该方法的有效性，并进一步扩展到基于神经网络的能量模型中，其中我们在多种分子系统中进行了拟合一致构像生成。为了鼓励进一步研究开发高度可扩展的采样方法，我们计划开源这些具有挑战性的基准，成功的算法可以直接影响计算化学的进步。