Journals
Resources
About Us
Open Access

Time Integration Schemes Based on Neural Networks for Solving Partial Differential Equations on Coarse Grids

Time Integration Schemes Based on Neural Networks for Solving Partial Differential Equations on Coarse Grids

Year:    2024

Author:    Xinxin Yan, Zhideng Zhou, Xiaohan Cheng, Xiaolei Yang

Communications in Computational Physics, Vol. 36 (2024), Iss. 5 : pp. 1262–1306

Abstract

In this work, we propose to learn time integration schemes based on neural networks which satisfy three distinct sets of mathematical constraints, i.e., unconstrained, semi-constrained with the root condition, and fully-constrained with both root and consistency conditions. We focus on the learning of 3-step linear multistep methods, which we subsequently applied to solve three model PDEs, i.e., the heat equation, the wave equation, and the Burgers’ equation. The results show that the prediction error of the learned fully-constrained scheme is close to that of the Runge-Kutta method and Adams-Bashforth method. Compared to the traditional methods, the learned unconstrained and semi-constrained schemes significantly reduce the prediction error on coarse grids, with an overall better performance for the semi-constrained model. On a grid $4×$ coarser than the reference, the mean square error (MSE) shows a reduction of up to two order of magnitude for some heat equation cases, and a substantial improvement in phase prediction for the wave equation. On a $32×$ coarser grid, the MSE for the Burgers’ equation can be reduced by up to 40% to 45%. Tests on the two-dimensional heat equation and wave equation show a good generalization capability for the constant optimized coefficients.

You do not have full access to this article.

Already a Subscriber? Sign in as an individual or via your institution

Journal Article Details

Publisher Name:    Global Science Press

Language:    English

DOI:    https://doi.org/10.4208/cicp.OA-2023-0266

Communications in Computational Physics, Vol. 36 (2024), Iss. 5 : pp. 1262–1306

Published online:    2024-01

AMS Subject Headings:    Global Science Press

Copyright:    COPYRIGHT: © Global Science Press

Pages:    45

Keywords:    Time integration scheme partial differential equation numerical simulation on coarse grids neural networks mathematical constraints.

Author Details

Xinxin Yan

Zhideng Zhou

Xiaohan Cheng

Xiaolei Yang