用空间时间感知的神经网络提升流体模拟精度与稳定性。
Mesh Based Simulations with Spatial and Temporal awareness

- 以局部拓扑为单位预测,确保空间导数一致性。
- 用注意力机制替代显式时间步,提升长期模拟稳定性。
- 融合3D旋转位置编码,适应不规则网格的对称性。
机器学习代理模型(如图神经网络和Transformer)已成为加速计算流体力学(CFD)的新方法。然而,当前训练范式仍受限于节点级监督和显式欧拉时间积分等过时假设,忽视了有限元、差分或体积法中固有的刚性动力学与局部通量连续性。本文提出统一框架,连接几何深度学习与严格数值分析。提出三项创新:(1) 多节点预测,基于模板级别的目标,预测节点及其局部拓扑的场值,强制空间导数一致性;(2) 时间校正,用时间交叉注意力实现预测-校正机制,取代不稳定的显式格式;(3) 几何归纳偏置,采用3D旋转位置编码(RoPE),有效捕捉非结构化网格中的旋转对称性。在三种架构(MeshGraphNet、Transolver 和 Transformer)上评估多个物理数据集,结果在长时滚动预测中显著提升精度与稳定性,并生成可泛化到未见子任务(如壁面剪切应力、压力预测)的潜在表示。代码已开源:https://github.com/DonsetPG/graph-physics。
原文摘要 · Abstract (English)
Machine Learning surrogates for Computational Fluid Dynamics (CFD), particularly Graph Neural Networks (GNNs) and Transformers, have become a new important approach for accelerating physics simulations. However, we identify a critical bottleneck in the field: while architectures have advanced significantly, the common underlying training paradigms remain bound to naive assumptions, such as node-wise supervision and explicit Euler time-stepping. These legacy choices ignore the stiff dynamics and local flux continuity inherent to numerous partial differential equations resolution methods, such as Finite Element, Difference, or Volume (FEM). In this work, we propose a unified framework to bridge the gap between geometric deep learning and rigorous numerical analysis. We introduce three key innovations: (1) Multi Node Prediction, a stencil-level objective that predicts field values for a node's full local topology, enforcing spatial derivative consistency; (2) Temporal Correction, replacing unstable explicit schemes with a predictor-corrector via temporal Cross-Attention; and (3) Geometric Inductive Biases, leveraging 3D Rotary Positional Embeddings (RoPE) to robustly capture rotational symmetries in unstructured meshes. We evaluate this framework across three architectures (MeshGraphNet, Transolver, and a Transformer) on diverse physics datasets. Our approach yields consistent improvements in accuracy and stability, particularly in long-horizon rollouts, while producing latent representations that generalize to unseen subtasks such as Wall Shear Stress or Pressure prediction. Code is available at https://github.com/DonsetPG/graph-physics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。