摘要
arXiv:2504.11713v1 类型: cross
摘要: 我们引入了一种名为伴随采样的高效算法,该算法适用于从未规范概率密度或能量函数中采样扩散过程的学习。这是首个允许进行的梯度更新数远多于能量评价和模型样本数量的在策略方法,从而使我们能够将问题规模扩展到先前类似方法未曾探索的更大范围。我们的框架理论基础为随机最优控制,并与伴随匹配方法具有相同的理论保证,能够在无需使用将样本推向目标分布的纠正措施的情况下进行训练。我们展示了如何在笛卡尔坐标和扭转坐标下整合关键对称性和周期边界条件,以建模分子。通过在经典能量函数上的大量实验,我们展示了该方法的有效性,并进一步扩展到基于神经网络的能量模型,在这些模型中我们在许多分子系统中进行了优化构象生成。为了鼓励进一步研究开发大规模采样方法,我们计划开源这些具有挑战性的基准,成功的方法可以直接影响计算化学的进步。