Saved in:
Bibliographic Details
Main Authors: Linot, Alec J., Hang, Haotian, Kanso, Eva, Taira, Kunihiko
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2501.00626
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909446210322432
author Linot, Alec J.
Hang, Haotian
Kanso, Eva
Taira, Kunihiko
author_facet Linot, Alec J.
Hang, Haotian
Kanso, Eva
Taira, Kunihiko
contents Data-driven modeling of collective dynamics is a challenging problem because emergent phenomena in multi-agent systems are often shaped by long-range interactions among individuals. For example, in bird flocks and fish schools, long-range vision and flow coupling drive individual behaviors across the collective. Such collective motion can be modeled using graph neural networks (GNNs), but GNNs struggle when graphs become large and often fail to capture long-range interactions. Here, we construct hierarchical and equivariant GNNs, and show that these GNNs accurately predict local and global behavior in systems with collective motion. As representative examples, we apply this approach to simulations of clusters of point vortices and populations of microswimmers. For the point vortices, we define a local graph of vortices within a cluster and a global graph of interactions between clusters. For the microswimmers, we define a local graph around each microswimmer and a global graph that groups long-range interactions. We then combine this hierarchy of graphs with an approach that enforces equivariance to rotations and translations. This combination results in a significant improvement over a fully-connected GNN. For point vortices, our method conserves the Hamiltonian over long times, and, for microswimmers, our method predicts the transition from aggregation to swirling.
format Preprint
id arxiv_https___arxiv_org_abs_2501_00626
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Hierarchical equivariant graph neural networks for forecasting collective motion in vortex clusters and microswimmers
Linot, Alec J.
Hang, Haotian
Kanso, Eva
Taira, Kunihiko
Fluid Dynamics
Data-driven modeling of collective dynamics is a challenging problem because emergent phenomena in multi-agent systems are often shaped by long-range interactions among individuals. For example, in bird flocks and fish schools, long-range vision and flow coupling drive individual behaviors across the collective. Such collective motion can be modeled using graph neural networks (GNNs), but GNNs struggle when graphs become large and often fail to capture long-range interactions. Here, we construct hierarchical and equivariant GNNs, and show that these GNNs accurately predict local and global behavior in systems with collective motion. As representative examples, we apply this approach to simulations of clusters of point vortices and populations of microswimmers. For the point vortices, we define a local graph of vortices within a cluster and a global graph of interactions between clusters. For the microswimmers, we define a local graph around each microswimmer and a global graph that groups long-range interactions. We then combine this hierarchy of graphs with an approach that enforces equivariance to rotations and translations. This combination results in a significant improvement over a fully-connected GNN. For point vortices, our method conserves the Hamiltonian over long times, and, for microswimmers, our method predicts the transition from aggregation to swirling.
title Hierarchical equivariant graph neural networks for forecasting collective motion in vortex clusters and microswimmers
topic Fluid Dynamics
url https://arxiv.org/abs/2501.00626