Neural network distillation of orbital dependent density functional theory

Abstract

Density functional theory (DFT) offers a desirable balance between quantitative accuracy and computational efficiency in practical many-electron calculations. Its central component, the exchange-correlation energy functional, has been approximated with increasing levels of complexity ranging from strictly local approximations to nonlocal and orbital-dependent expressions with many tuned parameters. In this paper, we formulate a general way of rewriting complex density functionals using deep neural networks in a way that allows for simplified computation of Kohn-Sham potentials as well as higher functional derivatives through automatic differentiation, enabling access to highly nonlinear response functions and forces. These goals are achieved by using a recently developed class of robust neural network models capable of modeling functionals, as opposed to functions, with explicitly enforced spatial symmetries. Functionals treated in this way are then called global density approximations and can be seamlessly integrated with existing DFT workflows. Tests are performed for a dataset featuring a large variety of molecular structures and popular meta-generalized gradient approximation density functionals, where we successfully eliminate orbital dependencies coming from the kinetic energy density, and discover a high degree of transferability to a variety of physical systems. The presented framework is general and could be extended to more complex orbital and energy dependent functionals as well as refined with specialized datasets.

6 pages, 3 figures (supplementary material 8 pages, 3 figures)

References in corpus (19)

• Climbing the Density Functional Ladder: Non-Empirical Meta-Generalized Gradient Approximation Designed for Molecules and Solids
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• The GW method
• Perspective on density functional theory
• Recent developments in the PySCF program package
• Finding Density Functionals with Machine Learning
• Libxc: a library of exchange and correlation functionals for density functional theory
• The GW compendium: A practical guide to theoretical photoemission spectroscopy
• A Review of automatic differentiation and its efficient implementation
• Regularized SCAN functional
• Semilocal and Hybrid Meta-Generalized Gradient Approximations Based on the Understanding of the Kinetic-Energy-Density Dependence
• DeePKS: a comprehensive data-driven approach towards chemically accurate density functional theory
• Design and Analysis of Machine Learning Exchange-Correlation Functionals via Rotationally Invariant Convolutional Descriptors
• Variational Benchmarks for Quantum Many-Body Problems
• Neural-network quantum states for many-body physics
• Nonlocal Machine-Learned Exchange Functional for Molecules and Solids
• Grad DFT: a software library for machine learning enhanced density functional theory
• Variational principle to regularize machine-learned density functionals: the non-interacting kinetic-energy functional
• Addressing the Band Gap Problem with a Machine-Learned Exchange Functional