skip to main content

Title: Machine-learning Kohn–Sham potential from dynamics in time-dependent Kohn–Sham systems

The construction of a better exchange-correlation potential in time-dependent density functional theory (TDDFT) can improve the accuracy of TDDFT calculations and provide more accurate predictions of the properties of many-electron systems. Here, we propose a machine learning method to develop the energy functional and the Kohn–Sham potential of a time-dependent Kohn–Sham (TDKS) system is proposed. The method is based on the dynamics of the Kohn–Sham system and does not require any data on the exact Kohn–Sham potential for training the model. We demonstrate the results of our method with a 1D harmonic oscillator example and a 1D two-electron example. We show that the machine-learned Kohn–Sham potential matches the exact Kohn–Sham potential in the absence of memory effect. Our method can still capture the dynamics of the Kohn–Sham system in the presence of memory effects. The machine learning method developed in this article provides insight into making better approximations of the energy functional and the Kohn–Sham potential in the TDKS system.

more » « less
Author(s) / Creator(s):
Publisher / Repository:
IOP Publishing
Date Published:
Journal Name:
Machine Learning: Science and Technology
Page Range / eLocation ID:
Article No. 035022
Medium: X
Sponsoring Org:
National Science Foundation
More Like this
  1. Abstract

    The Hohenberg-Kohn theorem of density-functional theory establishes the existence of a bijection between the ground-state electron density and the external potential of a many-body system. This guarantees a one-to-one map from the electron density to all observables of interest including electronic excited-state energies. Time-Dependent Density-Functional Theory (TDDFT) provides one framework to resolve this map; however, the approximations inherent in practical TDDFT calculations, together with their computational expense, motivate finding a cheaper, more direct map for electronic excitations. Here, we show that determining density and energy functionals via machine learning allows the equations of TDDFT to be bypassed. The framework we introduce is used to perform the first excited-state molecular dynamics simulations with a machine-learned functional on malonaldehyde and correctly capture the kinetics of its excited-state intramolecular proton transfer, allowing insight into how mechanical constraints can be used to control the proton transfer reaction in this molecule. This development opens the door to using machine-learned functionals for highly efficient excited-state dynamics simulations.

    more » « less
  2. We reformulate the thermally assisted-occupation density functional theory (TAO-DFT) into the Kohn–Sham single-determinant framework and construct two new post-self-consistent field (post-SCF) static correlation correction schemes, named rTAO and rTAO-1. In contrast to the original TAO-DFT with the density in an ensemble form, in which each orbital density is weighted with a fractional occupation number, the ground-state density is given by a single-determinant wavefunction, a regular Kohn–Sham (KS) density, and total ground state energy is expressed in the normal KS form with a static correlation energy formulated in terms of the KS orbitals. In post-SCF calculations with rTAO functionals, an efficient energy scanning to quantitatively determine θ is also proposed. The rTAOs provide a promising method to simulate systems with strong static correlation as original TAO, but simpler and more efficient. We show that both rTAO and rTAO-1 is capable of reproducing most results from TAO-DFT without the additional functional Eθ used in TAO-DFT. Furthermore, our numerical results support that, without the functional Eθ, both rTAO and rTAO-1 can capture correct static correlation profiles in various systems. 
    more » « less
  3. We derive and implement analytic gradients and derivative couplings for time-dependent density functional theory plus one double (TDDFT-1D) which is a semiempirical configuration interaction method whereby the Hamiltonian is diagonalized in a basis of all singly excited configurations and one doubly excited configuration as constructed from a set of reference Kohn–Sham orbitals. We validate the implementation by comparing against finite difference values. Furthermore, we show that our implementation can locate both optimized geometries and minimum-energy crossing points along conical seams of S1/S0 surfaces for a set of test cases.

    more » « less
  4. Abstract

    The theorems of density functional theory (DFT) establish bijective maps between the local external potential of a many-body system and its electron density, wavefunction and, therefore, one-particle reduced density matrix. Building on this foundation, we show that machine learning models based on the one-electron reduced density matrix can be used to generate surrogate electronic structure methods. We generate surrogates of local and hybrid DFT, Hartree-Fock and full configuration interaction theories for systems ranging from small molecules such as water to more complex compounds like benzene and propanol. The surrogate models use the one-electron reduced density matrix as the central quantity to be learned. From the predicted density matrices, we show that either standard quantum chemistry or a second machine-learning model can be used to compute molecular observables, energies, and atomic forces. The surrogate models can generate essentially anything that a standard electronic structure method can, ranging from band gaps and Kohn-Sham orbitals to energy-conserving ab-initio molecular dynamics simulations and infrared spectra, which account for anharmonicity and thermal effects, without the need to employ computationally expensive algorithms such as self-consistent field theory. The algorithms are packaged in an efficient and easy to use Python code, QMLearn, accessible on popular platforms.

    more » « less
  5. Abstract

    Informed by an abstraction of Kohn–Sham (KS) computation called a KS machine, a functional analytic perspective is developed on mathematical aspects of density functional theory. A natural semantics for the machine is bivariate, consisting of a sequence of potentials paired with a ground density. Although the question of when the KS machine can converge to a solution (where the potential component matches a designated target) is not resolved here, a number of related ones are. For instance: can the machine progress toward a solution? Barring presumably exceptional circumstances, yes in an energetic sense, but using a potential-mixing scheme rather than the usual density-mixing variety. Are energetic and function space distance notions of proximity-to-solution commensurate? Yes, to a significant degree. If the potential components of a sequence of ground pairs converges to a target density, do the density components cluster on ground densities thereof? Yes, barring particle number drifting to infinity.

    more » « less