Research reports

Multi-Scale Message Passing Neural PDE Solvers

by L. Equer and T. K. Rusch and S. Mishra

(Report number 2023-14)

Abstract
We propose a novel multi-scale message passing neural network algorithm for learning the solutions of time-dependent PDEs. Our algorithm possesses both temporal and spatial multi-scale resolution features by incorporating multi-scale sequence models and graph gating modules in the encoder and processor, respectively. Benchmark numerical experiments are presented to demonstrate that the proposed algorithm outperforms baselines, particularly on a PDE with a range of spatial and temporal scales.

Keywords: Graph neural network (GNN), Partial Differential Equation (PDE), Message-Passing, Multi-scale, Long Expressive Memory (LEM), Gradient Gating (G^2)

BibTeX
@Techreport{ERM23_1051,
  author = {L. Equer and T. K. Rusch and S. Mishra},
  title = {Multi-Scale Message Passing Neural PDE Solvers},
  institution = {Seminar for Applied Mathematics, ETH Z{\"u}rich},
  number = {2023-14},
  address = {Switzerland},
  url = {https://www.sam.math.ethz.ch/sam_reports/reports_final/reports2023/2023-14.pdf },
  year = {2023}
}

Disclaimer
© Copyright for documents on this server remains with the authors. Copies of these documents made by electronic or mechanical means including information storage and retrieval systems, may only be employed for personal use. The administrators respectfully request that authors inform them when any paper is published to avoid copyright infringement. Note that unauthorised copying of copyright material is illegal and may lead to prosecution. Neither the administrators nor the Seminar for Applied Mathematics (SAM) accept any liability in this respect. The most recent version of a SAM report may differ in formatting and style from published journal version. Do reference the published version if possible (see SAM Publications).

JavaScript has been disabled in your browser