论文标题

使用图神经网络的多边形网格的集聚,并应用于多机求解器

Agglomeration of Polygonal Grids using Graph Neural Networks with applications to Multigrid solvers

论文作者

Antonietti, P. F., Farenga, N., Manuzzi, E., Martinelli, G., Saverio, L.

论文摘要

基于聚集的策略在自适应改进算法中都很重要,并且要构建可扩展的多级代数求解器。为了自动执行多边形网格的聚集,我们提出了机器学习(ML)策略的使用,这些策略可以自然利用有关网格的几何信息,以保持网格质量,增强数值方法的性能并降低整体计算成本。特别是,我们采用K-均值聚类算法和图神经网络(GNNS)来分区计算网格的连接图。此外,GNN具有较高的在线推理速度,并且同时自然,同时处理网格的图形结构和几何信息(例如元素的面积或其Barycentric坐标)的优势。将这些技术与METIS(用于图形分区的标准算法)进行了比较,该算法仅处理网格的图信息。我们证明,在ML策略方面,从质量指标方面提高了性能。当应用于更复杂的几何形状(例如大脑MRI扫描)以及保留网格质量的能力时,此类模型还显示出良好的概括程度。当在多边形不连续的Galerkin(Polydg)框架中,应用于多族(MG)求解器时,这些策略的有效性也得到了证明。在考虑的实验中,GNNS在推理速度,准确性和方法的灵活性方面表现出最佳性能。

Agglomeration-based strategies are important both within adaptive refinement algorithms and to construct scalable multilevel algebraic solvers. In order to automatically perform agglomeration of polygonal grids, we propose the use of Machine Learning (ML) strategies, that can naturally exploit geometrical information about the mesh in order to preserve the grid quality, enhancing performance of numerical methods and reducing the overall computational cost. In particular, we employ the k-means clustering algorithm and Graph Neural Networks (GNNs) to partition the connectivity graph of a computational mesh. Moreover, GNNs have high online inference speed and the advantage to process naturally and simultaneously both the graph structure of mesh and the geometrical information, such as the areas of the elements or their barycentric coordinates. These techniques are compared with METIS, a standard algorithm for graph partitioning, which is meant to process only the graph information of the mesh. We demonstrate that performance in terms of quality metrics is enhanced for ML strategies. Such models also show a good degree of generalization when applied to more complex geometries, such as brain MRI scans, and the capability of preserving the quality of the grid. The effectiveness of these strategies is demonstrated also when applied to MultiGrid (MG) solvers in a Polygonal Discontinuous Galerkin (PolyDG) framework. In the considered experiments, GNNs show overall the best performance in terms of inference speed, accuracy and flexibility of the approach.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源