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 (6)

• Hidden Physics Models: Machine Learning of Nonlinear Partial Differential Equations
• Restricted-Boltzmann-Machine Learning for Solving Strongly Correlated Quantum Systems
• Self-Learning Monte Carlo Method
• Accelerate Monte Carlo Simulations with Restricted Boltzmann Machines
• Machine learning vortices at the Kosterlitz-Thouless transition
• Machine learning technique to find quantum many-body ground states of bosons on a lattice