用分段分层图网络提升动态模拟的精度与速度
M4GN: Mesh-based Multi-segment Hierarchical Graph Network for Dynamic Simulations
- 分段驱动的三级分层结构,兼顾拓扑与物理特性
- 精度最高提升56%,推理速度最快快22%
- 适合需要高效高精度物理模拟的科研与工程场景
基于网格的图神经网络已成为求解偏微分方程(PDE)模拟的有效代理模型,但其深层消息传递在大型长程网格上导致计算开销大且过度平滑;分层图网络虽缩短传播路径,仍面临两大挑战:(i) 构建尊重网格拓扑、几何及物理不连续性的粗粒度图,(ii) 在保持加速优势的同时维持细粒度精度。本文提出M4GN,一种三层次、以分段为中心的分层网络。M4GN采用混合分割策略,结合快速图划分器与基于模态分解特征的超像素式精炼,生成动态一致的连续节点段。这些段通过排列不变聚合器编码,避免了以往方法中对顺序敏感及二次复杂度的聚合问题。该信息桥接微观级GNN(捕捉局部动态)与宏观级Transformer(跨段高效推理),实现精度与效率的合理平衡。在多个代表性基准数据集上评估,M4GN预测精度最高提升56%,推理速度最快快22%优于当前最优基线。
原文摘要 · Abstract (English)
Mesh-based graph neural networks (GNNs) have become effective surrogates for PDE simulations, yet their deep message passing incurs high cost and over-smoothing on large, long-range meshes; hierarchical GNNs shorten propagation paths but still face two key obstacles: (i) building coarse graphs that respect mesh topology, geometry, and physical discontinuities, and (ii) maintaining fine-scale accuracy without sacrificing the speed gained from coarsening. We tackle these challenges with M4GN, a three-tier, segment-centric hierarchical network. M4GN begins with a hybrid segmentation strategy that pairs a fast graph partitioner with a superpixel-style refinement guided by modal-decomposition features, producing contiguous segments of dynamically consistent nodes. These segments are encoded by a permutation-invariant aggregator, avoiding the order sensitivity and quadratic cost of aggregation approaches used in prior works. The resulting information bridges a micro-level GNN, which captures local dynamics, and a macro-level transformer that reasons efficiently across segments, achieving a principled balance between accuracy and efficiency. Evaluated on multiple representative benchmark datasets, M4GN improves prediction accuracy by up to 56% while achieving up to 22% faster inference than state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。