Transferable E(3) equivariant parameterization for Hamiltonian of molecules and solids

Abstract

Using the message-passing mechanism in machine learning (ML) instead of self-consistent iterations to directly build the mapping from structures to electronic Hamiltonian matrices will greatly improve the efficiency of density functional theory (DFT) calculations. In this work, we proposed a general analytic Hamiltonian representation in an E(3) equivariant framework, which can fit the ab initio Hamiltonian of molecules and solids by a complete data-driven method and are equivariant under rotation, space inversion, and time reversal operations. Our model reached state-of-the-art precision in the benchmark test and accurately predicted the electronic Hamiltonian matrices and related properties of various periodic and aperiodic systems, showing high transferability and generalization ability. This framework provides a general transferable model that can be used to accelerate the electronic structure calculations on different large systems with the same network weights trained on small structures.

33 pages, 6 figures

References in corpus (8)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• On the Convergence of Adam and Beyond
• SIESTA: recent developments and applications
• GPUMD: A package for constructing accurate machine-learned potentials and performing highly efficient atomistic simulations
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Unified Graph Neural Network Force-field for the Periodic Table
• GeoMol: Torsional Geometric Generation of Molecular 3D Conformer Ensembles
• SE(3)-equivariant prediction of molecular wavefunctions and electronic densities

Cited by in corpus (13)

• AI-driven inverse design of materials: Past, present and future
• Universal materials model of deep-learning density functional theory Hamiltonian
• Universal Machine Learning Kohn-Sham Hamiltonian for Materials
• Recent Advances in Unconventional Ferroelectrics and Multiferroics
• 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
• General time-reversal equivariant neural network potential for magnetic materials
• Equivariant neural network for Green's functions of molecules and materials
• Materials discovery acceleration by using condition generative methodology
• Equivariant Electronic Hamiltonian Prediction with Many-Body Message Passing
• Revealing Higher-Order Topological Bulk-boundary Correspondence in Bismuth Crystal with Spin-helical Hinge State Loop and Proximity Superconductivity
• Review of the tight-binding method applicable to the properties of moiré superlattices
• Accelerated Discovery of Topological Conductors for Nanoscale Interconnects