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

用于求解不规则域和非结构化网格上计算力学问题的空谱图神经算子

Spatio-spectral graph neural operator for solving computational mechanics problems on irregular domain and unstructured grid

Yardi School of Artificial Intelligence(亚迪人工智能学院);Indian Institute of Technology Delhi(印度理工学院德里分校)

Subhankar Sarkar, Souvik Chakraborty

AI总结:

本文提出空谱图神经算子Sp$^2$GNO,融合空间与谱图神经网络以克服非结构化网格和不规则域上的求解难题,在规则与不规则域上的时间相关及无关偏微分方程求解中表现优异,并通过计算力学与科学计算基准及实际应用验证其有效性。

摘要:

随着算子学习的出现,科学机器学习取得了显著进展。然而,现有方法在应用于非结构化网格和不规则域上的问题时遇到困难。空间图神经网络利用邻域中的局部卷积来潜在地应对这些挑战,但它们在深层架构中常常遭受过度平滑和过度挤压等问题。相反,谱图神经网络利用全局卷积来捕获域图中的广泛特征和长程依赖关系,尽管由于特征值分解而导致计算成本高昂。在本文中,我们引入了一种新方法,称为空谱图神经算子(Sp$^2$GNO),它有效地整合了空间图神经网络和谱图神经网络。该框架缓解了各单独方法的局限性,并使得能够学习任意几何形状上的解算子,从而适用于广泛的现实世界问题。Sp$^2$GNO 在求解规则和不规则域上的时间相关和时间无关偏微分方程方面表现出卓越的性能。我们的方法通过来自计算力学和科学计算文献的综合基准测试和实际应用得到了验证。

英文摘要:

Scientific machine learning has seen significant progress with the emergence of operator learning. However, existing methods encounter difficulties when applied to problems on unstructured grids and irregular domains. Spatial graph neural networks utilize local convolution in a neighborhood to potentially address these challenges, yet they often suffer from issues such as over-smoothing and over-squashing in deep architectures. Conversely, spectral graph neural networks leverage global convolution to capture extensive features and long-range dependencies in domain graphs, albeit at a high computational cost due to Eigenvalue decomposition. In this paper, we introduce a novel approach, referred to as Spatio-Spectral Graph Neural Operator (Sp$^2$GNO) that integrates spatial and spectral GNNs effectively. This framework mitigates the limitations of individual methods and enables the learning of solution operators across arbitrary geometries, thus catering to a wide range of real-world problems. Sp$^2$GNO demonstrates exceptional performance in solving both time-dependent and time-independent partial differential equations on regular and irregular domains. Our approach is validated through comprehensive benchmarks and practical applications drawn from computational mechanics and scientific computing literature.

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

↑