A Transferable Machine-Learning Model of the Electron Density

Abstract

The electronic charge density plays a central role in determining the behavior of matter at the atomic scale, but its computational evaluation requires demanding electronic-structure calculations. We introduce an atom-centered, symmetry-adapted framework to machine-learn the valence charge density based on a small number of reference calculations. The model is highly transferable, meaning it can be trained on electronic-structure data of small molecules and used to predict the charge density of larger compounds with low, linear-scaling cost. Applications are shown for various hydrocarbon molecules of increasing complexity and flexibility, and demonstrate the accuracy of the model when predicting the density on octane and octatetraene after training exclusively on butane and butadiene. This transferable, data-driven model can be used to interpret experiments, initialize electronic structure calculations, and compute electrostatic interactions in molecules and condensed-phase systems.

References in corpus (23)

• Maximally-localized generalized Wannier functions for composite energy bands
• 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
• Deep Potential Molecular Dynamics: a scalable model with the accuracy of quantum mechanics
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Comparing molecules and solids across structural and alchemical space
• Machine Learning Unifies the Modelling of Materials and Molecules
• By-passing the Kohn-Sham equations with machine learning
• Machine-learning based interatomic potential for amorphous carbon
• Nearsightedness of Electronic Matter
• i-PI 2.0: A Universal Force Engine for Advanced Molecular Simulations
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Interatomic potentials for ionic systems with density functional accuracy based on charge densities obtained by a neural network
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• Transferability in Machine Learning for Electronic Structure via the Molecular Orbital Basis
• Accurate Interatomic Force Fields via Machine Learning with Covariant Kernels
• Accurate molecular polarizabilities with coupled-cluster theory and machine learning
• Machine learning of molecular properties: locality and active learning
• Transferable atomic multipole machine learning models for small organic molecules
• Guest Editorial: Special Topic on Data-enabled Theoretical Chemistry
• Solvent Fluctuations and Nuclear Quantum Effects Modulate the Molecular Hyperpolarizability of Water
• An efficient and accurate decomposition of the Fermi operator

Cited by in corpus (80)

• Combining Machine Learning and Computational Chemistry for Predictive Insights Into Chemical Systems
• Physics-inspired structural representations for molecules and materials
• Deep neural network solution of the electronic Schrödinger equation
• Machine learning for electronically excited states of molecules
• Towards self-driving laboratories: The central role of density functional theory in the AI age
• OrbNet: Deep Learning for Quantum Chemistry Using Symmetry-Adapted Atomic-Orbital Features
• Deep-Learning Density Functional Theory Hamiltonian for Efficient ab initio Electronic-Structure Calculation
• Perspective on integrating machine learning into computational chemistry and materials science
• On the Completeness of Atomic Structure Representations
• Atom-Density Representations for Machine Learning
• Deep neural network for the dielectric response of insulators
• Learning the electronic density of states in condensed matter
• A Deep Dive into Machine Learning Density Functional Theory for Materials Science and Chemistry
• A Universal Density Matrix Functional from Molecular Orbital-Based Machine Learning: Transferability across Organic Molecules
• OrbNet Denali: A machine learning potential for biological and organic chemistry with semi-empirical cost and DFT accuracy
• Uncertainty estimation for molecular dynamics and sampling
• Electronic Density Response of Warm Dense Matter
• DeePKS: a comprehensive data-driven approach towards chemically accurate density functional theory
• Quantum deep field: data-driven wave function, electron density generation, and atomization energy prediction and extrapolation with machine learning
• Recursive evaluation and iterative contraction of $N$-body equivariant features
• Regression-clustering for Improved Accuracy and Training Cost with Molecular-Orbital-Based Machine Learning
• Predicting charge density distribution of materials using a local-environment-based graph convolutional network
• Equivariant graph neural networks for fast electron density estimation of molecules, liquids, and solids
• Gaussian Process-Based Refinement of Dispersion Corrections
• Equivariant representations for molecular Hamiltonians and N-center atomic-scale properties
• PANNA: Properties from Artificial Neural Network Architectures
• Improved accuracy and transferability of molecular-orbital-based machine learning: Organics, transition-metal complexes, non-covalent interactions, and transition states
• Spectroscopy from Machine Learning by Accurately Representing the Atomic Polar Tensor
• Learning from the Density to Correct Total Energy and Forces in First Principle Simulations
• Hydration free energies from kernel-based machine learning: Compound-database bias
• Predicting the Charge Density Response in Metal Electrodes
• Finite-temperature materials modeling from the quantum nuclei to the hot electrons regime
• Linear Jacobi-Legendre expansion of the charge density for machine learning-accelerated electronic structure calculations
• Ligand additivity relationships enable efficient exploration of transition metal chemical space
• Reinforcement learning for traversing chemical structure space: Optimizing transition states and minimum energy paths of molecules
• 3DReact: Geometric deep learning for chemical reactions
• Machine learning density functionals from the random-phase approximation
• Atomic structure optimization with machine-learning enabled interpolation between chemical elements
• Accuracy of Charge Densities in Electronic Structure Calculations
• Representations and Strategies for Transferable Machine Learning Models in Chemical Discovery
• Quantum chemical roots of machine-learning molecular similarity descriptors
• Accurate Hellmann-Feynman forces from density functional calculations with augmented Gaussian basis sets
• Bridging electronic and classical density-functional theory using universal machine-learned functional approximations
• Machine-learned prediction of the electronic fields in a crystal
• Predicting the Electronic Density Response of Condensed-Phase Systems to Electric Field Perturbations
• Unified Deep Learning Framework for Many-Body Quantum Chemistry via Green's Functions
• Learning the energy curvature versus particle number in approximate density functionals
• Predicting The One-Particle Density Matrix With Machine Learning
• Exploring the design space of machine-learning models for quantum chemistry with a fully differentiable framework
• Learning on-top: regressing the on-top pair density for real-space visualization of electron correlation
• Deep Density: circumventing the Kohn-Sham equations via symmetry preserving neural networks
• Density functionals and Kohn-Sham potentials with minimal wavefunction preparations on a quantum computer
• Predicting hot-electron free energies from ground-state data
• Low-data deep quantum chemical learning for accurate MP2 and coupled-cluster correlations
• SPA${}^{\mathrm{H}}$M(a,b): encoding the density information from guess Hamiltonian in quantum machine learning representations
• Metatensor and metatomic: foundational libraries for interoperable atomistic machine learning
• Machine learning for accuracy in density functional approximations
• Machine learning at the atomic-scale
• Graph-neural-network predictions of solid-state NMR parameters from spherical tensor decomposition
• Equivariant Machine Learning Interatomic Potentials with Global Charge Redistribution
• Molecular Dipole Moment Learning via Rotationally Equivariant Gaussian Process Regression with Derivatives in Molecular-orbital-based Machine Learning
• Impact of quantum-chemical metrics on the machine learning prediction of electron density
• On the equivalence of molecular graph convolution and molecular wave function with poor basis set
• Embedding quantum statistical excitations in a classical force field
• Atomistic Simulations of Oxide-Water Interfaces using Machine Learning Potentials
• Machine-Learning Surrogate Model for Accelerating the Search of Stable Ternary Alloys
• Machine Learning Wavefunction
• Learning the Electrostatic Response of the Electron Density through a Symmetry-Adapted Vector Field Model
• Physical machine learning outperforms "human learning" in Quantum Chemistry
• Learning Density Functionals from Noisy Quantum Data
• Deep-learning atomistic semi-empirical pseudopotential model for nanomaterials
• Incorporating electronic information into Machine Learning potential energy surfaces via approaching the ground-state electronic energy as a function of atom-based electronic populations
• Electronic structures of crystalline and amorphous GeSe and GeSbTe compounds using machine learning empirical pseudopotentials
• Learning electron densities in the condensed phase
• Machine learning based prediction of the electronic structure of quasi-one-dimensional materials under strain
• A deep learning approach to the structural analysis of proteins
• Machine learning of electronic structure and atomistic properties from the external potential
• Fast evaluation of interaction integrals for confined systems with machine learning
• Integer linear programming for unsupervised training set selection in molecular machine learning
• A charge-density machine-learning workflow for computing the infrared spectrum of molecules