English

GPUTB-2:An efficient E(3) network method for learning high-precision orthogonal Hamiltonian

Materials Science 2026-01-21 v1

Abstract

Although equivariant neural networks have become a cornerstone for learning electronic Hamiltonians, the intrinsic non-orthogonality of linear combinations of atomic orbitals (LCAO) basis sets poses a fundamental challenge. The computational cost of Hamiltonian orthogonalization scales as O(N^3), which severely hinders electronic structure calculations for large-scale systems containing hundreds of thousands to millions of atoms. To address this issue, we develop GPUTB-2, a framework that learns implicitly orthogonality-preserving Hamiltonians by training directly on electronic band structures. Benefiting from an E(3)-equivariant network accelerated by Gaunt tensor product and SO(2) tensor product layers, GPUTB-2 achieves significantly higher accuracy than GPUTB across multiple benchmark systems. Moreover, GPUTB-2 accurately predicts large-scale electronic structures, including transport properties of temperature-perturbed SnSe and the band structures of magic-angle twisted bilayer graphene. By further integrating this framework with the linear-scaling quantum transport (LSQT) method, we investigate the electronic properties of million-atom amorphous graphene and uncover pressure-induced electronic structure transitions in more complex amorphous silicon. Together, these results establish GPUTB-2 as a high-accuracy and scalable approach for predicting orthogonal Hamiltonians.

Keywords

Cite

@article{arxiv.2601.13656,
  title  = {GPUTB-2:An efficient E(3) network method for learning high-precision orthogonal Hamiltonian},
  author = {Yunlong Wang and Zhixin Liang and Chi Ding and Junjie Wang and Zheyong Fan and Hui-Tian Wang and Dingyu Xing and Jian Sun},
  journal= {arXiv preprint arXiv:2601.13656},
  year   = {2026}
}
R2 v1 2026-07-01T09:11:56.594Z