arXiv:2506.01225cs.LG2025-06

用自精炼训练减少对大量数据的依赖,高效预测分子能量。

Self-Refining Training for Amortized Density Functional Theory

  • 通过自生成训练数据并同步优化模型,降低对预收集数据的依赖。
  • 在分子动力学模拟中,相比传统方法提升计算效率,保持精度。
  • 适合需要大规模分子能量预测的研究者,如材料与药物设计。

密度泛函理论(DFT)通过近似求解多体薛定谔方程,可从头计算分子体系的化学与物理性质。然而,随着能量评估规模增大,如用于分子动力学模拟时,计算成本急剧上升。近期研究发现,基于深度学习的模型可通过摊销经典DFT求解器的优化问题,在大规模分子构象数据集上实现输出预测。本文提出一种新型自精炼训练策略,显著降低摊销DFT求解器对预收集数据的依赖。该方法同时训练深度学习模型预测DFT输出,并生成用于训练的分子构象样本。我们将其表述为最小化变分上界下的KL散度,衡量生成样本与由基态能量定义的目标玻尔兹曼分布之间的差异。通过广泛实验证明,该方法在性能上优于使用预收集数据训练的模型。我们开源了优化后的算法实现,采用异步训练与采样阶段,支持并行采样与训练。代码见:https://github.com/majhas/self-refining-dft。

原文摘要 · Abstract (English)

Density Functional Theory (DFT) allows for predicting all the chemical and physical properties of molecular systems from first principles by finding an approximate solution to the many-body Schrödinger equation. However, the cost of these predictions becomes infeasible when increasing the scale of the energy evaluations, e.g., when calculating the ground-state energy for simulating molecular dynamics. Recent works have demonstrated that, for substantially large datasets of molecular conformations, Deep Learning-based models can predict the outputs of the classical DFT solvers by amortizing the corresponding optimization problems. In this paper, we propose a novel method that reduces the dependency of amortized DFT solvers on large pre-collected datasets by introducing a self-refining training strategy. Namely, we propose an efficient method that simultaneously trains a deep-learning model to predict the DFT outputs and samples molecular conformations that are used as training data for the model. We derive our method as a minimization of the variational upper bound on the KL-divergence measuring the discrepancy between the generated samples and the target Boltzmann distribution defined by the ground state energy. To demonstrate the utility of the proposed scheme, we perform an extensive empirical study comparing it with the models trained on the pre-collected datasets. Finally, we open-source our implementation of the proposed algorithm, optimized with asynchronous training and sampling stages, which enables simultaneous sampling and training. Code is available at https://github.com/majhas/self-refining-dft.

密度泛函深度学习自精炼分子模拟

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