Universal Machine Learning Kohn-Sham Hamiltonian for Materials

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.

20 pages, 9 figures

References in corpus (19)

• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• A Universal Graph Deep Learning Interatomic Potential for the Periodic Table
• Towards Universal Neural Network Potential for Material Discovery Applicable to Arbitrary Combination of 45 Elements
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Deep-Learning Density Functional Theory Hamiltonian for Efficient ab initio Electronic-Structure Calculation
• Bose condensation in flat bands
• Accurate Interatomic Force Fields via Machine Learning with Covariant Kernels
• General framework for E(3)-equivariant neural network representation of density functional theory Hamiltonian
• Machine learning and density functional theory
• Transferable E(3) equivariant parameterization for Hamiltonian of molecules and solids
• Direct Imaging and Electronic Structure Modulation of Moiré Superlattices at the 2D/3D Interface
• Physically inspired deep learning of molecular excitations and photoemission spectra
• Universal superlattice potential for 2D materials from twisted interface inside h-BN substrate
• Equivariant representations for molecular Hamiltonians and N-center atomic-scale properties
• Equivariant analytical mapping of first principles Hamiltonians to accurate and transferable materials models
• A deep neural network for molecular wave functions in quasi-atomic minimal basis representation
• Machine learning approach to genome of two-dimensional materials with flat electronic bands

Cited by in corpus (8)

• AI-driven inverse design of materials: Past, present and future
• Recent Advances in Unconventional Ferroelectrics and Multiferroics
• Bridging electronic and classical density-functional theory using universal machine-learned functional approximations
• Exploring the design space of machine-learning models for quantum chemistry with a fully differentiable framework
• Adaptive energy reference for machine-learning models of the electronic density of states
• Materials discovery acceleration by using condition generative methodology
• Review of the tight-binding method applicable to the properties of moiré superlattices
• Machine learning of electronic structure and atomistic properties from the external potential