arXiv:2506.09398cs.LGphysics.comp-ph2025-06被引 7

用二维局部坐标系高效预测分子哈密顿矩阵,提升电子结构计算速度。

Efficient Prediction of SO(3)-Equivariant Hamiltonian Matrices via SO(2) Local Frames

  • 在二维局部坐标系中进行特征更新与消息传递,避免昂贵的三维旋转变换计算。
  • 在QH9和MD17数据集上表现优于现有方法,对不同分子结构有强泛化能力。
  • 适合需要快速准确计算电子结构的量子化学与材料模拟研究者使用。

我们研究了加速电子结构计算中哈密顿矩阵预测的任务,该任务在物理、化学和材料科学中具有重要意义。受哈密顿矩阵非对角块与SO(2)局部坐标系之间内在关系的启发,提出一种新型高效网络QHNetV2,实现全局SO(3)等变性而无需代价高昂的SO(3) Clebsch-Gordan张量积。通过引入新的高效且强大的SO(2)等变操作,并在每个节点的SO(2)局部坐标系内完成所有非对角特征更新与消息传递,从而消除SO(3)张量积需求。此外,在每个节点的SO(2)局部坐标系中执行连续的SO(2)张量积以融合节点特征,模拟对称收缩操作。在大型QH9和MD17数据集上的大量实验表明,该模型在多种分子结构和轨迹上均取得优异性能,凸显其强大的泛化能力。提出的基于SO(2)局部坐标系的SO(2)操作为可扩展且具备对称性的电子结构学习提供了新方向。代码将作为AIRS库的一部分发布:https://github.com/divelab/AIRS。

原文摘要 · Abstract (English)

We consider the task of predicting Hamiltonian matrices to accelerate electronic structure calculations, which plays an important role in physics, chemistry, and materials science. Motivated by the inherent relationship between the off-diagonal blocks of the Hamiltonian matrix and the SO(2) local frame, we propose a novel and efficient network, called QHNetV2, that achieves global SO(3) equivariance without the costly SO(3) Clebsch-Gordan tensor products. This is achieved by introducing a set of new efficient and powerful SO(2)-equivariant operations and performing all off-diagonal feature updates and message passing within SO(2) local frames, thereby eliminating the need of SO(3) tensor products. Moreover, a continuous SO(2) tensor product is performed within the SO(2) local frame at each node to fuse node features, mimicking the symmetric contraction operation. Extensive experiments on the large QH9 and MD17 datasets demonstrate that our model achieves superior performance across a wide range of molecular structures and trajectories, highlighting its strong generalization capability. The proposed SO(2) operations on SO(2) local frames offer a promising direction for scalable and symmetry-aware learning of electronic structures. Our code will be released as part of the AIRS library https://github.com/divelab/AIRS.

电子结构对称性学习图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。