Wasserstein Quantum Monte Carlo: A Novel Approach for Solving the Quantum Many-Body Schrödinger Equation

Kirill Neklyudov,Jannes Nys,Luca Thiede,Juan Carrasquilla,Qiang Liu,Max Welling,Alireza Makhzani
2023-10-26
Abstract:Solving the quantum many-body Schrödinger equation is a fundamental and challenging problem in the fields of quantum physics, quantum chemistry, and material sciences. One of the common computational approaches to this problem is Quantum Variational Monte Carlo (QVMC), in which ground-state solutions are obtained by minimizing the energy of the system within a restricted family of parameterized wave functions. Deep learning methods partially address the limitations of traditional QVMC by representing a rich family of wave functions in terms of neural networks. However, the optimization objective in QVMC remains notoriously hard to minimize and requires second-order optimization methods such as natural gradient. In this paper, we first reformulate energy functional minimization in the space of Born distributions corresponding to particle-permutation (anti-)symmetric wave functions, rather than the space of wave functions. We then interpret QVMC as the Fisher-Rao gradient flow in this distributional space, followed by a projection step onto the variational manifold. This perspective provides us with a principled framework to derive new QMC algorithms, by endowing the distributional space with better metrics, and following the projected gradient flow induced by those metrics. More specifically, we propose "Wasserstein Quantum Monte Carlo" (WQMC), which uses the gradient flow induced by the Wasserstein metric, rather than Fisher-Rao metric, and corresponds to transporting the probability mass, rather than teleporting it. We demonstrate empirically that the dynamics of WQMC results in faster convergence to the ground state of molecular systems.
Computational Physics,Machine Learning,Chemical Physics
What problem does this paper attempt to address?
The paper aims to address the problem of solving the quantum many-body Schrödinger equation, which is a fundamental and challenging issue in the fields of quantum physics, quantum chemistry, and materials science. Specifically, the paper focuses on how to more efficiently find the ground state wave function of quantum many-body systems. To solve this problem, the paper proposes a new method—Wasserstein Quantum Monte Carlo (WQMC), which uses the Wasserstein metric instead of the traditional Fisher–Rao metric for gradient flow optimization of energy functional minimization. This method can avoid the "teleportation" phenomenon of probability mass that occurs in traditional methods. Experimental validation shows that WQMC has a significant advantage in convergence speed compared to traditional Quantum Variational Monte Carlo (QVMC) methods, especially when dealing with chemical systems, allowing for faster convergence to the ground state energy. The experimental results indicate that for different chemical systems (such as beryllium atom, boron atom, lithium molecule, and hydrogen chain), both WQMC and W(FR)QMC exhibit faster convergence speeds and higher accuracy.