D4FT: A Deep Learning Approach to Kohn-Sham Density Functional Theory

AI-generated keywords: Kohn-Sham Density Functional Theory

AI-generated Key Points

  • Traditional method for solving KS-DFT equations is through the SCF approach
  • Proposed deep learning approach to KS-DFT
  • Reparameterization of orthogonal constraint as a feed-forward computation
  • Same expressivity as SCF method but reduces computational complexity from O(N^4) to O(N^3)
  • Incorporates numerical integration by amortizing it into optimization steps
  • Stochastic gradient descent with sampled minibatch of quadrature grids improves efficiency and stability
  • Extensive experiments demonstrate advantages in efficiency and stability
  • Enables exploration of more complex neural-based wave functions
  • Paper accepted by ICLR 2023 and recognized as one of the top 25% papers
  • Authors: Tianbo Li, Min Lin, Zheyuan Hu, Kunhao Zheng, Giovanni Vignale, Kenji Kawaguchi, A.H. Castro Neto, Kostya S. Novoselov, Shuicheng Yan
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Tianbo Li, Min Lin, Zheyuan Hu, Kunhao Zheng, Giovanni Vignale, Kenji Kawaguchi, A. H. Castro Neto, Kostya S. Novoselov, Shuicheng Yan

Accepted by The Eleventh International Conference on Learning Representations (ICLR 2023, notable-top-25%)
License: CC BY 4.0

Abstract: Kohn-Sham Density Functional Theory (KS-DFT) has been traditionally solved by the Self-Consistent Field (SCF) method. Behind the SCF loop is the physics intuition of solving a system of non-interactive single-electron wave functions under an effective potential. In this work, we propose a deep learning approach to KS-DFT. First, in contrast to the conventional SCF loop, we propose to directly minimize the total energy by reparameterizing the orthogonal constraint as a feed-forward computation. We prove that such an approach has the same expressivity as the SCF method, yet reduces the computational complexity from O(N^4) to O(N^3). Second, the numerical integration which involves a summation over the quadrature grids can be amortized to the optimization steps. At each step, stochastic gradient descent (SGD) is performed with a sampled minibatch of the grids. Extensive experiments are carried out to demonstrate the advantage of our approach in terms of efficiency and stability. In addition, we show that our approach enables us to explore more complex neural-based wave functions.

Submitted to arXiv on 01 Mar. 2023

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2303.00399v1

In the field of Kohn-Sham Density Functional Theory (KS-DFT), the traditional method for solving the equations is through the Self-Consistent Field (SCF) approach. This involves iteratively solving a system of non-interactive single-electron wave functions under an effective potential. However, in this work, a deep learning approach to KS-DFT is proposed. The first key contribution of this work is a reparameterization of the orthogonal constraint as a feed-forward computation, which allows for direct minimization of the total energy. It has been proven that this approach has the same expressivity as the SCF method but significantly reduces computational complexity from O(N^4) to O(N^3), where N represents the number of atomic orbitals. Additionally, this deep learning approach incorporates numerical integration by amortizing it into optimization steps. Stochastic gradient descent (SGD) is performed with a sampled minibatch of quadrature grids at each step, improving both efficiency and stability. Extensive experiments have been conducted to demonstrate the advantages of this approach in terms of both efficiency and stability. Furthermore, it has been shown that this deep learning approach enables exploration of more complex neural-based wave functions. The paper has been accepted by The Eleventh International Conference on Learning Representations (ICLR 2023) and has received notable recognition as one of the top 25% papers. The authors include Tianbo Li, Min Lin, Zheyuan Hu, Kunhao Zheng, Giovanni Vignale, Kenji Kawaguchi, A.H. Castro Neto, Kostya S. Novoselov, and Shuicheng Yan. Overall, this work presents a novel deep learning approach to KS-DFT that offers improved computational efficiency and stability while allowing for exploration of more complex wave functions based on neural networks.
Created on 08 Feb. 2024

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.