用新方法让机器学习加速量子计算,对大分子更有效
Transferable SCF-Acceleration through Solver-Aligned Initialization Learning

- 通过端到端优化初始猜测,让模型更契合求解器需求
- 在更大分子上将迭代次数减少37%~28%,提速超1.3倍
- 适合需要快速模拟大分子的药物研发人员
Kohn-Sham密度泛函理论(KS-DFT)计算成本取决于求解器迭代次数,而迭代次数由初始猜测质量决定。基于分子几何预测初始猜测的机器学习方法可降低成本,但矩阵预测模型在扩展至更大分子时会失效,反而降低收敛速度 [Liu et al., 2025]。我们发现该失败本质是监督问题,而非外推问题:在基态目标上训练良好的模型,虽能准确拟合目标,却产生拖慢收敛的初始猜测。为解决此问题,提出求解器对齐初始化学习(SAIL),通过端到端微分自洽场(SCF)求解器,适用于哈密顿量与密度矩阵模型。引入有效相对迭代次数(ERIC),修正了常用RIC指标,考虑了隐藏的福克构建开销。在包含最大达训练分布4倍大小分子的QM40数据集上,SAIL使ERIC降低37%(PBE)、33%(SCAN)、28%(B3LYP),B3LYP下提升超过前代最佳水平一倍以上。在分子规模达训练集10倍的QMugs数据集上,于杂化理论层级实现1.35倍实际运行时间加速,首次将机器学习加速扩展至类药物大分子。
原文摘要 · Abstract (English)
The cost of Kohn-Sham density functional theory (KS-DFT) calculations scales with the number of solver iterations, which depends on the quality of the initial guess. Machine learning methods that predict initial guesses from molecular geometry can reduce this cost, but matrix-prediction models fail when extrapolating to larger molecules, degrading rather than accelerating convergence [Liu et al., 2025]. We show that this failure is a supervision problem, not an extrapolation problem: models trained on ground-state targets fit those targets well out of distribution, yet produce initial guesses that slow convergence. Solver-Aligned Initialization Learning (SAIL) resolves this for both Hamiltonian and density matrix models by differentiating through the self-consistent field (SCF) solver end-to-end. We introduce the Effective Relative Iteration Count (ERIC), a correction to the commonly used RIC that accounts for hidden Fock-build overhead. On QM40, which contains molecules up to 4$\times$ larger than the training distribution, SAIL reduces ERIC by 37\% (PBE), 33\% (SCAN), and 28\% (B3LYP), more than doubling the previous state-of-the-art reduction on B3LYP. On QMugs molecules 10$\times$ larger than the training set, SAIL delivers a 1.35$\times$ wall-time speedup at the hybrid level of theory, extending ML SCF acceleration to large drug-like molecules.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。