Multi-Grid Graph Neural Networks with Self-Attention for Computational Mechanics
Not on the map. Why?
Abstract (arXiv)
Advancement in finite element methods have become essential in various disciplines, and in particular for Computational Fluid Dynamics (CFD), driving research efforts for improved precision and efficiency. While Convolutional Neural Networks (CNNs) have found success in CFD by mapping meshes into images, recent attention has turned to leveraging Graph Neural Networks (GNNs) for direct mesh processing. This paper introduces a novel model merging Self-Attention with Message Passing in GNNs, achieving a 15% reduction in RMSE on the well known flow past a cylinder benchmark. Furthermore, a dynamic mesh pruning technique based on Self-Attention is proposed, that leads to a robust GNN-based multigrid approach, also reducing RMSE by 15%. Additionally, a new self-supervised training method based on BERT is presented, resulting in a 25% RMSE reduction. The paper includes an ablation study and outperforms state-of-the-art models on several challenging datasets, promising advancements similar to those recently achieved in natural language and image processing. Finally, the paper introduces a dataset with meshes larger than existing ones by at least an order of magnitude. Code and Datasets will be released at https://github.com/DonsetPG/multigrid-gnn.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 28 of its 43 references that are in the map. First posted on arXiv 18 Sep 2024.
Builds on
28 of 43 references.
References · 28
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
- 2021ViViT: A Video Vision Transformercited by 1,620
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 880
16 more
- 2020Learning Mesh-Based Simulation with Graph Networkscited by 868
- 2019Graph Transformer Networkscited by 523
- 2020Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classificationcited by 506
- 2019Self-Attention Graph Poolingcited by 421
- 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 373
- 2019Understanding Attention and Generalization in Graph Neural Networkscited by 169
- 2018Deep Learning Methods for Reynolds-Averaged Navier-Stokes Simulations of Airfoil Flowscited by 150
- 2016Accelerating Eulerian Fluid Simulation With Convolutional Networkscited by 130
- 2022MultiScale MeshGraphNetscited by 98
- 2023Attending to Graph Transformerscited by 83
- 2020Learning Algebraic Multigrid Using Graph Neural Networkscited by 54
- 2018tempoGAN: A Temporally Coherent, Volumetric GAN for Super-resolution Fluid Flowcited by 44
- 2021Simulating Continuum Mechanics with Multi-Scale Graph Neural Networkscited by 40
- 2022MGAE: Masked Autoencoders for Self-Supervised Learning on Graphscited by 28
- 2019U-net architectures for fast prediction of incompressible laminar flowscited by 17
- 2023MG-GNN: Multigrid Graph Neural Networks for Learning Multilevel Domain Decomposition Methodscited by 12
Built on by
Citing papers per year: 2024: 1; 2025: 3; 2026: 5.
9 papers.
- 2026MeshGraphNet-Transformer: Scalable Mesh-based Learned Simulation for Solid Mechanicscited by 10
- 2025MeshMask: Physics-Based Simulations with Masked Graph Neural Networkscited by 9
- 2024EvoMesh: Adaptive Physical Simulation with Hierarchical Graph Evolutionscited by 4
- 2025Curriculum Learning for Mesh-based simulationscited by 1
- 2026Mask-Morph Graph U-Net: A Generalisable Mesh-Based Surrogate for Crashworthiness Field Prediction under Large Geometric Variationcited by 1
- 2025Automated discovery of finite volume schemes using Graph Neural Networkscited by 0
- 2026Multi-fidelity graph-based neural networks architectures to learn Navier-Stokes solutions on non-parametrized 2D domainscited by 0
- 2026Mesh Based Simulations with Spatial and Temporal awarenesscited by 0
- 2026Physics-Informed Coarsening for Multigrid Graph Neural Surrogatescited by 0
- 2026Physics-Informed Coarsening for Multigrid Graph Neural Surrogates29 May 2026
- 2026Mask-Morph Graph U-Net: A Generalisable Mesh-Based Surrogate for Crashworthiness Field Prediction under Large Geometric Variation13 May 2026
- 2026Mesh Based Simulations with Spatial and Temporal awareness2 May 2026
- 2026MeshGraphNet-Transformer: Scalable Mesh-based Learned Simulation for Solid Mechanics30 Jan 2026
- 2026Multi-fidelity graph-based neural networks architectures to learn Navier-Stokes solutions on non-parametrized 2D domains5 Jan 2026
- 2025Curriculum Learning for Mesh-based simulations16 Sep 2025
- 2025Automated discovery of finite volume schemes using Graph Neural Networks26 Aug 2025
- 2025MeshMask: Physics-Based Simulations with Masked Graph Neural Networks15 Jan 2025
- 2024EvoMesh: Adaptive Physical Simulation with Hierarchical Graph Evolutions3 Oct 2024