Universal Machine Learning Kohn-Sham Hamiltonian for Materials

Yang Zhong,Hongyu Yu,Jihui Yang,Xingyu Guo,Hongjun Xiang,Xingao Gong

2024-04-15

Abstract:While density functional theory (DFT) serves as a prevalent computational approach in electronic structure calculations, its computational demands and scalability limitations persist. Recently, leveraging neural networks to parameterize the Kohn-Sham DFT Hamiltonian has emerged as a promising avenue for accelerating electronic structure computations. Despite advancements, challenges such as the necessity for computing extensive DFT training data to explore each new system and the complexity of establishing accurate ML models for multi-elemental materials still exist. Addressing these hurdles, this study introduces a universal electronic Hamiltonian model trained on Hamiltonian matrices obtained from first-principles DFT calculations of nearly all crystal structures on the Materials Project. We demonstrate its generality in predicting electronic structures across the whole periodic table, including complex multi-elemental systems, solid-state electrolytes, Moiré twisted bilayer heterostructure, and metal-organic frameworks (MOFs). Moreover, we utilize the universal model to conduct high-throughput calculations of electronic structures for crystals in GeNOME datasets, identifying 3,940 crystals with direct band gaps and 5,109 crystals with flat bands. By offering a reliable efficient framework for computing electronic properties, this universal Hamiltonian model lays the groundwork for advancements in diverse fields, such as easily providing a huge data set of electronic structures and also making the materials design across the whole periodic table possible.

Computational Physics,Materials Science,Artificial Intelligence

What problem does this paper attempt to address?

The paper aims to address the challenges in predicting the electronic Hamiltonian of materials using machine learning, such as the need for a large amount of training data from Density Functional Theory (DFT) and the complexity of building accurate models for multi-element materials. The study proposes a universal electronic Hamiltonian model that utilizes a two-step training method and trains with 55,000 crystal structure data from the Materials Project. This model can predict the electronic structure of various materials in the periodic table, including complex multi-element systems. This approach improves prediction efficiency and accuracy, providing an efficient framework for material design and understanding electronic properties.