[论文解读] Learning-based solutions to nonlinear hyperbolic PDEs: Empirical insights on generalization errors

Bilal Thonnam Thodi, Sai Venkata Ramana Ambadipudi+1 more|arXiv (Cornell University)|Feb 16, 2023

Model Reduction and Neural Networks被引用 4

一句话总结

本文提出了一种物理信息傅里叶神经算子（π-FNO），用于学习非线性双曲PDE的弱解，例如Lighthill-Whitham-Richards交通流模型。实证结果表明，泛化误差随输入复杂度线性增长，且物理正则化显著提升了间断解的分辨率，实现了对未见过的初始和边界条件的准确预测。

ABSTRACT

We study learning weak solutions to nonlinear hyperbolic partial differential equations (H-PDE), which have been difficult to learn due to discontinuities in their solutions. We use a physics-informed variant of the Fourier Neural Operator ($π$-FNO) to learn the weak solutions. We empirically quantify the generalization/out-of-sample error of the $π$-FNO solver as a function of input complexity, i.e., the distributions of initial and boundary conditions. Our testing results show that $π$-FNO generalizes well to unseen initial and boundary conditions. We find that the generalization error grows linearly with input complexity. Further, adding a physics-informed regularizer improved the prediction of discontinuities in the solution. We use the Lighthill-Witham-Richards (LWR) traffic flow model as a guiding example to illustrate the results.

研究动机与目标

• 研究基于深度学习的PDE求解器在具有不连续弱解的非线性双曲PDE中的泛化性能。
• 量化当初始和边界条件复杂度增加时，分布外误差的缩放关系。
• 评估物理信息正则化在提升双曲PDE中激波捕捉解保真度方面的有效性。
• 展示基于学习的求解器在真实世界应用（如交通流建模）中的可行性。

提出的方法

• 通过将H-PDE的积分形式作为损失函数（而非残差），将傅里叶神经算子（FNO）改进为物理信息变体（π-FNO）。
• π-FNO学习从初始和边界条件到时空域上弱解的解算子映射。
• 训练使用简单输入条件（如均匀或分段常数），而测试则评估其在复杂、未见过的初始和边界数据分布上的泛化能力。
• 通过测试输入上的平均绝对误差（MAE）量化泛化误差，并对误差图拟合趋势线以评估复杂度增加时的误差增长。
• 采用Lighthill-Whitham-Richards（LWR）模型作为测试平台，其通量函数为 f(u) = u·v_max·(1 - u/u_max)。
• 应用物理信息正则化以改善对间断（特别是交通流中的激波形成）的预测。

实验结果

研究问题

• RQ1 对于非线性双曲PDE，基于深度学习的PDE求解器的泛化误差如何随初始和边界条件复杂度的增加而变化？
• RQ2 与标准FNO相比，物理信息神经算子是否能改善双曲PDE中不连续弱解的预测？
• RQ3 在LWR交通流模型中，一个在简单输入条件下训练的模型，能在多大程度上泛化到复杂、分布外的初始和边界条件？
• RQ4 使用PDE的积分形式作为物理损失函数，是否能产生更符合物理规律的解，特别是在激波附近？

主要发现

• π-FNO模型的泛化误差随输入复杂度线性增长，每增加一个道路出口的交通信号灯（边界条件），误差斜率为0.141 vehs/km。
• 初始条件每增加一个非均匀性，泛化误差增加0.142 vehs/km，表明存在一致的线性增长趋势。
• 在正向问题中，π-FNO将初始条件复杂度下的误差从0.222降至0.142 vehs/km，边界条件复杂度下的误差从0.279降至0.141 vehs/km，显著优于标准FNO。
• 在反向问题中，π-FNO将初始条件的误差从0.210降至0.085 vehs/km，边界条件的误差从0.211降至0.194 vehs/km，表现出对部分观测的鲁棒性。
• π-FNO模型成功泛化到未见过的测试场景，无需反向问题中的显式边界数据，即可准确捕捉交通队列的形成与消散动态。
• 物理信息正则化显著降低了激波区域的噪声，生成的解更平滑、更符合物理规律；而标准FNO在不连续区域附近表现出伪影。