用自适应网格与剪枝技术,让神经流体模拟更准更快。
AMR-Transformer: Enabling Efficient Long-range Interaction for Complex Neural Fluid Simulation
- 结合自适应网格和剪枝模块,增强长距离流体交互能力。
- 在多个数据集上准确率提升一个数量级,计算量减少60倍。
- 适合高分辨率、含湍流/激波等复杂物理现象的仿真任务。
精确高效地模拟复杂流体动力学是一项挑战性任务,传统方法依赖计算密集型手段。基于神经网络的方法如卷积网络和图神经网络虽能高效提取局部特征,但受限于感受野,难以捕捉长程依赖;而基于Transformer的模型虽具备全局上下文,却带来高昂计算成本。为此,我们提出AMR-Transformer,一种融合新型自适应网格细化策略与纳维-斯托克斯约束感知快速剪枝模块的神经流体求解流水线。该设计促进模拟单元间的长程交互,支持对湍流、激波等全局流体波模式的建模。实验表明,该方法在保持关键细节的同时显著提升效率:在CFDBench、PDEBench及新构建的激波数据集上,相比基线模型准确率最高提升一个数量级;相较ViT,FLOPs降低高达60倍。
原文摘要 · Abstract (English)
Accurately and efficiently simulating complex fluid dynamics is a challenging task that has traditionally relied on computationally intensive methods. Neural network-based approaches, such as convolutional and graph neural networks, have partially alleviated this burden by enabling efficient local feature extraction. However, they struggle to capture long-range dependencies due to limited receptive fields, and Transformer-based models, while providing global context, incur prohibitive computational costs. To tackle these challenges, we propose AMR-Transformer, an efficient and accurate neural CFD-solving pipeline that integrates a novel adaptive mesh refinement scheme with a Navier-Stokes constraint-aware fast pruning module. This design encourages long-range interactions between simulation cells and facilitates the modeling of global fluid wave patterns, such as turbulence and shockwaves. Experiments show that our approach achieves significant gains in efficiency while preserving critical details, making it suitable for high-resolution physical simulations with long-range dependencies. On CFDBench, PDEBench and a new shockwave dataset, our pipeline demonstrates up to an order-of-magnitude improvement in accuracy over baseline models. Additionally, compared to ViT, our approach achieves a reduction in FLOPs of up to 60 times.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。