Solving Schrodinger equations using a physically constrained neural network* *Supported by the National Natural Science Foundation of China (12035006, 12075098), the Natural Science Foundation of Hubei Province (2019CFB563), the Hubei Province Department of Education (D20201108), Hubei Province Department of Science and Technology (2021BLB171)

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.