General framework for E(3)-equivariant neural network representation of density functional theory Hamiltonian

Abstract

Combination of deep learning and ab initio calculation has shown great promise in revolutionizing future scientific research, but how to design neural network models incorporating a priori knowledge and symmetry requirements is a key challenging subject. Here we propose an E(3)-equivariant deep-learning framework to represent density functional theory (DFT) Hamiltonian as a function of material structure, which can naturally preserve the Euclidean symmetry even in the presence of spin-orbit coupling. Our DeepH-E3 method enables very efficient electronic-structure calculation at ab initio accuracy by learning from DFT data of small-sized structures, making routine study of large-scale supercells ($>10^4$ atoms) feasible. Remarkably, the method can reach sub-meV prediction accuracy at high training efficiency, showing state-of-the-art performance in our experiments. The work is not only of general significance to deep-learning method development, but also creates new opportunities for materials research, such as building Moiré-twisted material database.

References in corpus (6)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Nearsightedness of Electronic Matter
• The crucial role of atomic corrugation on the flat bands and energy gaps of twisted bilayer graphene at the "magic angle" $\theta \sim 1.08^{\circ }$
• General framework for E(3)-equivariant neural network representation of density functional theory Hamiltonian
• SE(3)-equivariant prediction of molecular wavefunctions and electronic densities

Cited by in corpus (27)

• General framework for E(3)-equivariant neural network representation of density functional theory Hamiltonian
• Current and future directions in network biology
• AI-driven inverse design of materials: Past, present and future
• Deep-learning electronic-structure calculation of magnetic superstructures
• Deep-learning density functional perturbation theory
• Universal materials model of deep-learning density functional theory Hamiltonian
• Universal Machine Learning Kohn-Sham Hamiltonian for Materials
• Machine Learning and Data-Driven Methods in Computational Surface and Interface Science
• AI-Driven Accelerated Discovery of Intercalation-type Cathode Materials for Magnesium Batteries
• Neural-network Density Functional Theory Based on Variational Energy Minimization
• Bridging electronic and classical density-functional theory using universal machine-learned functional approximations
• Unified Deep Learning Framework for Many-Body Quantum Chemistry via Green's Functions
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• Exploring the design space of machine-learning models for quantum chemistry with a fully differentiable framework
• Transferable empirical pseudopotenials from machine learning
• General time-reversal equivariant neural network potential for magnetic materials
• Evolution of flat bands in MoSe${}_2$/WSe${}_2$ moiré lattices: A study combining machine learning and band unfolding methods
• Equivariant neural network for Green's functions of molecules and materials
• Superconductor discovery in the emerging paradigm of Materials Informatics
• Trade-off between Gradient Measurement Efficiency and Expressivity in Deep Quantum Neural Networks
• Review of the tight-binding method applicable to the properties of moiré superlattices
• Accelerating point defect simulations using data-driven and machine learning approaches
• Fault-tolerant Quantum Chemical Calculations with Improved Machine-Learning Models
• Electronic structures of crystalline and amorphous GeSe and GeSbTe compounds using machine learning empirical pseudopotentials
• Machine-Learned Hamiltonians for Quantum Transport Simulation of Valence Change Memories
• Machine-learning modeling of magnetization dynamics in quasi-equilibrium and driven metallic spin systems
• Accelerated Discovery of Topological Conductors for Nanoscale Interconnects