REMuS-GNN: A rotation-equivariant model for simulating continuum dynamics
File(s)2205.07852v1.pdf (1.14 MB)
Accepted version
Author(s)
Lino, M
Cantwell, C
Fotiadis, S
Bharath, AA
Type
Conference Paper
Abstract
Numerical simulation is an essential tool in many areas of science and engineering, but its performance often limits application in practice or when used to explore large parameter spaces. On the other hand, surrogate deep learning models, while accelerating simulations, often exhibit poor accuracy and ability to generalise. In order to improve these two factors, we introduce REMuS-GNN, a rotation-equivariant multi-scale model for simulating continuum dynamical systems encompassing a range of length scales. REMuS-GNN is designed to predict an output vector field from an input vector field on a physical domain discretised into an unstructured set of nodes. Equivariance to rotations of the domain is a desirable inductive bias that allows the network to learn the underlying physics more efficiently, leading to improved accuracy and generalisation compared with similar architectures that lack such symmetry. We demonstrate and evaluate this method on the incompressible flow around elliptical cylinders.
Date Issued
2022-07-22
Date Acceptance
2022-07-01
Citation
Proceedings of Machine Learning Research, 2022, 196, pp.226-236
Publisher
ML Research Press
Start Page
226
End Page
236
Journal / Book Title
Proceedings of Machine Learning Research
Volume
196
Copyright Statement
© 2022 The Author(s).
Identifier
http://arxiv.org/abs/2205.07852v1
Source
Algebraic and Geometric Learning Workshops 2022
Subjects
cs.LG
cs.LG
physics.flu-dyn
Publication Status
Published
Start Date
2022-07-22
Finish Date
2022-07-25
Coverage Spatial
Virtual
Date Publish Online
2022-03-02