Subspace method based on neural networks for eigenvalue problems

Xiaoying Dai, Yunying Fan, Zhiqiang Sheng

Published 2024-10-17Version 1

With the rapid development of machine learning, numerical discretization methods based on deep neural networks have been widely used in many fields, especially in solving high-dimensional problems where traditional methods face bottlenecks. However, for low-dimensional problems, existing machine learning methods are not as accurate and efficient as traditional methods. In this paper, we propose a subspace method based on neural networks for eigenvalue problems with high accuracy and low cost. Our basic idea is to use neural network based basis functions to span a subspace, then calculate the parameters of the neural network based basis functions through appropriate training, and finally calculate the Galerkin projection of the eigenvalue problem onto the subspace and obtain an approximate solution. In addition, we reduce the dimension of the subspace by applying some dimensionality reduction technique, which can improve the accuracy of the approximate solution in further. Numerical experiments show that we can obtain approximate eigenvalues with accuracy of $10^{-11}$ but with less then 400 epochs, which is significantly superior to other existing neural network based methods.