一种基于物理信息极限学习机的PDE求解方法

**作者|**PINN山里娃，作者主页
**研究方向|**物理信息驱动深度学习 不确定性 人工智能 偏微分方程 极限学习机

该作者聚焦深度学习模型与物理信息结合前沿研究，提供了一系列AI for science研究进展报告及代码实现，旨在实现物理信息驱动深度学习方法入门到精通。
内嵌物理知识神经网络（PINN）入门及相关论文
深度学习求解微分方程系列一：PINN求解框架（Poisson 1d）
深度学习求解微分方程系列二：PINN求解burger方程正问题
深度学习求解微分方程系列三：PINN求解burger方程逆问题
深度学习求解微分方程系列四：基于自适应激活函数PINN求解burger方程逆问题
深度学习求解微分方程系列五：PINN求解Navier-Stokes方程正逆问题
一种基于物理信息极限学习机的PDE求解方法

近年来，物理信息驱动的深度学习方法用于科学计算问题受到了越来越多的关注，其中，physic informed neural network(PINN)在求解微分方程(PDE)正逆问题上展现出巨大的优势，但是并不适用于某些需要实时响应的应用。由此，下面将介绍一种基于物理信息极限学习机的PDE求解方法，用于线性微分方程的快速求解，能实现近实时高精度求解(秒级求解)。首先介绍PINN及极限学习机基本方法，其次介绍基于物理信息的极限学习方法，随后介绍求解问题与背景，最后展示基于Pytorch实现的算例结果。
优点：基于物理信息极限学习机的PDE求解方法能够毫秒级求解线性微分方程，而且该方法为无网格方法，可用于复杂形状系统。

1.问题背景

线性微分方程还是理论联系实践的重要渠道之一，**在物理、工程、力学、天文学、生物学、医学、经济学等诸多领域都有重要作用。**如电子计算机与无线电装置的计算问题可归为微分方程求解；弹道计算与飞机飞行中的稳定性研究可归为线性微分方程的求解。对于线性PDE可描述为

$$\begin{array}{rl}& Lu=f_{\lambda }(x,y),(x,y)\in \Omega \\ & \mathcal{B}[u(x,y)]=b(x,y),(x,y)\in \partial \Omega \end{array}$$
式中： $L$ 为线性微分算子， $B$ 为边界或初值条件，$f_{\lambda }(x,y)$ 为系统源项，$\lambda$为物理参数，$u$为物理系统的解。

2.PINN及极限学习机简介

神经网络作为一种强大的信息处理工具在计算机视觉、生物医学、 油气工程领域得到广泛应用, 引发多领域技术变革。深度学习网络具有非常强的学习能力, 不仅能发现物理规律, 还能求解偏微分方程。近年来，基于深度学习的偏微分方程求解已是研究新热点。PINN是一种科学机器在传统数值领域的应用方法，能够用于解决与PDE相关的各种问题，包括方程求解、参数反演、模型发现、控制与优化等。随着神经网络的发展，极限学习机(ELM)由于其好的泛化性能以及极快的学习能力，在图像分类和回归问题上得到了非常大的。

2.1PINN方法求解逆问题

PINN的主要思想如图1，先构建一个输出结果为$\hat{u}$的神经网络，将其作为PDE解的代理模型，将PDE信息作为约束，编码到神经网络损失函数中进行训练。损失函数主要包括4部分：偏微分结构损失(PDE loss)，边值条件损失(BC loss)、初值条件损失(IC loss)以及真实数据条件损失(Data loss)。

图1：PINN示意图

特别的，考虑下面这个的PDE问题，其中PDE的解$u(x)$在$\Omega \subset {\mathbb{R}}^d$定义，其中$\mathbf{x}=\left(x_1,\dots ,x_d\right)$：
$$f\left(\mathbf{x};\frac{\partial u}{\partial x_1},\dots ,\frac{\partial u}{\partial x_d};\frac{{\partial }^{2}u}{\partial x_1\partial x_1},\dots ,\frac{{\partial }^{2}u}{\partial x_1\partial x_d}\right)=0,\mathbf{x}\in \Omega$$
同时，满足下面的边界
$$\mathcal{B}(u,\mathbf{x})=0\text{ on }\partial \Omega$$

PINN求解过程主要包括：

• 第一步，首先定义D层全连接层的神经网络模型：
  $$N_{\Theta }:=L_D\circ \sigma \circ L_{D-1}\circ \sigma \circ \cdots \circ \sigma \circ L_1$$
  式中：
  $$\begin{array}{rl}{L}_1(x)& :=W_{1}x+b_1,W_1\in {\mathbb{R}}^{d_1\times d},b_1\in {\mathbb{R}}^{d_1}\\ L_i(x)& :=W_{i}x+b_i,W_i\in {\mathbb{R}}^{d_i\times d_{i-1}},b_i\in {\mathbb{R}}^{d_i},\forall i=2,3,\cdots D-1,\\ L_D(x)& :=W_{D}x+b_D,W_D\in {\mathbb{R}}^{N\times d_{D-1}},b_D\in {\mathbb{R}}^N.\end{array}$$
  以及 $\sigma$ 为激活函数， $W$ 和 $b$ 为权重和偏差参数。
• 第二步，为了衡量神经网络$\hat{u}$和约束之间的差异，考虑损失函数定义：
  $$\mathcal{L}\left(\mathbf{\theta }\right)=w_f{\mathcal{L}}_{PDE}\left(\mathbf{\theta };{\mathcal{T}}_f\right)+w_i{\mathcal{L}}_{IC}\left(\mathbf{\theta };{\mathcal{T}}_i\right)+w_b{\mathcal{L}}_{BC}\left(\mathbf{\theta },;{\mathcal{T}}_b\right)+w_d{\mathcal{L}}_{Data}\left(\mathbf{\theta },;{\mathcal{T}}_{data}\right)$$
  式中：
  $$\begin{array}{rl}{\mathcal{L}}_{PDE}\left(\mathbf{\theta };{\mathcal{T}}_f\right)& =\frac{1}{\left|{\mathcal{T}}_f\right|}\sum _{\mathbf{x}\in {\mathcal{T}}_f}{\Vert f\left(\mathbf{x};\frac{\partial \hat{u}}{\partial x_1},\dots ,\frac{\partial \hat{u}}{\partial x_d};\frac{{\partial }^2\hat{u}}{\partial x_1\partial x_1},\dots ,\frac{{\partial }^2\hat{u}}{\partial x_1\partial x_d}\right)\Vert }_2^2\\ {\mathcal{L}}_{IC}\left(\mathbf{\theta };{\mathcal{T}}_i\right)& =\frac{1}{\left|{\mathcal{T}}_i\right|}\sum _{\mathbf{x}\in {\mathcal{T}}_i}\Vert \hat{u}(\mathbf{x})-u(\mathbf{x}){\Vert }_2^2\\ {\mathcal{L}}_{BC}\left(\mathbf{\theta };{\mathcal{T}}_b\right)& =\frac{1}{\left|{\mathcal{T}}_b\right|}\sum _{\mathbf{x}\in {\mathcal{T}}_b}\Vert \mathcal{B}(\hat{u},\mathbf{x}){\Vert }_2^2\\ {\mathcal{L}}_{Data}\left(\mathbf{\theta };{\mathcal{T}}_{data}\right)& =\frac{1}{\left|{\mathcal{T}}_{data}\right|}\sum _{\mathbf{x}\in {\mathcal{T}}_{data}}\Vert \hat{u}(\mathbf{x})-u(\mathbf{x}){\Vert }_2^2\end{array}$$
  $w_f$，$w_i$、$w_b$和$w_d$是权重。${\mathcal{T}}_f$，${\mathcal{T}}_i$、${\mathcal{T}}_b$和${\mathcal{T}}_{data}$表示来自PDE，初值、边值以及真值的residual points。这里的${\mathcal{T}}_f\subset \Omega$是一组预定义的点来衡量神经网络输出$\hat{u}$与PDE的匹配程度。
• 最后，利用梯度优化算法最小化损失函数，直到找到满足预测精度的网络参数 KaTeX parse error: Undefined control sequence: \theat at position 1: \̲t̲h̲e̲a̲t̲^{*}。

值得注意的是，对于逆问题，即方程中的某些参数未知。若只知道PDE方程及边界条件，PDE参数未知，该逆问题为非定问题，所以必须要知道其他信息，如部分观测点$u$ 的值。在这种情况下，PINN做法可将方程中的参数作为未知变量，加到训练器中进行优化，损失函数包括Data loss。

2.2 ELM方法求解

考虑如图2的ELM，它由 n 个神经元的单层前馈神经网络构成，假设输入为$\mathbf{x}=(x_1,\cdots ,x_j{)}^T$，输出为一个神经元$y$ ，ELM主要思想是，输入层权值和偏差是预先设定的随机值，并且在整个训练过程中都是固定的，通过训练学习得到输出层权值。

图2：极限学习机示意图

ELM求解方法
首先，确定前馈神经网络结构，初始化输入权重和偏差(初始化后固定)、输出层权重(待求解)。输入输出的映射可以表示为：
$$\mathbf{G}(\mathbf{x})\nu =\mathbf{y}$$
式中：$\mathbf{G}={\left[g\left({\vec{x}}_1\right),g\left({\vec{x}}_2\right),\dots ,g\left({\vec{x}}_j\right)\right]}^T$，$G\left(x_k\right)=\left[\phi \left(a_{j1}^T{x}_k+b_1\right),\phi \left(a_{j2}^T{x}_k+b_2\right),\dots ,\phi \left(a_{jn}^T{x}_k+b_n\right)\right]$
然后通过求解最小二乘法方法求得输出层权重
$$\nu =\mathrm{pin}v(\mathbf{G})\mathbf{x}$$

3.基于物理信息的极限学习机方法

基于物理信息极限学习机方法作为PINN的快速版本，在求解偏微分方程方面受到了广泛关注。其关键特征是用随机值固定输入层权值，并对输出层权值使用Moore-Penrose广义逆求解。
步骤：

• 首先，确定神经网络结构，单层神经网络($n$个神经元)，随机初始化输入层权值和偏差并固定，随机初始化输出层权值。假设
  $$\chi =[x,y,1{]}^T,\mathbf{\alpha }={\left[{\alpha }_1,{\alpha }_2,\cdots ,{\alpha }_n\right]}^T,\mathbf{\beta }={\left[{\beta }_1,{\beta }_2,\cdots ,{\beta }_n\right]}^T，\gamma ={\left[{\gamma }_1,{\gamma }_2,\cdots ,{\gamma }_n\right]}^T$$
  式中： $\alpha ,\beta$和$\gamma$为输出层参数，通过随机初始化后固定。网络中使用非线性激活函数$\phi =\tanh$，输出层参数为$\mathbf{\omega }={\left[{\omega }_1,{\omega }_2,\cdots ,{\omega }_n\right]}^T$，第$k$个神经元的输出可表示为 $\phi \left(z_k\right)$，$z_k=\left[{\alpha }_k,{\beta }_k,{\gamma }_k\right]\chi$，神经网络输出可表示为
  $$\hat{u}(\chi )=\phi (z)\omega$$
• 其次，将物理信息编码到极限学习网络中，由于极限学习机网络的特殊性，线性PDE方程和边界条件可描述为
  $$\begin{array}{rl}& {\mathbf{\xi }}_f=L\phi (\mathbf{z})\mathbf{\omega }-f_{\lambda }(x,y)=0\\ & {\mathbf{\xi }}_b=\mathcal{B}[\phi (\mathbf{z})\mathbf{\omega }]-b(x,y)=0\end{array}$$
  通过上式，可得到
  $$\mathbf{H}\mathbf{\omega }=\mathbf{Y}$$
  式中：$\mathbf{H}\mathbf{H}$和$\mathbf{Y}$是由$L$和$\mathcal{B}$所确定的matrix。
  然后，求解网络权重参数，利用Moore-Penrose广义逆求解可得
  $${\mathbf{w}}^{\ast }=\mathrm{pin}v(\mathbf{H})\mathbf{Y}$$
• 最后，得到PDE的解
  $$\hat{u}(\chi )=\phi (z){\mathbf{w}}^{\ast }$$

4.算例展示

4.1 2D Poisson Equation

$$u_{xx}+u_{yy}=\left(16x^2+64y^2-12\right)e^{-\left(2x^2+4y^2\right)},(x,y)\in \Omega$$
式中： Ω = { ( x , y ) ∣ x = 0.55 ρ ( θ ) cos ⁡ ( θ ) , y = 0.75 ρ ( θ ) sin ⁡ ( θ ) } \Omega=\{(x, y) \mid x=0.55 \rho(\theta) \cos (\theta), y=0.75 \rho(\theta) \sin (\theta)\} Ω={(x,y)∣x=0.55ρ(θ)cos(θ),y=0.75ρ(θ)sin(θ)}，$\rho (\theta )=1+\cos (\theta )\sin (4\theta ),0\le \theta \le 2\pi$
方程真实解为：
$$u=\frac{1}{2}+e^{-\left(2x^2+4y^2\right)}$$

4.2 结果

基于物理信息的极限学习机预测结果如图3所示，可看出，预测结果能方程好接近真解，同时计算时间为0.85s。

图3：基于物理信息的极限学习机预测

PINN预测结果如图4所示，PINN计算时间为144s，同时预测精度比基于物理信息的极限学习机差两个数量级。

图4：PINN预测结果