arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.04544cs.LGstat.ML

基于软注意力机制的自适应物理信息神经网络

Self-Adaptive Physics-Informed Neural Networks using a Soft Attention Mechanism

Texas A&M University(德克萨斯农工大学)

Levi McClenny, Ulisses Braga-Neto

AI总结:

针对刚性偏微分方程求解中PINN难以拟合困难区域的问题,提出带可训练软注意力权重的SA-PINN,结合高斯过程实现连续权重映射,推导其NTK矩阵解释训练机制,实验中以更少训练轮数取得更优L2误差表现。

摘要:

物理信息神经网络(Physics-Informed Neural Networks, PINNs)近期作为深度神经网络在非线性偏微分方程(partial differential equations, PDEs)数值求解领域的一类有前景的应用而兴起。然而,学界已认识到需要自适应过程来迫使神经网络准确拟合“刚性”偏微分方程求解过程中的顽固区域。本文提出了一种全新的自适应训练PINNs的方法,其中自适应权重完全可训练,且单独作用于每个训练点,因此神经网络能够自主学习解的哪些区域是困难的,并被迫聚焦于这些区域。自适应权重构成一个软乘法软注意力掩码,这与计算机视觉中使用的类似机制相似。这类SA-PINNs的核心思路是让权重随对应损失的增大而增大,这通过训练网络同时最小化损失并最大化权重来实现。此外,我们展示了如何利用高斯过程回归构建自适应权重的连续映射,这使得在传统梯度下降不足以生成精确解的问题中可以使用随机梯度下降。最后,我们推导了SA-PINNs的神经切线核(Neural Tangent Kernel, NTK)矩阵,并利用它在无限宽PINN的极限情况下,启发式地理解自适应权重对训练动力学的影响,这表明SA-PINNs的工作原理是对对应不同损失项的NTK矩阵特征值进行平滑均衡。在多个线性和非线性基准问题的数值实验中,SA-PINN在L2误差上优于其他最先进的PINN算法,同时使用的训练轮数更少。

英文摘要:

Physics-Informed Neural Networks (PINNs) have emerged recently as a promising application of deep neural networks to the numerical solution of nonlinear partial differential equations (PDEs). However, it has been recognized that adaptive procedures are needed to force the neural network to fit accurately the stubborn spots in the solution of "stiff" PDEs. In this paper, we propose a fundamentally new way to train PINNs adaptively, where the adaptation weights are fully trainable and applied to each training point individually, so the neural network learns autonomously which regions of the solution are difficult and is forced to focus on them. The self-adaptation weights specify a soft multiplicative soft attention mask, which is reminiscent of similar mechanisms used in computer vision. The basic idea behind these SA-PINNs is to make the weights increase as the corresponding losses increase, which is accomplished by training the network to simultaneously minimize the losses and maximize the weights. In addition, we show how to build a continuous map of self-adaptive weights using Gaussian Process regression, which allows the use of stochastic gradient descent in problems where conventional gradient descent is not enough to produce accurate solutions. Finally, we derive the Neural Tangent Kernel matrix for SA-PINNs and use it to obtain a heuristic understanding of the effect of the self-adaptive weights on the dynamics of training in the limiting case of infinitely-wide PINNs, which suggests that SA-PINNs work by producing a smooth equalization of the eigenvalues of the NTK matrix corresponding to the different loss terms. In numerical experiments with several linear and nonlinear benchmark problems, the SA-PINN outperformed other state-of-the-art PINN algorithm in L2 error, while using a smaller number of training epochs.

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

↑