Kinetic energy densities based on the fourth order gradient expansion: performance in different classes of materials and improvement via machine learning

TL;DR

This paper evaluates fourth-order gradient expansions of kinetic energy density in various materials, demonstrating that machine learning can significantly improve their accuracy and stability over traditional methods.

Contribution

It introduces neural network-based approaches trained on fourth-order expansion terms to enhance kinetic energy density predictions in density functional theory.

Findings

Fourth-order expansion does not outperform second-order in formal form.

Neural networks trained on expansion terms improve KED accuracy.

Pseudopotential choice critically affects expansion stability.

Abstract

We study the performance of fourth-order gradient expansions of the kinetic energy density (KED) in semi-local kinetic energy functionals depending on the density-dependent variables. The formal fourth-order expansion is convergent for periodic systems and small molecules but does not improve over the second-order expansion (Thomas-Fermi term plus one-ninth of von Weizs\"acker term). Linear fitting of the expansion coefficients somewhat improves on the formal expansion. The tuning of the fourth order expansion coefficients allows for better reproducibility of Kohn-Sham kinetic energy density than the tuning of the second-order expansion coefficients alone. The possibility of a much more accurate match with the Kohn-Sham kinetic energy density by using neural networks trained using the terms of the 4th order expansion as density-dependent variables is demonstrated. We obtain ultra-low fitting errors without overfitting. Small single hidden layer neural networks can provide good accuracy in separate KED fits of each compound, while for joint fitting of KEDs of multiple compounds multiple hidden layers were required to achieve good fit quality. The critical issue of data distribution is highlighted. We also show the critical role of pseudopotentials in the performance of the expansion, where in the case of a too rapid decay of the valence density at the nucleus with some pseudopotentials, numeric instabilities arise.