mlsanityData to 14 Oct 2026

Multi-Grid Graph Neural Networks with Self-Attention for Computational Mechanics

arXiv 2409.11899 ↗First posted 18 Sep 2024cs.LG cs.CEcited by 9

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  4. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  5. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  6. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  7. 2017Graph Attention Networkscited by 8,923
  8. 2016Feature Pyramid Networks for Object Detectioncited by 5,036
  9. 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
  10. 2021ViViT: A Video Vision Transformercited by 1,620
  11. 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
  12. 2020Learning to Simulate Complex Physics with Graph Networkscited by 880
16 more
  1. 2020Learning Mesh-Based Simulation with Graph Networkscited by 868
  2. 2019Graph Transformer Networkscited by 523
  3. 2020Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classificationcited by 506
  4. 2019Self-Attention Graph Poolingcited by 421
  5. 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 373
  6. 2019Understanding Attention and Generalization in Graph Neural Networkscited by 169
  7. 2018Deep Learning Methods for Reynolds-Averaged Navier-Stokes Simulations of Airfoil Flowscited by 150
  8. 2016Accelerating Eulerian Fluid Simulation With Convolutional Networkscited by 130
  9. 2022MultiScale MeshGraphNetscited by 98
  10. 2023Attending to Graph Transformerscited by 83
  11. 2020Learning Algebraic Multigrid Using Graph Neural Networkscited by 54
  12. 2018tempoGAN: A Temporally Coherent, Volumetric GAN for Super-resolution Fluid Flowcited by 44
  13. 2021Simulating Continuum Mechanics with Multi-Scale Graph Neural Networkscited by 40
  14. 2022MGAE: Masked Autoencoders for Self-Supervised Learning on Graphscited by 28
  15. 2019U-net architectures for fast prediction of incompressible laminar flowscited by 17
  16. 2023MG-GNN: Multigrid Graph Neural Networks for Learning Multilevel Domain Decomposition Methodscited by 12

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 1; 2025: 3; 2026: 5.

9 papers.