Deep Stochastic Mechanics

Elena Orlova,Aleksei Ustimenko,Ruoxi Jiang,Peter Y. Lu,Rebecca Willett
2024-07-08
Abstract:This paper introduces a novel deep-learning-based approach for numerical simulation of a time-evolving Schrödinger equation inspired by stochastic mechanics and generative diffusion models. Unlike existing approaches, which exhibit computational complexity that scales exponentially in the problem dimension, our method allows us to adapt to the latent low-dimensional structure of the wave function by sampling from the Markovian diffusion. Depending on the latent dimension, our method may have far lower computational complexity in higher dimensions. Moreover, we propose novel equations for stochastic quantum mechanics, resulting in quadratic computational complexity with respect to the number of dimensions. Numerical simulations verify our theoretical findings and show a significant advantage of our method compared to other deep-learning-based approaches used for quantum mechanics.
Machine Learning,Quantum Physics
What problem does this paper attempt to address?
The paper attempts to address the problem of the "curse of dimensionality" encountered in numerically solving the Schrödinger equation in high-dimensional cases. Traditional numerical methods for solving high-dimensional problems exhibit computational complexity that grows exponentially with the increase in problem dimensions, making it difficult to handle high-dimensional Schrödinger equations in practical applications such as multi-electron systems, drug discovery, and condensed matter physics. To overcome this challenge, the authors propose a new method based on deep learning—Deep Stochastic Mechanics (DSM). This method introduces the concept of stochastic mechanics and generative diffusion models, enabling effective solutions to the Schrödinger equation in high-dimensional cases. Specifically, the DSM method adapts to the potential low-dimensional structure of the wave function by sampling the Markov diffusion process, thereby significantly reducing computational complexity. Compared to existing deep learning methods, DSM has lower computational complexity for high-dimensional problems and demonstrates significant advantages in numerical simulations.