arXiv:2106.03126 · SciPost Physics Core

Predicting quantum potentials by deep neural network and metropolis sampling

Rui Hong, Peng-Fei Zhou, Bin Xi, Jie Hu, An-Chun Ji, Shi-Ju Ran

Published Sep 13, 2021• 3 citations

Abstract

The hybridizations of machine learning and quantum physics have caused essential impacts to the methodology in both fields. Inspired by quantum potential neural network, we here propose to solve the potential in the Schrödinger equation provided the eigenstate, by combining Metropolis sampling with deep neural network, which we dub as Metropolis potential neural network (MPNN). A loss function is proposed to explicitly involve the energy in the optimization for its accurate evaluation. Benchmarking on the harmonic oscillator and hydrogen atom, MPNN shows excellent accuracy and stability on predicting not just the potential to satisfy the Schrödinger equation, but also the eigen-energy. Our proposal could be potentially applied to the ab-initio simulations, and to inversely solving other partial differential equations in physics and beyond.

DOI: 10.21468/scipostphyscore.4.3.022

Related papers

Solving Schrodinger equations using a physically constrained neural network*

Kai-Fang Pu, Hanlin Li, Hongliang Lü +1 more

Abstract Deep neural networks (DNNs) and auto differentiation have been widely used in computational physics to solve variational problems. When a DNN is used to represent the wave function and solve quantum many-body problems using variational optimization, various physical constraints have to be injected into the neural network by construction to increase the data and learning efficiency. We build the unitary constraint to the variational wave function using a monotonic neural network to represent the cumulative distribution function (CDF) . Using this constrained neural network to represent the variational wave function, we solve Schrodinger equations using auto-differentiation and stochastic gradient descent (SGD) by minimizing the violation of the trial wave function to the Schrodinger equation. For several classical problems in quantum mechanics, we obtain their ground state wave function and energy with very low errors. The method developed in the present paper may pave a new way for solving nuclear many-body problems in the future.

arXiv:2303.03934Chinese Physics CMar 17, 20238 citations

SchrödingerNet: A Universal Neural Network Solver for the Schrödinger Equation

Yaolong Zhang, Bin Jiang, Hua Guo

Recent advances in machine learning have facilitated numerically accurate solution of the electronic Schrödinger equation (SE) by integrating various neural network (NN)-based wave function ansatzes with variational Monte Carlo methods. Nevertheless, such NN-based methods are all based on the Born-Oppenheimer approximation (BOA) and require computationally expensive training for each nuclear configuration. In this work, we propose a novel NN architecture, SchrödingerNet, to solve the full electronic-nuclear SE by defining a loss function designed to equalize local energies across the system. This approach is based on a translationally, rotationally and permutationally symmetry-adapted total wave function ansatz that includes both nuclear and electronic coordinates. This strategy not only allows for an efficient and accurate generation of a continuous potential energy surface at any geometry within the well-sampled nuclear configuration space, but also incorporates non-BOA corrections, through a single training process. Comparison with benchmarks of atomic and small molecular systems demonstrates its accuracy and efficiency.

arXiv:2408.04497Journal of Chemical Theory and ComputationJan 7, 20258 citations

Ab initio solution of the many-electron Schrödinger equation with deep neural networks

David Pfau, James S. Spencer, Alexander G. D. G. Matthews +1 more

Given access to accurate solutions of the many-electron Schrödinger equation, nearly all chemistry could be derived from first principles.Exact wave functions of interesting chemical systems are out of reach because they are NP-hard to compute in general, but approximations can be found using polynomially scaling algorithms.The key challenge for many of these algorithms is the choice of wave function approximation, or Ansatz, which must trade off between efficiency and accuracy.Neural networks have shown impressive power as accurate practical function approximators and promise as a compact wave-function Ansatz for spin systems, but problems in electronic structure require wave functions that obey Fermi-Dirac statistics.Here we introduce a novel deep learning architecture, the Fermionic neural network, as a powerful wave-function Ansatz for many-electron systems.The Fermionic neural network is able to achieve accuracy beyond other variational quantum Monte Carlo Ansatz on a variety of atoms and small molecules.Using no data other than atomic positions and charges, we predict the dissociation curves of the nitrogen molecule and hydrogen chain, two challenging strongly correlated systems, to significantly higher accuracy than the coupled cluster method, widely considered the most accurate scalable method for quantum chemistry at equilibrium geometry.This demonstrates that deep neural networks can improve the accuracy of variational quantum Monte Carlo to the point where it outperforms other ab initio quantum chemistry methods, opening the possibility of accurate direct optimization of wave functions for previously intractable many-electron systems.

arXiv:1909.02487Physical Review ResearchSep 16, 2020552 citations

Smart Machine Learning for Solving the Full Electron-Nuclear Schrödinger Equation with a Quantum-Chemically Inspired Wavefunction Ansatz

Yaolong Zhang, Bin Jiang, Hua Guo

Directly solving the full electron-nuclear Schrödinger equation remains one of the grand challenges in quantum mechanics. Here, we propose a quantum-chemically motivated orbital-based neural network wavefunction ansatz with explicit many-body electron-electron correlation terms, with parameters depending on the entire electron-nuclear configuration, enabling an accurate description of strongly correlated and/or multi-reference electronic structures by a single Slater-determinant across a broad nuclear configuration space. To train this wavefunction, we further develop an efficient hybrid optimization strategy that combines the variational Monte Carlo technique with local-energy constraints, mitigating the intrinsic sampling bias of the former and considerably reducing the statistical error. These advances are embedded in a unified neural network framework, Schrödinger, enabling the determination of globally accurate potential energy surfaces with a single training for systems ranging from strongly correlated diatomic molecules, triatomic reactions containing conical intersections, to polyatomic multi-reference molecules. This work offers a practical machine-learning tool for solving the electron-nuclear Schrödinger equation and capturing full quantum effects beyond the Born-Oppenheimer approximation.

physics.chem-phCross-listedarXiv:2609.40112Oct 1, 20260 citations

Solutions to the Schrödinger equation using deep neural networks for integrated photonics

Sourabh Kumar Dubey, Hibah Islahi, Raghvendra Singh

Abstract This paper introduces a novel method for solving the Schrödinger equation through the use of deep neural networks (DNNs), presenting a significant departure from traditional techniques. Conventional approaches to solving the Schrödinger equation, such as analytical methods and numerical algorithms, often face challenges when dealing with complex quantum systems due to their inherent limitations. These traditional methods can become cumbersome or even infeasible as the complexity of the systems increases. In contrast, our approach harnesses the capabilities of deep neural networks to approximate both the wavefunction and the energy eigenvalues of quantum systems. By leveraging the flexible and powerful nature of DNNs, we provide a new pathway to solving the Schrödinger equation that can potentially overcome the constraints of classical methods. To validate the effectiveness of our approach, we apply it to the particle in a box problem – a fundamental quantum mechanics model with well-established analytical solutions. This benchmark problem serves as a useful test case, allowing us to demonstrate that DNNs can not only replicate the known results accurately but also offer insights into how these networks can handle more intricate quantum systems. Our results reveal that DNNs are capable of accurately reproducing the analytical solutions for the particle in a box, illustrating their potential as a versatile tool for quantum mechanics.

Journal of Optical CommunicationsSep 30, 20240 citations

Approximating Ground State Energies and Wave Functions of Physical Systems with Neural Networks

Lema, Cesar, Anna Choromanska

Quantum theory has been remarkably successful in providing an understanding of physical systems at foundational scales. Solving the Schrödinger equation provides full knowledge of all dynamical quantities of the physical system. However closed form solutions to this equation are only available for a few systems and approximation methods are typically used to find solutions. In this paper we address the problem of solving the time independent Schrödinger equation for the ground state solution of physical systems. We propose using end-to-end deep learning approach in a variational optimization scheme for approximating the ground state energies and wave functions of these systems. A neural network realizes a universal trial wave function and is trained in an unsupervised learning framework by optimizing the expectation value of the Hamiltonian of a physical system. The proposed approach is evaluated on physical systems consisting of a particle in a box with and without a perturbation. We demonstrate that our approach obtains approximations of ground state energies and wave functions that are highly accurate, which makes it a potentially plausible candidate for solving more complex physical systems for which analytical solutions are beyond reach.

arXiv:2011.10694Nov 21, 20201 citation