论文标题

对流扩散方程的PINN的神经切线内核分析

Neural tangent kernel analysis of PINN for advection-diffusion equation

论文作者

Saadat, M. H., Gjorgiev, B., Das, L., Sansavini, G.

论文摘要

物理知识的神经网络(PINNS)在数值上通过将PDE的残差及其初始/边界条件纳入损耗函数,从而近似偏微分方程(PDE)的解。尽管他们有部分成功,但即使在封闭形式的分析解决方案可获得的简单情况下,PINN仍在挣扎。为了更好地了解PINN的学习机制,这项工作着重于使用神经切线核(NTK)理论对线性对流扩散方程(LAD)的PINN的系统分析。多亏了NTK分析,研究并阐明了对流速度/扩散参数对PINN的训练动力学的影响。我们表明,PINN的训练难度是1)所谓的光谱偏见,这导致难以学习高频行为; 2)不同损耗成分之间的收敛速率差异导致训练失败。后者即使在基础PDE的解决方案没有表现出高频行为的情况下也会发生。此外,我们观察到,这种训练难度在某种程度上表现出自身,在以对流为主和扩散为主的方向上不同。还讨论了解决这些问题的不同策略。特别是,可以证明可定期激活函数可用于部分解决光谱偏置问题。

Physics-informed neural networks (PINNs) numerically approximate the solution of a partial differential equation (PDE) by incorporating the residual of the PDE along with its initial/boundary conditions into the loss function. In spite of their partial success, PINNs are known to struggle even in simple cases where the closed-form analytical solution is available. In order to better understand the learning mechanism of PINNs, this work focuses on a systematic analysis of PINNs for the linear advection-diffusion equation (LAD) using the Neural Tangent Kernel (NTK) theory. Thanks to the NTK analysis, the effects of the advection speed/diffusion parameter on the training dynamics of PINNs are studied and clarified. We show that the training difficulty of PINNs is a result of 1) the so-called spectral bias, which leads to difficulty in learning high-frequency behaviours; and 2) convergence rate disparity between different loss components that results in training failure. The latter occurs even in the cases where the solution of the underlying PDE does not exhibit high-frequency behaviour. Furthermore, we observe that this training difficulty manifests itself, to some extent, differently in advection-dominated and diffusion-dominated regimes. Different strategies to address these issues are also discussed. In particular, it is demonstrated that periodic activation functions can be used to partly resolve the spectral bias issue.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源