arXiv:2411.17164cs.LGphysics.comp-ph2024-11被引 26

X-MeshGraphNet让物理模拟图神经网络更高效、无需网格,支持实时仿真。

X-MeshGraphNet: Scalable Multi-Scale Graph Neural Networks for Physics Simulation

  • 分块处理大图并加边缘区域,实现跨分区消息传递与梯度聚合。
  • 在保持全图精度前提下,推理速度显著提升,支持复杂场景实时模拟。
  • 直接从STL文件生成点云图,避免繁琐网格生成,适合工程应用。

图神经网络在复杂物理系统模拟中表现优异,但面临可扩展性差、推理需网格化及长程交互难等问题。本文提出X-MeshGraphNet,通过将大图分块并引入边界区域,实现跨分区无缝消息传递与梯度聚合,使分布式训练等效于单图处理。为消除对仿真网格的依赖,模型直接从STL等几何文件生成表面或体积分点云,并基于k近邻构建图。进一步通过迭代融合粗细分辨率点云构建多尺度图,提升长程交互效率。实验表明,该模型在保持全图GNN预测精度的同时,大幅改善可扩展性与灵活性,无需推理时生成网格,为广泛应用场景提供实用实时仿真方案。代码已开源至NVIDIA Modulus。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) have gained significant traction for simulating complex physical systems, with models like MeshGraphNet demonstrating strong performance on unstructured simulation meshes. However, these models face several limitations, including scalability issues, requirement for meshing at inference, and challenges in handling long-range interactions. In this work, we introduce X-MeshGraphNet, a scalable, multi-scale extension of MeshGraphNet designed to address these challenges. X-MeshGraphNet overcomes the scalability bottleneck by partitioning large graphs and incorporating halo regions that enable seamless message passing across partitions. This, combined with gradient aggregation, ensures that training across partitions is equivalent to processing the entire graph at once. To remove the dependency on simulation meshes, X-MeshGraphNet constructs custom graphs directly from tessellated geometry files (e.g., STLs) by generating point clouds on the surface or volume of the object and connecting k-nearest neighbors. Additionally, our model builds multi-scale graphs by iteratively combining coarse and fine-resolution point clouds, where each level refines the previous, allowing for efficient long-range interactions. Our experiments demonstrate that X-MeshGraphNet maintains the predictive accuracy of full-graph GNNs while significantly improving scalability and flexibility. This approach eliminates the need for time-consuming mesh generation at inference, offering a practical solution for real-time simulation across a wide range of applications. The code for reproducing the results presented in this paper is available through NVIDIA Modulus.

图神经网络物理模拟多尺度建模实时仿真

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。