Weakly-supervised learning on Schrodinger equation

Abstract

We propose a machine learning method to solve Schrodinger equations for a Hamiltonian that consists of an unperturbed Hamiltonian and a perturbation. We focus on the cases where the unperturbed Hamiltonian can be solved analytically or solved numerically with some fast way. Given a potential function as input, our deep learning model predicts wave functions and energies using a weakly-supervised method. Information of first-order perturbation calculation for randomly chosen perturbations is used to train the model. In other words, no label (or exact solution) is necessary for the training, which is why the method is called weakly-supervised, not supervised. The trained model can be applied to calculation of wave functions and energies of Hamiltonian containing arbitrary perturbation. As an example, we calculated wave functions and energies of a harmonic oscillator with a perturbation and results were in good agreement with those obtained from exact diagonalization.

References in corpus (18)

• The density-matrix renormalization group
• DGM: A deep learning algorithm for solving partial differential equations
• Machine learning phases of matter
• Hidden Physics Models: Machine Learning of Nonlinear Partial Differential Equations
• Discovering Phase Transitions with Unsupervised Learning
• Neural Ordinary Differential Equations
• Unsupervised learning of phase transitions: from principal component analysis to variational autoencoders
• Restricted-Boltzmann-Machine Learning for Solving Strongly Correlated Quantum Systems
• Deep autoregressive models for the efficient variational simulation of many-body quantum systems
• Self-Learning Monte Carlo Method
• Accelerate Monte Carlo Simulations with Restricted Boltzmann Machines
• Solving Many-Electron Schrödinger Equation Using Deep Neural Networks
• Machine learning vortices at the Kosterlitz-Thouless transition
• Deep learning and the Schrödinger equation
• Neural Network Renormalization Group
• Machine learning technique to find quantum many-body ground states of bosons on a lattice
• Disordered ultracold atomic gases in optical lattices: A case study of Fermi-Bose mixtures
• Deep Neural Network Detects Quantum Phase Transition