arXiv:2209.12572

Solving Schrödinger Equation Using Tensor Neural Network

Yangfei Liao, Lin, Zhongshuo, Liu, Jianghao, Sun, Qingyuan, Yifan Wang, Wu, Teng, Hehu Xie, He, Mingfeng

Published Sep 26, 2022• 0 citations

Abstract

In this paper, we introduce a novel approach to solve the many-body Schrodinger equation by the tensor neural network. Based on the tensor product structure, we can do the direct numerical integration by using fixed quadrature points for the functions constructed by the tensor neural network within tolerable computational complexity. Especially, we design several types of efficient numerical methods to treat the variable-coupled Coulomb potentials with high accuracy. The corresponding machine learning method is built for solving many-body Schrodinger equation. Some numerical examples are provided to validate the accuracy and efficiency of the proposed algorithms.

DOI: 10.48550/arxiv.2209.12572

Related papers

Functional tensor network solving many-body Schrödinger equation

Rui Hong, Ya-Xuan Xiao, Jie Hu +2 more

Solving the many-body Schr\"odinger equation in continuous spaces with the presence of strong correlations is an extremely important and challenging issue in quantum physics. In this work, we propose the functional tensor network (FTN) approach to solve the many-body Schr\"odinger equation. Provided the orthonormal functional bases, we represent the coefficients of the many-body wave function as a tensor network. The observables, such as energy, can be calculated simply by tensor contractions. Simulating the ground state becomes solving a minimization problem defined by the tensor network. An efficient gradient-decent algorithm based on automatically differentiable tensors is proposed. We here take the matrix product state (MPS), whose complexity scales only linearly with the system size, as an example. We apply our approach to solve the ground state of coupled harmonic oscillators and achieve high accuracy by comparing our results with the exact solutions. Reliable results are also given in the presence of three-body interactions, where the system cannot be decoupled from isolated oscillators. Our approach is simple and, with well-controlled error, essentially different from highly nonlinear neural-network solvers. Our work extends the applications of the tensor network from quantum lattice models to systems in continuous space. The FTN can be used as a general solver of the differential equations with many variables. The MPS exemplified here can be generalized to, e.g., fermionic tensor networks to solve the electronic Schr\"odinger equation.

arXiv:2201.12823Physical review. B./Physical review. BApr 12, 202212 citations

Solving the Electronic Schrödinger Equation by Pairing Tensor-Network State with Neural Network Quantum State

Bowen Kan, Yingqi Tian, Daiyou Xie +3 more

Neural network methods have shown promise for solving complex quantum many-body systems. In this study, we develop a novel approach through incorporating the density-matrix renormalization group (DMRG) method with the neural network quantum state method. The results demonstrate that, when tensor-network pre-training is introduced into the neural network, a high efficiency can be achieved for quantum many-body systems with strong correlations.

MathematicsJan 29, 20245 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

Neural-network-based multistate solver for a static Schrödinger equation

Hong Li, Qilong Zhai, Jeff Z. Y. Chen

Solving a multivariable static Schr\"odinger equation for a quantum system, to produce multiple excited-state energy eigenvalues and wave functions, is one of the basic tasks in mathematical and computational physics. Here we propose a neural-network-based solver, which enables us to cover the high-dimensional variable space for this purpose. The efficiency of the solver is analyzed by examples aimed at demonstrating the concept and various aspects of the task: the simultaneous finding of multiple excited states of lowest energies, the computation of energy-degenerate states with orthogonalized wave functions, the scalability to handle a multivariable problem, and the self-consistent determination and automatic adjustment of the imbedded Monte Carlo procedure. The solver adheres to the computational techniques developed in machine learning and is vastly different from traditional numerical methods.

Physical Review AMar 8, 202131 citations

Quantum-inspired tensor networks in machine learning models

Guillermo Valverde, Igor García-Olaizola, Giannicola Scarpa +1 more

Tensor networks were developed in the context of many-body physics as compressed representations of multiparticle quantum states. These representations mitigate the exponential complexity of many-body systems by capturing only the most relevant dependencies. Due to the formal similarity between quantum entanglement and statistical correlations, tensor networks have recently been integrated in machine learning, operating both as alternative learning architectures and as decompositions of components of neural networks. The expectation is that the theoretical understanding of tensor networks developed within quantum many-body physics leads to novel methods that offer advantages in terms of computational efficiency, explainability, or privacy. Here we review the use of tensor networks in the context of machine learning, providing a critical assessment of the state of the art, the potential advantages, and the challenges that must be overcome.

cs.LGCross-listedarXiv:2604.14287Apr 17, 20260 citations

Spectral convergence of sum-of-Gaussians tensor neural networks for many-electron Schrödinger equation

Teng Wu, Qi Zhou, Huangjie Zheng +2 more

An improved version of the sum-of-Gaussians tensor neural network (SOG-TNN) architecture is presented for solving the many-electron Schrödinger equation for one-dimensional soft-Coulomb systems. Model reduction techniques are introduced to reduce the number of tensor-factorized bases under the SOG approximation of the kernel. The Slater determinant ansatz is employed so that the antisymmetric property of the wave function can be strictly preserved. Numerical results show that the SOG-TNN achieves high accuracy with remarkably small basis sizes. Robust spectral convergence with respect to the basis size is also observed, consistently characterized by a mixed algebraic-exponential model for the error decay. These findings validate that the SOG-TNN architecture provides an ultra-efficient and low-rank representation of complex multi-electron wave functions, shedding light on high-fidelity quantum calculations in larger-scale many-electron systems.

The Journal of Chemical PhysicsJun 22, 20260 citations