We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations

DBLP - CS Bibliography


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Learning the Solution Operator of Boundary Value Problems using Graph Neural Networks

Abstract: As an alternative to classical numerical solvers for partial differential equations (PDEs) subject to boundary value constraints, there has been a surge of interest in investigating neural networks that can solve such problems efficiently. In this work, we design a general solution operator for two different time-independent PDEs using graph neural networks (GNNs) and spectral graph convolutions. We train the networks on simulated data from a finite elements solver on a variety of shapes and inhomogeneities. In contrast to previous works, we focus on the ability of the trained operator to generalize to previously unseen scenarios. Specifically, we test generalization to meshes with different shapes and superposition of solutions for a different number of inhomogeneities. We find that training on a diverse dataset with lots of variation in the finite element meshes is a key ingredient for achieving good generalization results in all cases. With this, we believe that GNNs can be used to learn solution operators that generalize over a range of properties and produce solutions much faster than a generic solver. Our dataset, which we make publicly available, can be used and extended to verify the robustness of these models under varying conditions.
Subjects: Machine Learning (cs.LG); Numerical Analysis (math.NA)
Journal reference: 2nd AI4Science Workshop at the 39th International Conference on Machine Learning (ICML), 2022
Cite as: arXiv:2206.14092 [cs.LG]
  (or arXiv:2206.14092v1 [cs.LG] for this version)

Submission history

From: Winfried Lötzsch [view email]
[v1] Tue, 28 Jun 2022 15:39:06 GMT (1176kb,D)

Link back to: arXiv, form interface, contact.