arrow
Volume 32, Issue 4
A New Artificial Neural Network Method for Solving Schrödinger Equations on Unbounded Domains

Joshua P. Wilson, Weizhong Dai, Aniruddha Bora & Jacob C. Boyt

Commun. Comput. Phys., 32 (2022), pp. 1039-1060.

Published online: 2022-10

Export citation
  • Abstract

The simulation for particle or soliton propagation based on linear or nonlinear Schrödinger equations on unbounded domains requires the computational domain to be bounded, and therefore, a special boundary treatment such as an absorbing boundary condition (ABC) or a perfectly matched layer (PML) is needed so that the reflections of outgoing waves at the boundary can be minimized in order to prevent the destruction of the simulation. This article presents a new artificial neural network (ANN) method for solving linear and nonlinear Schrödinger equations on unbounded domains. In particular, this method randomly selects training points only from the bounded computational space-time domain, and the loss function involves only the initial condition and the Schrödinger equation itself in the computational domain without any boundary conditions. Moreover, unlike standard ANN methods that calculate gradients using expensive automatic differentiation, this method uses accurate finite-difference approximations for the physical gradients in the Schrödinger equation. In addition, a Metropolis-Hastings algorithm is implemented for preferentially selecting regions of high loss in the computational domain allowing for the use of fewer training points in each batch. As such, the present training method uses fewer training points and less computation time for convergence of the loss function as compared with the standard ANN methods. This new ANN method is illustrated using three examples.

  • AMS Subject Headings

65N06, 65N12

  • Copyright

COPYRIGHT: © Global Science Press

  • Email address
  • BibTex
  • RIS
  • TXT
@Article{CiCP-32-1039, author = {Wilson , Joshua P.Dai , WeizhongBora , Aniruddha and Boyt , Jacob C.}, title = {A New Artificial Neural Network Method for Solving Schrödinger Equations on Unbounded Domains}, journal = {Communications in Computational Physics}, year = {2022}, volume = {32}, number = {4}, pages = {1039--1060}, abstract = {

The simulation for particle or soliton propagation based on linear or nonlinear Schrödinger equations on unbounded domains requires the computational domain to be bounded, and therefore, a special boundary treatment such as an absorbing boundary condition (ABC) or a perfectly matched layer (PML) is needed so that the reflections of outgoing waves at the boundary can be minimized in order to prevent the destruction of the simulation. This article presents a new artificial neural network (ANN) method for solving linear and nonlinear Schrödinger equations on unbounded domains. In particular, this method randomly selects training points only from the bounded computational space-time domain, and the loss function involves only the initial condition and the Schrödinger equation itself in the computational domain without any boundary conditions. Moreover, unlike standard ANN methods that calculate gradients using expensive automatic differentiation, this method uses accurate finite-difference approximations for the physical gradients in the Schrödinger equation. In addition, a Metropolis-Hastings algorithm is implemented for preferentially selecting regions of high loss in the computational domain allowing for the use of fewer training points in each batch. As such, the present training method uses fewer training points and less computation time for convergence of the loss function as compared with the standard ANN methods. This new ANN method is illustrated using three examples.

}, issn = {1991-7120}, doi = {https://doi.org/10.4208/cicp.OA-2022-0135}, url = {http://global-sci.org/intro/article_detail/cicp/21138.html} }
TY - JOUR T1 - A New Artificial Neural Network Method for Solving Schrödinger Equations on Unbounded Domains AU - Wilson , Joshua P. AU - Dai , Weizhong AU - Bora , Aniruddha AU - Boyt , Jacob C. JO - Communications in Computational Physics VL - 4 SP - 1039 EP - 1060 PY - 2022 DA - 2022/10 SN - 32 DO - http://doi.org/10.4208/cicp.OA-2022-0135 UR - https://global-sci.org/intro/article_detail/cicp/21138.html KW - Linear and nonlinear Schrödinger equations, artificial neural network method, convergence, soliton and particle propagations. AB -

The simulation for particle or soliton propagation based on linear or nonlinear Schrödinger equations on unbounded domains requires the computational domain to be bounded, and therefore, a special boundary treatment such as an absorbing boundary condition (ABC) or a perfectly matched layer (PML) is needed so that the reflections of outgoing waves at the boundary can be minimized in order to prevent the destruction of the simulation. This article presents a new artificial neural network (ANN) method for solving linear and nonlinear Schrödinger equations on unbounded domains. In particular, this method randomly selects training points only from the bounded computational space-time domain, and the loss function involves only the initial condition and the Schrödinger equation itself in the computational domain without any boundary conditions. Moreover, unlike standard ANN methods that calculate gradients using expensive automatic differentiation, this method uses accurate finite-difference approximations for the physical gradients in the Schrödinger equation. In addition, a Metropolis-Hastings algorithm is implemented for preferentially selecting regions of high loss in the computational domain allowing for the use of fewer training points in each batch. As such, the present training method uses fewer training points and less computation time for convergence of the loss function as compared with the standard ANN methods. This new ANN method is illustrated using three examples.

Joshua P. Wilson, Weizhong Dai, Aniruddha Bora & Jacob C. Boyt. (2022). A New Artificial Neural Network Method for Solving Schrödinger Equations on Unbounded Domains. Communications in Computational Physics. 32 (4). 1039-1060. doi:10.4208/cicp.OA-2022-0135
Copy to clipboard
The citation has been copied to your clipboard