arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.16183cs.LGcs.AIcs.CEphysics.comp-ph
收藏

图神经网络偏微分方程求解器:守恒性与相似性等变

Graph Neural PDE Solvers with Conservation and Similarity-Equivariance

RICOS Co. Ltd.(理科斯有限公司);Graduate School of Science and Technology, University of Tsukuba(筑波大学理工学研究科)

Masanobu Horie, Naoto Mitsume

AI总结:

本研究提出一种基于图神经网络、融入守恒定律和物理对称性的新型机器学习架构,用于求解偏微分方程,显著提升了对未见空间域的泛化能力。

摘要:

利用机器学习解决偏微分方程(PDEs)面临重大挑战,原因在于空间域及其对应状态配置的多样性,这使得仅通过数据驱动方法涵盖所有可能场景变得复杂。此外,此类方法在泛化性和可靠性方面存在合理担忧,因为它们常常忽略固有的物理约束。针对这些挑战,本研究提出了一种新型机器学习架构,该架构具有高度泛化性,并遵循守恒定律和物理对称性,从而确保更高的可靠性。该架构的基础是图神经网络(GNNs),其擅长适应各种形状和形式。此外,我们探讨了GNNs与传统数值求解器之间的相似性,促进了守恒原理和对称性无缝集成到机器学习模型中。我们的实验结果表明,模型纳入物理定律显著增强了其泛化能力,即在未见过的空间域上精度无明显下降,而其他模型则出现退化。代码可在 https://github.com/yellowshippo/fluxgnn-icml2024 获取。

英文摘要:

Utilizing machine learning to address partial differential equations (PDEs) presents significant challenges due to the diversity of spatial domains and their corresponding state configurations, which complicates the task of encompassing all potential scenarios through data-driven methodologies alone. Moreover, there are legitimate concerns regarding the generalization and reliability of such approaches, as they often overlook inherent physical constraints. In response to these challenges, this study introduces a novel machine-learning architecture that is highly generalizable and adheres to conservation laws and physical symmetries, thereby ensuring greater reliability. The foundation of this architecture is graph neural networks (GNNs), which are adept at accommodating a variety of shapes and forms. Additionally, we explore the parallels between GNNs and traditional numerical solvers, facilitating a seamless integration of conservative principles and symmetries into machine learning models. Our findings from experiments demonstrate that the model's inclusion of physical laws significantly enhances its generalizability, i.e., no significant accuracy degradation for unseen spatial domains while other models degrade. The code is available at https://github.com/yellowshippo/fluxgnn-icml2024.

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

↑