Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physics
Including prior knowledge is important for effective machine learning models in physics, and is usually achieved by explicitly adding loss terms or constraints on model architectures. Prior knowledge embedded in the physics computation itself rarely draws attention. We show that solving the Kohn-Sham equations when training neural networks for the exchange-correlation functional provides an implicit regularization that greatly improves generalization. Two separations suffice for learning the entire one-dimensional H$_2$ dissociation curve within chemical accuracy, including the strongly correlated region. Our models also generalize to unseen types of molecules and overcome self-interaction error.
Code (1)
Tasks
BIG-bench Machine LearningSimilar Papers 제목 키워드 기반
How Well Does Kohn-Sham Regularizer Work for Weakly Correlated Systems?
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 te…
BIG-bench Machine LearningNeuralSCF: Neural network self-consistent fields for density functional theory
Kohn-Sham density functional theory (KS-DFT) has found widespread application in accurate electronic structure calculations. However, it can be computationally demanding especially for large-scale simulations, motivating…
Zero-shot GeneralizationBy-passing the Kohn-Sham equations with machine learning
Last year, at least 30,000 scientific papers used the Kohn-Sham scheme of density functional theory to solve electronic structure problems in a wide variety of scientific fields, ranging from materials science to biochem…
BIG-bench Machine LearningTowards Combinatorial Generalization for Catalysts: A Kohn-Sham Charge-Density Approach
The Kohn-Sham equations underlie many important applications such as the discovery of new catalysts. Recent machine learning work on catalyst modeling has focused on prediction of the energy, but has so far not yet demon…
Out-of-Distribution GeneralizationEnhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systems
Density Functional Theory (DFT) is a pivotal method within quantum chemistry and materials science, with its core involving the construction and solution of the Kohn-Sham Hamiltonian. Despite its importance, the applicat…