How Well Does Kohn-Sham Regularizer Work for Weakly Correlated Systems?

Abstract

Kohn-Sham regularizer (KSR) is a differentiable machine learning approach to finding the exchange-correlation functional in Kohn-Sham density functional theory (DFT) that works for strongly correlated systems. Here we test KSR for weak correlation. We propose spin-adapted KSR (sKSR) with trainable local, semilocal, and nonlocal approximations found by minimizing density and total energy loss. We assess the atoms-to-molecules generalizability by training on one-dimensional (1D) H, He, Li, Be, Be${}^{++}$ and testing on 1D hydrogen chains, LiH, BeH${}_2$, and helium hydride complexes. The generalization error from our semilocal approximation is comparable to other differentiable approaches, but our nonlocal functional outperforms any existing machine learning functionals, predicting ground-state energies of test systems with a mean absolute error of 2.7 milli-Hartrees.

References in corpus (12)

• Array Programming with NumPy
• Perspective on density functional theory
• By-passing the Kohn-Sham equations with machine learning
• Finding Density Functionals with Machine Learning
• Understanding and reducing errors in density functional calculations
• Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physics
• A Differentiable Programming System to Bridge Machine Learning and Scientific Computing
• Learning the exchange-correlation functional from nature with fully differentiable density functional theory
• DeePKS: a comprehensive data-driven approach towards chemically accurate density functional theory
• Reference electronic structure calculations in one dimension
• One Dimensional Mimicking of Electronic Structure: The Case for Exponentials
• Using differentiable programming to obtain an energy and density-optimized exchange-correlation functional

Cited by in corpus (6)

• Grad DFT: a software library for machine learning enhanced density functional theory
• Automatic Differentiation for Orbital-Free Density Functional Theory
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• Machine learning for accuracy in density functional approximations
• Inverting the Kohn-Sham equations with physics-informed machine learning
• Quantum-Enhanced Neural Exchange-Correlation Functionals