用物理约束提升量子化学计算速度与精度,让大分子模拟更可行
Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systems
- 引入基于波函数对齐的损失函数,增强模型对分子轨道和能量的预测能力
- 总能量误差降低1347倍,自洽场计算提速18%,突破大分子模拟瓶颈
- 适合需要高精度且大规模分子模拟的研究者,尤其在材料与药物设计中
密度泛函理论(DFT)是量子化学与材料科学中的核心方法,其关键在于构建与求解郭恩-申哈姆哈密顿量。尽管重要,但其应用常受限于构造该哈密顿量所需的大量计算资源。为应对这一挑战,现有研究采用深度学习模型高效预测分子与固体哈密顿量,并在神经网络中编码旋转平移对称性。然而,先前模型在处理大分子时可能面临可扩展性问题,导致基态性质预测出现非物理解。本研究构建了一个比以往大得多的训练集(PubChemQH),并基于此开发出一种可扩展且具有物理准确性的DFT计算模型。模型引入基于物理原理的损失函数——波函数对齐损失(WALoss),通过将预测哈密顿量进行基变换以对齐观测结果,使差值可作为轨道能差的代理,从而显著提升对分子轨道和总能量的预测精度。此外,WALoss大幅加速自洽场(SCF)DFT计算,实现总能量预测误差降低1347倍、SCF计算速度提升18%。这些显著改进为更大分子系统的高精度与可应用预测树立了新基准。
原文摘要 · Abstract (English)
Density Functional Theory (DFT) is a pivotal method within quantum chemistry and materials science, with its core involving the construction and solution of the Kohn-Sham Hamiltonian. Despite its importance, the application of DFT is frequently limited by the substantial computational resources required to construct the Kohn-Sham Hamiltonian. In response to these limitations, current research has employed deep-learning models to efficiently predict molecular and solid Hamiltonians, with roto-translational symmetries encoded in their neural networks. However, the scalability of prior models may be problematic when applied to large molecules, resulting in non-physical predictions of ground-state properties. In this study, we generate a substantially larger training set (PubChemQH) than used previously and use it to create a scalable model for DFT calculations with physical accuracy. For our model, we introduce a loss function derived from physical principles, which we call Wavefunction Alignment Loss (WALoss). WALoss involves performing a basis change on the predicted Hamiltonian to align it with the observed one; thus, the resulting differences can serve as a surrogate for orbital energy differences, allowing models to make better predictions for molecular orbitals and total energies than previously possible. WALoss also substantially accelerates self-consistent-field (SCF) DFT calculations. Here, we show it achieves a reduction in total energy prediction error by a factor of 1347 and an SCF calculation speed-up by a factor of 18%. These substantial improvements set new benchmarks for achieving accurate and applicable predictions in larger molecular systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。