arXiv:2501.19077cs.LG2025-01ICML被引 14

通过温度渐进训练,提升分子系统采样精度与效率。

Temperature-Annealed Boltzmann Generators

  • 用逆KL散度在高温下训练生成模型,避免模式崩溃。
  • 引入重加权目标函数实现温度渐进下降,精准捕获构象空间。
  • 在复杂分子系统中显著提升采样效果,减少能量评估次数。

高效采样未归一化概率密度(如分子系统的玻尔兹曼分布)是一个长期挑战。除了传统的分子动力学或马尔可夫链蒙特卡洛方法外,基于变分方法的正常化流(normalizing flows)也已被提出,但这类方法容易发生模式崩溃,难以覆盖完整的构象空间。本文提出温度渐进玻尔兹曼生成器(TA-BG)以解决该问题。首先,我们证明在高温下使用逆KL散度训练正常化流是可行的且不会出现模式崩溃。其次,引入基于重加权的训练目标,实现从高温到低温的分布渐进退火。我们将该方法应用于三个复杂度递增的分子系统,相比基线方法,在几乎所有指标上表现更优,且目标能量评估次数最多减少三分之二。在最大系统中,本方法是唯一能准确分辨亚稳态的方案。

原文摘要 · Abstract (English)

Efficient sampling of unnormalized probability densities such as the Boltzmann distribution of molecular systems is a longstanding challenge. Next to conventional approaches like molecular dynamics or Markov chain Monte Carlo, variational approaches, such as training normalizing flows with the reverse Kullback-Leibler divergence, have been introduced. However, such methods are prone to mode collapse and often do not learn to sample the full configurational space. Here, we present temperature-annealed Boltzmann generators (TA-BG) to address this challenge. First, we demonstrate that training a normalizing flow with the reverse Kullback-Leibler divergence at high temperatures is possible without mode collapse. Furthermore, we introduce a reweighting-based training objective to anneal the distribution to lower target temperatures. We apply this methodology to three molecular systems of increasing complexity and, compared to the baseline, achieve better results in almost all metrics while requiring up to three times fewer target energy evaluations. For the largest system, our approach is the only method that accurately resolves the metastable states of the system.

生成模型分子模拟温度退火正常化流

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。