Equivariant Electronic Hamiltonian Prediction with Many-Body Message Passing

Abstract: Machine learning surrogates of Kohn-Sham Density Functional Theory Hamiltonians offer a powerful tool to accelerate the prediction of electronic properties of materials, such as electronic band structures and densities-of-states. For large-scale applications, an ideal model would exhibit high generalization ability and computational efficiency. Here, we introduce the MACE-H graph neural network, which combines high body-order message passing with a node-order expansion to efficiently obtain all relevant $O(3)$ irreducible representations. The model achieves high accuracy and computational efficiency and captures the full local chemical environment features of, currently, up to $f$ orbital matrix interaction blocks. We demonstrate the model's accuracy and transferability on several open materials benchmark datasets of two-dimensional materials and a new dataset for bulk gold, achieving sub-meV prediction errors on matrix elements and eigenvalues across all systems. We further analyse the interplay of high body order message passing and locality that makes this model a good candidate for high-throughput material screening.