用流匹配生成分子哈密顿量,提升量子化学计算速度与精度
High-order Equivariant Flow Matching for Density Functional Theory Hamiltonian Prediction
- 基于高阶等变流匹配框架,从分子结构生成哈密顿矩阵
- 在MD17和QH9数据集上分别降低71%和53%的哈密顿误差
- 适合需要快速高精度量子化学模拟的研究者
密度泛函理论(DFT)是模拟量子化学性质的基础方法,但因其需迭代求解库恩-夏姆方程而计算成本高昂。近年来,深度学习被用于跳过该步骤,直接预测哈密顿量。然而现有方法依赖确定性回归,未考虑哈密顿量的复杂结构。本文提出QHFlow,一种高阶等变流匹配框架,可基于分子几何条件生成哈密顿矩阵。该模型通过连续时间轨迹学习哈密顿量的结构化分布,而非直接回归。为引入对称性,采用预测SE(3)-等变向量场的神经架构,提升跨不同几何结构的准确性与泛化能力。为进一步增强物理保真度,引入微调策略以对齐预测轨道能级与目标值。QHFlow在MD17和QH9数据集上分别实现71%和53%的哈密顿误差降低。此外,将预测哈密顿量用于初始化自洽场(SCF)迭代,显著减少迭代次数与运行时间,同时不牺牲解的质量。
原文摘要 · Abstract (English)
Density functional theory (DFT) is a fundamental method for simulating quantum chemical properties, but it remains expensive due to the iterative self-consistent field (SCF) process required to solve the Kohn-Sham equations. Recently, deep learning methods are gaining attention as a way to bypass this step by directly predicting the Hamiltonian. However, they rely on deterministic regression and do not consider the highly structured nature of Hamiltonians. In this work, we propose QHFlow, a high-order equivariant flow matching framework that generates Hamiltonian matrices conditioned on molecular geometry. Flow matching models continuous-time trajectories between simple priors and complex targets, learning the structured distributions over Hamiltonians instead of direct regression. To further incorporate symmetry, we use a neural architecture that predicts SE(3)-equivariant vector fields, improving accuracy and generalization across diverse geometries. To further enhance physical fidelity, we additionally introduce a fine-tuning scheme to align predicted orbital energies with the target. QHFlow achieves state-of-the-art performance, reducing Hamiltonian error by 71% on MD17 and 53% on QH9. Moreover, we further show that QHFlow accelerates the DFT process without trading off the solution quality when initializing SCF iterations with the predicted Hamiltonian, significantly reducing the number of iterations and runtime.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。