Learning electron densities in the condensed phase

Abstract

We introduce a local machine-learning method for predicting the electron densities of periodic systems. The framework is based on a numerical, atom-centred auxiliary basis, which enables an accurate expansion of the all-electron density in a form suitable for learning isolated and periodic systems alike. We show that using this formulation the electron densities of metals, semiconductors and molecular crystals can all be accurately predicted using symmetry-adapted Gaussian process regression models, properly adjusted for the non-orthogonal nature of the basis. These predicted densities enable the efficient calculation of electronic properties which present errors on the order of tens of meV/atom when compared to ab initio density-functional calculations. We demonstrate the key power of this approach by using a model trained on ice unit cells containing only 4 water molecules to predict the electron densities of cells containing up to 512 molecules, and see no increase in the magnitude of the errors of derived electronic properties when increasing the system size. Indeed, we find that these extrapolated derived energies are more accurate than those predicted using a direct machine-learning model. Finally, on heterogeneous datasets SALTED can predict electron densities with errors below 4%.

arXiv comment13 pages, 7 figures; accepted version, SI added

References in corpus (16)

• On representing chemical environments
• Resolution-of-identity approach to Hartree-Fock, hybrid density functionals, RPA, MP2, and \textit{GW} with numeric atom-centered orbital basis functions
• By-passing the Kohn-Sham equations with machine learning
• Machine learning a general purpose interatomic potential for silicon
• i-PI 2.0: A Universal Force Engine for Advanced Molecular Simulations
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• A Transferable Machine-Learning Model of the Electron Density
• Incorporating long-range physics in atomic-scale machine learning
• Accurate molecular polarizabilities with coupled-cluster theory and machine learning
• Atom-Density Representations for Machine Learning
• Challenges in Large Scale Quantum Mechanical Calculations
• Large scale and linear scaling DFT with the CONQUEST code
• Predicting molecular dipole moments by combining atomic partial charges and atomic dipoles
• Using Gaussian Process Regression to Simulate the Vibrational Raman Spectra of Molecular Crystals
• Recursive evaluation and iterative contraction of $N$-body equivariant features

Cited by in corpus (3)

• A Deep Dive into Machine Learning Density Functional Theory for Materials Science and Chemistry
• Equivariant graph neural networks for fast electron density estimation of molecules, liquids, and solids
• Linear Jacobi-Legendre expansion of the charge density for machine learning-accelerated electronic structure calculations