arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.03485cs.LGcs.NAmath.NAstat.ML
收藏

神经算子:偏微分方程的图核网络

Neural Operator: Graph Kernel Network for Partial Differential Equations

Zongyi Li, Nikola Kovachki, Kamyar Azizzadenesheli, Burigede Liu, Kaushik Bhattacharya, Andrew Stuart, Anima Anandkumar

摘要:

神经网络的经典发展主要是有限维欧氏空间和一组类之间的映射,或者两个有限维欧氏空间之间的映射。这项工作的目的是推广神经网络,使它们能够学习无限维空间(运算符)之间的映射。我们工作的关键创新在于,在精心设计的网络架构中,可以使用一组网络参数来描述无限维空间之间以及这些空间的不同有限维近似之间的映射。通过构造非线性激活函数和一类积分算子,给出了无限维映射的逼近。核积分的计算是通过图网络上的消息传递。这种方法有实质性的实际后果,我们将说明在输入数据的偏微分方程(PDE)和他们的解决方案之间的映射的上下文中。在这种情况下,这种学习网络可以在PDE的不同近似方法(例如有限差分或有限元法)之间以及在对应于分辨率和离散化的不同底层水平的近似之间进行概括。实验证实,所提出的图核网络确实具有所需的属性,并显示出竞争力的性能相比,最先进的求解器。

英文摘要:

The classical development of neural networks has been primarily for mappings between a finite-dimensional Euclidean space and a set of classes, or between two finite-dimensional Euclidean spaces. The purpose of this work is to generalize neural networks so that they can learn mappings between infinite-dimensional spaces (operators). The key innovation in our work is that a single set of network parameters, within a carefully designed network architecture, may be used to describe mappings between infinite-dimensional spaces and between different finite-dimensional approximations of those spaces. We formulate approximation of the infinite-dimensional mapping by composing nonlinear activation functions and a class of integral operators. The kernel integration is computed by message passing on graph networks. This approach has substantial practical consequences which we will illustrate in the context of mappings between input data to partial differential equations (PDEs) and their solutions. In this context, such learned networks can generalize among different approximation methods for the PDE (such as finite difference or finite element methods) and among approximations corresponding to different underlying levels of resolution and discretization. Experiments confirm that the proposed graph kernel network does have the desired properties and show competitive performance compared to the state of the art solvers.

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

↑