Linear Jacobi-Legendre expansion of the charge density for machine learning-accelerated electronic structure calculations

Abstract

Kohn-Sham density functional theory (KS-DFT) is a powerful method to obtain key materials' properties, but the iterative solution of the KS equations is a numerically intensive task, which limits its application to complex systems. To address this issue, machine learning (ML) models can be used as surrogates to find the ground-state charge density and reduce the computational overheads. We develop a grid-centred structural representation, based on Jacobi and Legendre polynomials combined with a linear regression, to accurately learn the converged DFT charge density. This integrates into a ML pipeline that can return any density-dependent observable, including energy and forces, at the quality of a converged DFT calculation, but at a fraction of the computational cost. Fast scanning of energy landscapes and producing starting densities for the DFT self-consistent cycle are among the applications of our scheme.

References in corpus (35)

• On representing chemical environments
• Quantum Spin Hall Effect and Topological Field Effect Transistor in Two-Dimensional Transition Metal Dichalcogenides
• Perspective on density functional theory
• Machine Learning Force Fields
• Ab initio Random Structure Searching
• A Performance and Cost Assessment of Machine Learning Interatomic Potentials
• DScribe: Library of Descriptors for Machine Learning in Materials Science
• Machine Learning Unifies the Modelling of Materials and Molecules
• By-passing the Kohn-Sham equations with machine learning
• On-the-fly machine learning force field generation: Application to melting points
• Finding Density Functionals with Machine Learning
• Deep neural network solution of the electronic Schrödinger equation
• Opportunities and Challenges for Machine Learning in Materials Science
• Machine learning modeling of superconducting critical temperature
• Accelerating crystal structure prediction by machine-learning interatomic potentials with active learning
• Unified Representation of Molecules and Crystals for Machine Learning
• A Transferable Machine-Learning Model of the Electron Density
• A critical examination of compound stability predictions from machine-learned formation energies
• On the Completeness of Atomic Structure Representations
• Machine learning with force-field inspired descriptors for materials: fast screening and mapping energy landscape
• Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physics
• Orbital-free Bond Breaking via Machine Learning
• Large scale and linear scaling DFT with the CONQUEST code
• Machine learning and density functional theory
• Predicting the Curie temperature of ferromagnets using machine learning
• Functional Form of the Superconducting Critical Temperature from Machine Learning
• Accelerating Finite-temperature Kohn-Sham Density Functional Theory with Deep Neural Networks
• Machine learning for materials discovery: two-dimensional topological insulators
• Detection of Topological Materials with Machine Learning
• Machine learning force fields based on local parametrization of dispersion interactions: Application to the phase diagram of C${}_{60}$
• Equivariant analytical mapping of first principles Hamiltonians to accurate and transferable materials models
• Machine learning density functional theory for the Hubbard model
• Toward Orbital-Free Density Functional Theory with Small Data Sets and Deep Learning
• Inversion of the chemical environment representations
• Learning electron densities in the condensed phase

Cited by in corpus (7)

• Performance Assessment of Universal Machine Learning Interatomic Potentials: Challenges and Directions for Materials' Surfaces
• Predicting The One-Particle Density Matrix With Machine Learning
• Convolutional network learning of self-consistent electron density via grid-projected atomic fingerprints
• Machine learning for accuracy in density functional approximations
• Machine-Learning Surrogate Model for Accelerating the Search of Stable Ternary Alloys
• A general formalism for machine-learning models based on multipolar-spherical harmonics
• A charge-density machine-learning workflow for computing the infrared spectrum of molecules