SITA:可扩展推理时退火采样,避免昂贵散度计算

Scalable Inference-Time Annealing with Surrogate Likelihood Estimators

精选理由

SITA 解决了大分子系统退火采样中散度计算不可行的痛点,做计算化学和分子模拟的团队可以直接用开源代码复现,值得关注。

AI 摘要

计算化学中高效采样分子玻尔兹曼分布是一个长期挑战。传统方法计算成本高,而基于扩散模型的退火采样方法需要计算分数场的散度来估计重要性权重,对于大分子系统不可行。本文提出可扩展推理时退火(SITA),通过能量模型提供快速替代似然,重新训练流模型以逐步在更低温度下生成样本。在丙氨酸二肽和三肽上取得最先进性能,且避免了昂贵的散度项。代码已开源。

AI 翻译 · 中文

计算化学中高效采样分子玻尔兹曼分布是一个长期挑战。传统方法计算成本高,而基于扩散模型的退火采样方法需要计算分数场的散度来估计重要性权重,对于大分子系统不可行。本文提出可扩展推理时退火(SITA),通过能量模型提供快速替代似然,重新训练流模型以逐步在更低温度下生成样本。在丙氨酸二肽和三肽上取得最先进性能,且避免了昂贵的散度项。代码已开源。

arXiv cs.LGA long standing challenge in computational chemistry and biophysics is efficiently sampling the Boltzmann distribution of molecules. Advances in generative modeling have been proposed to address the limitations of conven