用神经网络预测电子密度,实现DFT计算的可迁移加速。
Towards A Transferable Acceleration Method for Density Functional Theory
- 用E(3)等变神经网络在辅助基组中预测电子密度作为初始猜测。
- 小分子训练后,60原子系统平均减少33.3%自洽迭代次数。
- 对900原子聚合物仍有效,无需重训,是首个可通用的DFT加速方法。
近期深度学习方法通过生成高效初始猜测以加速密度泛函理论(DFT)计算。尽管实际初始猜测常为密度矩阵(DM),其转换形式也可作为替代。现有工作多依赖哈密顿量矩阵预测,但该矩阵数值难预测且固有不可迁移,限制了实际应用。为此,我们提出一种新方法:利用E(3)-等变神经网络在紧凑辅助基组中预测电子密度,构建DFT初始猜测。模型仅在最多20个原子的小分子上训练,即在三倍大小(最多60原子)的分子上实现平均33.3%的自洽场(SCF)迭代减少。相比之下,基准哈密顿量方法在这些大体系上普遍失效,迭代次数增加超80%或完全无法收敛。此外,该方法具有强可扩展性:在未重新训练的情况下成功加速含900原子的聚合物与多肽体系。据我们所知,这是首个真正具备通用迁移能力的DFT加速方法。我们还发布了SCFbench数据集及配套代码,以推动该方向研究。
原文摘要 · Abstract (English)
Recently, sophisticated deep learning-based approaches have been developed for generating efficient initial guesses to accelerate the convergence of density functional theory (DFT) calculations. While the actual initial guesses are often density matrices (DM), quantities that can convert into density matrices also qualify as alternative forms of initial guesses. Hence, existing works mostly rely on the prediction of the Hamiltonian matrix for obtaining high-quality initial guesses. However, the Hamiltonian matrix is both numerically difficult to predict and intrinsically non-transferable, hindering the application of such models in real scenarios. In light of this, we propose a method that constructs DFT initial guesses by predicting the electron density in a compact auxiliary basis representation using E(3)-equivariant neural networks. Trained exclusively on small molecules with up to 20 atoms, our model achieves an average 33.3% reduction in SCF iterations for molecules three times larger (up to 60 atoms). This result is particularly significant given that baseline Hamiltonian-based methods fail to generalize, often increasing the iteration count by over 80% or failing to converge entirely on these larger systems. Furthermore, we demonstrate that this acceleration is robustly scalable: the model successfully accelerates calculations for systems with up to 900 atoms (polymers and polypeptides) without retraining. To the best of our knowledge, this work represents the first and robust candidate for a universally transferable DFT acceleration method. We also released the SCFbench dataset and its accompanying code to facilitate future research in this promising direction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。