Transferable Machine Learning of Electronic Hamiltonians with Superposition-of-Atomic-Potentials Features
基于原子势叠加特征的可迁移电子哈密顿量机器学习
Chaoqun Zhang, Christian Venturella, Enzhi Chen, Tianyu Zhu
AI总结 提出基于原子势叠加(SAP)近似的哈密顿量学习框架,结合对称性适配的原子轨道学习基和轨道图神经网络预测Kohn-Sham Fock矩阵,并通过降维方案扩展到大基组,在QM9和有机电荷传输材料中实现高精度可迁移预测。
详情
电子哈密顿量的机器学习为电子波函数和物理可观测量提供了一条统一途径。我们引入了一个哈密顿量学习框架,该框架基于从原子势叠加(SAP)近似导出的电子特征,这是一种有效的自洽场初始猜测,能够捕获基本的电子-电子屏蔽效应。SAP量定义了一个对称性适配的本征原子轨道学习基,并为基于轨道的图神经网络提供物理信息输入,该网络预测收敛的Kohn-Sham Fock矩阵。为了将方法扩展到更大的基组,我们进一步开发了一个降维方案,从最小基组特征预测大基组电子结构。在QM9数据集上,该模型准确再现了前沿和核心轨道能量、偶极矩以及完整态密度。对于有机电荷传输材料,它产生了苯、四氰基对苯二醌二甲烷(TCNQ)和四硫富瓦烯(TTF)二聚体的精确分子间转移积分,并迁移到未见过的取代苯异质二聚体,平均绝对误差为4.8 meV。这些结果确立了基于SAP的电子哈密顿量机器学习作为高通量电子结构预测的可迁移且可扩展的工具。
Machine learning (ML) of electronic Hamiltonians offers a unified route to electronic wave functions and physical observables. We introduce a Hamiltonian learning framework built on electronic features derived from the superposition-of-atomic-potentials (SAP) approximation, an efficient self-consistent-field initial guess that captures essential electron-electron screening. SAP quantities define a symmetry-adapted intrinsic atomic orbital learning basis and provide physics-informed inputs to an orbital-based graph neural network that predicts converged Kohn-Sham Fock matrices. To extend the approach to larger basis sets, we further develop a downfolding scheme that predicts large-basis electronic structure from minimal-basis features. On the QM9 dataset, the model accurately reproduces frontier and core orbital energies, dipole moments, and the full density of states. For organic charge-transport materials, it yields accurate intermolecular transfer integrals for benzene, tetracyanoquinodimethane (TCNQ), and tetrathiafulvalene (TTF) dimers, and transfers to unseen substituted-benzene heterodimers with a mean absolute error of 4.8 meV. These results establish SAP-based ML of electronic Hamiltonians as a transferable and scalable tool for high-throughput electronic-structure prediction.