arXiv · 2405.20287
Flexible SE(2) graph neural networks with applications to PDE surrogates
Abstract
This paper presents a novel approach for constructing graph neural networks equivariant to 2D rotations and translations and leveraging them as PDE surrogates on non-gridded domains. We show that aligning the representations with the principal axis allows us to sidestep many constraints while preserving SE(2) equivariance. By applying our model as a surrogate for fluid flow simulations and conducting thorough benchmarks against non-equivariant models, we demonstrate significant gains in terms of both data efficiency and accuracy.
Explore related subjects
Keep this discovery
Maria Bånkestad, Olof Mogren, Aleksis Pirinen. 2024-05-30. Flexible SE(2) graph neural networks with applications to PDE surrogates. https://arxiv.org/abs/2405.20287
Cite the original work for its findings. Save a collection to share your selection of sources.