Machine learning of electronic structure and atomistic properties from the external potential

Abstract

Electronic structure calculations remain a major bottleneck in atomistic simulations and, not surprisingly, have attracted significant attention in machine learning (ML). Most existing approaches learn a direct map from molecular geometries, typically represented as graphs or encoded local environments, to molecular properties or use ML as a surrogate for electronic structure theory by targeting quantities such as Fock or density matrices expressed in an atomic orbital (AO) basis. Inspired by the Hohenberg-Kohn theorem, in this work, we propose an operator-centered framework in which the external (nuclear) potential, expressed in an AO basis, serves as the model input. From this operator, we construct hierarchical, body-ordered representations of atomic configurations that closely mirror the principles underlying several popular atom-centered descriptors. At the same time, the matrix-valued nature of the external potential provides a natural connection to equivariant message-passing neural networks. In particular, we show that successive products of the external potential provide a scalable route to equivariant message passing and enable an efficient description of long-range effects. We demonstrate that this approach can be used to model molecular properties, such as energies and dipole moments, from the external potential, or learn effective operator-to-operator maps, including mappings to the Fock matrix and the reduced density matrix from which multiple molecular observables can be simultaneously derived.

References in corpus (28)

• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• On representing chemical environments
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Moment Tensor Potentials: a class of systematically improvable interatomic potentials
• By-passing the Kohn-Sham equations with machine learning
• Machine Learning of Molecular Electronic Properties in Chemical Compound Space
• Physics-inspired structural representations for molecules and materials
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• A Transferable Machine-Learning Model of the Electron Density
• OrbNet: Deep Learning for Quantum Chemistry Using Symmetry-Adapted Atomic-Orbital Features
• Transferability in Machine Learning for Electronic Structure via the Molecular Orbital Basis
• On the Completeness of Atomic Structure Representations
• Atom-Density Representations for Machine Learning
• Predicting molecular dipole moments by combining atomic partial charges and atomic dipoles
• Recursive evaluation and iterative contraction of $N$-body equivariant features
• Physically inspired deep learning of molecular excitations and photoemission spectra
• Equivariant representations for molecular Hamiltonians and N-center atomic-scale properties
• Equivariant analytical mapping of first principles Hamiltonians to accurate and transferable materials models
• Differentiable quantum chemistry with PySCF for molecules and materials at the mean-field level and beyond
• Universal Machine Learning Kohn-Sham Hamiltonian for Materials
• Optimal radial basis for density-based atomic representations
• SPA${}^{\mathrm{H}}$M: the Spectrum of Approximated Hamiltonian Matrices representations
• Unified Deep Learning Framework for Many-Body Quantum Chemistry via Green's Functions
• Exploring the design space of machine-learning models for quantum chemistry with a fully differentiable framework
• A smooth basis for atomistic machine learning
• Molecular Dipole Moment Learning via Rotationally Equivariant Gaussian Process Regression with Derivatives in Molecular-orbital-based Machine Learning
• Equivariant Electronic Hamiltonian Prediction with Many-Body Message Passing