Solving Partial Differential Equations Based on Physics-Informed Neural Networks
This study focuses on the efficient solution of partial differential equations by applying Physics-Informed Neural Networks (PINNs). The method transforms the traditional PDE solving problem into an optimization training process of neural networks. By embedding physical laws as soft constraints into the neural network architecture, a novel fast solver for PDEs is constructed. Numerical experiments demonstrate that this approach can achieve high-precision inversion of unknown parameters in equations and construct predictive solutions with high accuracy, even under limited data conditions.
Neural Networks
偏微分方程对于生物药学模型、地球物理模型、化学模型、力学以及材料模型等复杂模型的学习有着极其重要的作用。学习这些复杂模型的过程即为高效求解偏微分方程的过程。对于微分方程的求解,分为求解微分方程正问题和反问题。从求解数学模型的角度来看,由因推果即为正问题的求解,求解正问题即是指已知偏微分方程、初边值条件以及方程中各项系数求方程的解;求解反问题则是由果推因,在具体的实验中,通常是已知方程的形式,但是系数未知,类似扩散系数、粘度系数等,这些参数的准确性直接影响方程的预测精度和可靠性,因此参数识别显得尤为重要。传统数值方法(如有限差分法、有限元法)面临高维问题中的维度灾难和反问题求解中的不适定性。近年来,物理信息神经网络(PINNs)通过将微分算子嵌入神经网络架构,实现了“无网格求解”和“正反问题统一建模”的突破。Pinkus
本文通过优化PINNs来提高预测精度并求解偏微分方程并反演方程中的未知系数,采用Adam-LBFGS两阶段优化,前期快速下降,后期精细调优,设计自适应损失权重机制,动态平衡PDE残差、边界条件和初始条件的优化过程,并提出混合采样策略,结合拉丁超立方采样与残差驱动的自适应采样来提高神经网络的效率并通过数值实验展示优化后PINNs方法的高效性。
近年来,由于深度学习技术在微分问题求解中展现出巨大潜力,因此深度学习方法被用在各类微分方程正反问题的求解中。有学者已经利用深度学习进行微分方程的求解,例如,Yuwei Fan
随着机器学习的发展,深度学习方法已被应用于偏微分方程未知系数的反演、复杂域和高维复杂微分方程的求解。深度学习通过构建包含输入和输出的网络,利用自动微分和无网格技术,通过优化损失函数来更新网络的权重,来高效求解偏微分方程。PINNs是一种具有代表性的方法,它能够利用稀疏测量数据高效地求解常微分方程和偏微分方程,与传统方法相比展现出独特的优势,损失函数的设计遵循能量守恒定律,增强了模型的学习能力和鲁棒性。
PINNs通过构建网络来逼近真实值,在偏微分方程的正反问题求解中展现了高效和快速的特点。PINNs在构造损失函数时引入了数据损失和作为正则项的物理损失,这样的物理损失使得神经网络在训练过程中充分考虑物理规律,从而提高预测精度。
通过非线性的偏微分方程来阐述PINNs的原理,考虑非线性偏微分方程的一般形式如下:
构造一个神经网络 作为精确解 的近似,其中 是包含神经网络权重和偏置的参数集合,方程的残差定义为PINNs的输出,即:
(2)
在训练的过程中,损失函数被定义为方程及其初边值条件上残值的加权和:
其中
表示训练数据
集,为初始条件的采样数据
集,为边界条件的采样数据
集,为残差数据的采样点集,
表示各部分残值的权重,
表示各部分的采样数,PINNs的网络结构如
使用PINNs训练输入数据和预测数据之间的映射模型时,需要用到大量的数据点,包含函数的内部点以及初边值点,因此对于PINNs的优化是非常有必要的,在优化器的选择上本文考虑使用两种优化器即Adam优化器和L-BFGS优化器,从
关于损失函数的优化本文采用自适应损失权重机制引入可学习的对数方差参数,通过动态调整各项损失的权重。采样点的选择方法可分为均匀型、自适应型以及基于选择准则的方法。例如,自适应均匀采样方法、Sobol序列以及拉丁超立方抽样(LHS)都是常见的采样策略,它们在不同的应用场景下具备各自的优势和特点。此外还有其他根据残差的大小选择自适应采样方法,在残差或者损失函数值大的地方多添加一些训练数据,对该区域的点进行更精细地学习,见
本文对三种采样方法的测试均在参数设置相同的两种优化器下进行,其中Adam优化器的学习率设置为0.01,L-BFGS优化器的学习率设置为1,如
使用训练好的神经网络求解偏微分方程时,输入的数据点与预测值之间建立起近似关系,使用该模型可以很快地找到某一样本点的函数值。
考虑如下扩散方程,具体形式为:
(4)
在上述扩散方程中,给定 且u的精确解为 。在本算例中,求解该方程并对扩散系数进行反演计算。该方程中的源项可通过 已知参数和精确解推导得出。网络和算法里的其他参数如下:PINNs的输入维度是2,包含一个空间坐标,一个时间坐标,7个隐藏层、每层40个神经元,输出维度为1,每层使用Tanh激活函数,在训练的初始采用Adam优化器,之后使用L-BFGS,采用自适应采样的方式,由解析解采样得到观测数据。实验中边界点数选取初始点选取残差点 实验中,采用 内部点 。
实验中各项损失函数权值采用自适应方法动态调整,将各项损失的权重视为可训练的参数,在反向传播时优化这些参数。自适应权值方法量化各项损失的重要性程度,在实验的过程中动态调整各项损失的贡献程度。整个实验在PyTorch 2.2.1版本的深度学习框架下实现。
结果,从误差图中可以发现扩散方程的未知系数的真实值与预测值之间的差异较小,这意味着反演结果在很大程度上逼近了真实值。
本研究基于PINNs的方法研究在求解偏微分方程中的应用。优化了PINNs方法的预测精度,通过对比多种采样方法、优化器的设置方式以及自适应权值的设置减少训练时长提高精度。自适应权重方法通过动态调整各项损失的权重,解决了传统固定权重方法在训练后期难以平衡不同约束条件的问题。基于残差采样强化了高残差区域的采样密度。实验结果表明,所提出的方法能够有效预测扩散方程的解,并对未知参数提供预测,可视化结果进一步表明,PINNs预测的解与精确解有较好的一致性,体现了该方法在处理具有不确定性物理参数的偏微分方程求解中的潜力。