arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.02573cs.LG
收藏

基于BSMS-GNN的网格物理模拟高效学习

Efficient Learning of Mesh-Based Physical Simulation with BSMS-GNN

Department of Computer Science, UCLA, Los Angeles, USA(加州大学洛杉矶分校计算机科学系);AR Perception, Google, Los Angeles, USA(谷歌AR感知部门);Department of Mathematics, UCLA, Los Angeles, USA(加州大学洛杉矶分校数学系)

Yadi Cao, Menglei Chai, Minchen Li, Chenfanfu Jiang

AI总结:

针对大规模网格物理模拟中图神经网络扩展复杂度和过平滑问题,提出基于二分图确定的双步幅池化策略BSMS-GNN,无需人工粗网格且避免几何边界错误边,显著提升精度和计算效率。

摘要:

使用平面图神经网络(GNN)和堆叠消息传递(MP)在大规模网格上学习物理模拟具有挑战性,因为其扩展复杂度与节点数量相关且存在过平滑问题。社区对引入多尺度结构到GNN用于物理模拟的兴趣日益增长。然而,当前最先进的方法受限于依赖人工绘制粗网格或基于空间邻近性构建粗层级,这可能在几何边界引入错误边。受二分图确定启发,我们提出了一种新颖的池化策略——双步幅(bi-stride),以解决上述限制。双步幅在广度优先搜索(BFS)的每个其他前沿上池化节点,无需手动绘制粗网格,并避免了空间邻近性导致的错误边。此外,它实现了每层级单次MP方案以及通过插值进行非参数化池化和反池化,类似于U-Net,显著降低了计算成本。实验表明,所提出的框架BSMS-GNN在代表性物理模拟中,在精度和计算效率方面均显著优于现有方法。

英文摘要:

Learning the physical simulation on large-scale meshes with flat Graph Neural Networks (GNNs) and stacking Message Passings (MPs) is challenging due to the scaling complexity w.r.t. the number of nodes and over-smoothing. There has been growing interest in the community to introduce \textit{multi-scale} structures to GNNs for physical simulation. However, current state-of-the-art methods are limited by their reliance on the labor-intensive drawing of coarser meshes or building coarser levels based on spatial proximity, which can introduce wrong edges across geometry boundaries. Inspired by the bipartite graph determination, we propose a novel pooling strategy, \textit{bi-stride} to tackle the aforementioned limitations. Bi-stride pools nodes on every other frontier of the breadth-first search (BFS), without the need for the manual drawing of coarser meshes and avoiding the wrong edges by spatial proximity. Additionally, it enables a one-MP scheme per level and non-parametrized pooling and unpooling by interpolations, resembling U-Nets, which significantly reduces computational costs. Experiments show that the proposed framework, \textit{BSMS-GNN}, significantly outperforms existing methods in terms of both accuracy and computational efficiency in representative physical simulations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑