Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Lagrangian particle flow simulation by equivariant graph neural network
Quan JIANG, Wenqing HUANG, Zhiyong GOU
Journal of Computer Applications    2025, 45 (8): 2666-2671.   DOI: 10.11772/j.issn.1001-9081.2024111601
Abstract46)   HTML0)    PDF (2667KB)(6)       Save

Graph Neural Network (GNN) is increasingly applied to complex fluid system predictions due to the superior capability in handling structured grids and strong combinatorial generalization. However, from a Lagrangian mesh-free perspective, GNN has unpredictable output variations when processing fluid particle information subjected to translation, rotation, or reflection transformations. To address this problem, an Equivariant Graph Neural Network-based Simulator (EGNS) method was proposed. Geometric vectors were first converted into relative equivariants. Then, equivariant message passing was employed at each step to ensure equivariance of entire neural network, maintaining consistency of spatial transformations between output and input equivariants. Finally, the optimized EGNS model was obtained by training in particle trajectories simulated with the Smoothed Particle Hydrodynamics (SPH) method. Experimental results on public fluid simulation datasets demonstrate that EGNS has superior predictive performance compared with Graph Neural Network-based Simulator (GNS); specifically, EGNS achieves better accuracy in predicting fluid particle movement, velocity, and typical details, decreasing Mean Squared Error (MSE) in predicting particle positions by about 16%.

Table and Figures | Reference | Related Articles | Metrics