-
Comparison between tensor methods and neural networks in electronic structure calculations
Authors:
Mathias Dus,
Geneviève Dusson,
Virginie Ehrlacher,
Clément Guillot,
Joel Pascal Soffo Wambo
Abstract:
This article compares the tensor method density matrix renormalization group (DMRG) with two neural network based methods -namely FermiNet and PauliNet) for determining the ground state wavefunction of the many-body electronic Schr{ö}dinger problem. We provide numerical simulations illustrating the main features of the methods and showing convergence with respect to some parameter, such as the ra…
▽ More
This article compares the tensor method density matrix renormalization group (DMRG) with two neural network based methods -namely FermiNet and PauliNet) for determining the ground state wavefunction of the many-body electronic Schr{ö}dinger problem. We provide numerical simulations illustrating the main features of the methods and showing convergence with respect to some parameter, such as the rank for DMRG, and number of pretraining iterations for neural networks. We then compare the obtained energy with the methods for a few atoms and molecules, for some of which the exact value of the energy is known for the sake of comparison. In the last part of the article, we propose a new kind of neural network to solve the Schr{ö}dinger problem based on the training of the wavefunction on a simplex, and an explicit permutation for evaluating the wavefunction on the whole space. We provide numerical results on a toy problem for the sake of illustration.
△ Less
Submitted 4 December, 2024;
originally announced December 2024.
-
Two-layers neural networks for Schr{ö}dinger eigenvalue problems
Authors:
Mathias Dus,
Ehrlacher Virginie
Abstract:
The aim of this article is to analyze numerical schemes using two-layer neural networks withinfinite width for the resolution of high-dimensional Schr{ö}dinger eigenvalue problems with smoothinteraction potentials and Neumann boundary condition on the unit cube in any dimension. Moreprecisely, any eigenfunction associated to the lowest eigenvalue of the Schr{ö}dinger operator is a unitL 2 norm min…
▽ More
The aim of this article is to analyze numerical schemes using two-layer neural networks withinfinite width for the resolution of high-dimensional Schr{ö}dinger eigenvalue problems with smoothinteraction potentials and Neumann boundary condition on the unit cube in any dimension. Moreprecisely, any eigenfunction associated to the lowest eigenvalue of the Schr{ö}dinger operator is a unitL 2 norm minimizer of the associated energy. Using Barron's representation of the solution witha probability measure defined on the set of parameter values and following the approach initiallysuggested by Bach and Chizat [1], the energy is minimized thanks to a constrained gradient curvedynamic on the 2-Wasserstein space of the set of parameter values defining the neural network. Weprove the existence of solutions to this constrained gradient curve. Furthermore, we prove that,if it converges, the represented function is then an eigenfunction of the considered Schr{ö}dingeroperator. At least up to our knowledge, this is the first work where this type of analysis is carriedout to deal with the minimization of non-convex functionals.
△ Less
Submitted 3 September, 2024;
originally announced September 2024.
-
Numerical solution of Poisson partial differential equations in high dimension using two-layer neural networks
Authors:
Mathias Dus,
Virginie Ehrlacher
Abstract:
The aim of this article is to analyze numerical schemes using two-layer neural networks with infinite width for the resolution of the high-dimensional Poisson-Neumann partial differential equations (PDEs) with Neumann boundary conditions. Using Barron's representation of the solution with a measure of probability, the energy is minimized thanks to a gradient curve dynamic on the $2$ Wasserstein sp…
▽ More
The aim of this article is to analyze numerical schemes using two-layer neural networks with infinite width for the resolution of the high-dimensional Poisson-Neumann partial differential equations (PDEs) with Neumann boundary conditions. Using Barron's representation of the solution with a measure of probability, the energy is minimized thanks to a gradient curve dynamic on the $2$ Wasserstein space of parameters defining the neural network. Inspired by the work from Bach and Chizat, we prove that if the gradient curve converges, then the represented function is the solution of the elliptic equation considered. Numerical experiments are given to show the potential of the method.
△ Less
Submitted 13 July, 2023; v1 submitted 16 May, 2023;
originally announced May 2023.
