A machine-learned kinetic energy model for light weight metals and compounds of group III-V elements

Abstract

We present a machine-learned (ML) model of kinetic energy for orbital-free density functional theory (OF-DFT) suitable for bulk light weight metals and compounds made of group III-V elements. The functional is machine-learned with Gaussian process regression (GPR) from data computed with Kohn-Sham DFT with plane wave bases and local pseudopotentials. The dataset includes multiple phases of unary, binary, and ternary compounds containing Li, Al, Mg, Si, As, Ga, Sb, Na, Sn, P, and In. A total of 433 materials were used for training, and 18 strained structures were used for each material. Averaged (over the unit cell) kinetic energy density is fitted as a function of averaged terms of the 4th order gradient expansion and the product of the density and effective potential. The kinetic energy predicted by the model allows reproducing energy-volume curves around equilibrium geometry with good accuracy. We show that the GPR model beats linear and polynomial regressions. We also find that unary compounds sample a wider region of the descriptor space than binary and ternary compounds, and it is therefore important to include them in the training set; a GPR model trained on a small number of unary compounds is able to extrapolate relatively well to binary and ternary compounds but not vice versa.

References in corpus (10)

• Active learning of reactive Bayesian force fields: Application to heterogeneous hydrogen-platinum catalysis dynamics
• Laplacian-level kinetic energy approximations based on the fourth-order gradient expansion: Global assessment and application to the subsystem formulation of density functional theory
• Orbital-Free Density Functional Theory: Kinetic Potentials and Ab-Initio Local Pseudopotentials
• Data-driven kinetic energy density fitting for orbital-free DFT: linear vs Gaussian process regression
• Nonlocal Pseudopotential Energy Density Functional for Orbital-Free Density Functional Theory
• Order-$N$ orbital-free density-functional calculations with machine learning of functional derivatives for semiconductors and metals
• Highly Accurate Local Pseudopotentials of Li, Na, and Mg for Orbital Free Density Functional Theory
• Degeneration of kernel regression with Matern kernels into low-order polynomial regression in high dimension
• KineticNet: Deep learning a transferable kinetic energy functional for orbital-free density functional theory
• Machine learning of kinetic energy densities with target and feature averaging: better results with fewer training data

Cited by in corpus (1)

• Machine learning-guided construction of an analytic kinetic energy functional for orbital free density functional theory