CoRGI: GNNs with Convolutional Residual Global Interactions for Lagrangian Simulation
Ethan Ji, Yuanzhou Chen, Arush Ramteke, Fang Sun, Tianrun Yu, Jai Parera, Peipei Ping, Wei Wang, Yizhou Sun
Abstract
Partial differential equations (PDEs) govern dynamical systems in hydrodynamics, where classical solvers face well-known difficulties with nonlinearity and computational cost. Lagrangian neural surrogates such as GNS and SEGNN learn directly from particle-based simulations, but local message passing alone is limited in its ability to represent inherently global flow interactions. We therefore frame neural Lagrangian simulation as a coarse--fine interaction problem: fine-grained local particle dynamics should be coupled with a coarse global pathway. We instantiate this idea with Convolutional Residual Global Interactions (øurs), which projects particle features to an Eulerian grid, applies residual global updates, and maps them back to particles. In this paper, the local/global pair is GNS+CNN, but in principle, the coupling interface is modular and not tied to that specific choice. With a GNS backbone, øurs improves rollout accuracy by 62% with 13% more inference time and 31% more training time. Compared to SEGNN, øurs improves accuracy by 56% while reducing inference time by 48% and training time by 31%. Under matched runtime budgets, øurs still outperforms GNS by 47% on average, supporting the hypothesis that jointly modeling coarse and fine interactions is an effective strategy for neural CFD surrogates.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext cfa25b9c-6e9f-4b6b-b64e-1a1e2bb03c59Builds on13
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Learning to Simulate Complex Physics with Graph NetworksAlvaro Sanchez-Gonzalez, Jonathan Godwin, Tobias Pfaff, Rex Ying et al.ICML 2020 · 1,439 citations
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 1,432 citations
- Learning Mesh-Based Simulation with Graph NetworksTobias Pfaff, Meire Fortunato, Alvaro Sanchez-Gonzalez, Peter W. BattagliaICLR 2021 · 1,175 citations
- Equivariant message passing for the prediction of tensorial properties and molecular spectraKristof Schütt, Oliver T. Unke, Michael GasteggerICML 2021 · 736 citations
Related papers
- Neural SPH: Improved Neural Modeling of Lagrangian Fluid DynamicsArtur P. Toshev, Jonas A. Erbesdobler, Nikolaus A. Adams, Johannes BrandstetterICML 2024 · 9 citations
- Combining Differentiable PDE Solvers and Graph Neural Networks for Fluid Flow PredictionFilipe de Avila Belbute-Peres, Thomas D. Economon, J. Zico KolterICML 2020 · 271 citations
- Learning Controllable Adaptive Simulation for Multi-resolution PhysicsTailin Wu, Takashi Maruyama, Qingqing Zhao, Gordon Wetzstein et al.ICLR 2023 · 4 citations
- Finite Volume Features, Global Geometry Representations, and Residual Training for Deep Learning-based CFD SimulationLoh Sher En Jessica, Naheed Anjum Arafat, Wei Xian Lim, Wai Lee Chan et al.ICML 2024 · 4 citations
- Universal Physics Transformers: A Framework For Efficiently Scaling Neural OperatorsBenedikt Alkin, Andreas Fürst, Simon Schmid, Lukas Gruber et al.NeurIPS 2024 · 23 citations
