用多尺度图网络提升复杂结构流体仿真精度与效率
A multi-scale feature enhanced graph neural network for fluid dynamics prediction in complex geometries

- 双步消息传递+注意力U-Net,融合局部细节与全局特征
- 在三个数据集上误差低于现有方法,速度场误差低至0.0196
- 适合需要高精度流体模拟的汽车、航空工程设计场景
工业设计中车辆与航空航天领域的流体性能评估常依赖大规模数值模拟,计算成本高昂。深度神经网络,特别是图神经网络(GNN),因其对非结构化数据的灵活性展现出提升模拟效率的潜力。然而,处理复杂几何和大规模网格时,GNN面临挑战。本文提出多尺度特征增强图神经网络(ME-GNN),采用两步消息传递机制有效捕捉局部细节特征,并结合注意力U-Net与均匀网格离散化,实现细粒度与粗粒度特征的联合提取。模型还使用K跳采样构建子图,实现在大型数据集上的高效训练,同时保留局部细节。在三个基准数据集上测试,取得当前最优结果:ShapeNet-Car上速度场相对L2误差为0.0196,表面压力误差为0.0556;AirfRANS上流场归一化均方误差为0.0033;DrivAerNet上表面压力相对L2误差为0.1416。
原文摘要 · Abstract (English)
Industrial design in fields such as vehicle and aerospace engineering often relies on large-scale numerical simulations to evaluate fluid dynamics performance, which can incur substantial computational costs. Deep neural networks have shown promise in improving simulation efficiency, especially graph neural networks (GNNs), which demonstrate great potential due to their flexibility with unstructured data. However, GNNs face challenges when dealing with tasks involving complex geometries and large-scale meshes. In this paper, we propose the Multi-scale Feature Enhanced Graph Neural Network (ME-GNN) to tackle these challenges. ME-GNN employs a graph neural network with a two-step message-passing mechanism to capture detailed local features effectively. Additionally, it integrates an Attention U-Net with uniform grid discretization, enabling the extraction of both fine and coarse features. The model also utilizes K-hop sampling to construct subgraphs, facilitating efficient training on large datasets while preserving detailed local features. We evaluated ME-GNN on three benchmark datasets and achieved state-of-the-art results: a relative L2 error of 0.0196 for the velocity field and 0.0556 for the surface pressure on ShapeNet-Car, a normalized mean squared error of 0.0033 for the flow field on AirfRANS, and a relative L2 error of 0.1416 for the surface pressure on DrivAerNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。