Physics-Informed Neural Network (PINN) has proven itself a powerful tool to obtain the numerical solutions of nonlinear partial differential equations (PDEs) leveraging the expressivity of deep neural networks and the computing power of modern heterogeneous hardware. However, its training is still time-consuming, especially in the multi-query and real-time simulation settings, and its parameterization often overly excessive. In this paper, we propose the Generative Pre-Trained PINN (GPT-PINN) to mitigate both challenges in the setting of parametric PDEs. GPT-PINN represents a brand-new meta-learning paradigm for parametric systems. As a network of networks, its outer-/meta-network is hyper-reduced with only one hidden layer having significantly reduced number of neurons. Moreover, its activation function at each hidden neuron is a (full) PINN pre-trained at a judiciously selected system configuration. The meta-network adaptively ``learns'' the parametric dependence of the system and ``grows'' this hidden layer one neuron at a time. In the end, by encompassing a very small number of networks trained at this set of adaptively-selected parameter values, the meta-network is capable of generating surrogate solutions for the parametric system across the entire parameter domain accurately and efficiently.
翻译:物理信息神经网络(PINN)已证明其自身是利用深度神经网络的表达能力与现代异构硬件的计算能力求解非线性偏微分方程(PDEs)数值解的强大工具。然而,其训练过程仍耗时较长,尤其在多查询与实时仿真场景中,且参数化往往过度冗余。本文提出生成式预训练物理信息神经网络(GPT-PINN),以缓解参数化PDEs场景中的这两项挑战。GPT-PINN代表了一种全新的参数化系统元学习范式。作为一种网络之网络,其外部/元网络采用超降阶设计,仅含一个隐藏层且神经元数量显著减少。此外,每个隐藏神经元的激活函数是一个(完整)PINN,该PINN经过精心选择的系统配置的预训练。元网络自适应地"学习"系统的参数依赖性,并逐个神经元地"生长"其隐藏层。最终,通过整合少量在该组自适应选择参数值下训练的网络,元网络能够准确高效地生成整个参数域内参数化系统的替代解。