Practical Implications of Equivariant and Invariant Graph Neural Networks for Fluid Flow Modeling
Abstract
Graph neural networks (GNNs) have shown promise in learning unstructured mesh-based simulations of physical systems, including fluid dynamics. In tandem, geometric deep learning principles have informed the development of equivariant architectures. However, the practical implications of rotational equivariance in modeling fluids remains under-explored. We build a multi-scale equivariant GNN to forecast buoyancy-driven shear fluid flow and study the effect of modeling invariant and non-invariant representations of the flow state. Our results show that modeling invariant quantities produces more accurate long-term predictions and that these invariant quantities may be learned from the velocity field using a data-driven encoder.
Cite
Text
Shankar et al. "Practical Implications of Equivariant and Invariant Graph Neural Networks for Fluid Flow Modeling." ICLR 2023 Workshops: Physics4ML, 2023.Markdown
[Shankar et al. "Practical Implications of Equivariant and Invariant Graph Neural Networks for Fluid Flow Modeling." ICLR 2023 Workshops: Physics4ML, 2023.](https://mlanthology.org/iclrw/2023/shankar2023iclrw-practical/)BibTeX
@inproceedings{shankar2023iclrw-practical,
title = {{Practical Implications of Equivariant and Invariant Graph Neural Networks for Fluid Flow Modeling}},
author = {Shankar, Varun J and Barwey, Shivam and Maulik, Romit and Viswanathan, Venkatasubraman},
booktitle = {ICLR 2023 Workshops: Physics4ML},
year = {2023},
url = {https://mlanthology.org/iclrw/2023/shankar2023iclrw-practical/}
}