Several forms for constructing novel physics-informed neural-networks (PINN) for the solution of partial-differential-algebraic equations based on derivative operator splitting are proposed, using the nonlinear Kirchhoff rod as a prototype for demonstration. The open-source DeepXDE is likely the most well documented framework with many examples. Yet, we encountered some pathological problems and proposed novel methods to resolve them. Among these novel methods are the PDE forms, which evolve from the lower-level form with fewer unknown dependent variables to higher-level form with more dependent variables, in addition to those from lower-level forms. Traditionally, the highest-level form, the balance-of-momenta form, is the starting point for (hand) deriving the lowest-level form through a tedious (and error prone) process of successive substitutions. The next step in a finite element method is to discretize the lowest-level form upon forming a weak form and linearization with appropriate interpolation functions, followed by their implementation in a code and testing. The time-consuming tedium in all of these steps could be bypassed by applying the proposed novel PINN directly to the highest-level form. We developed a script based on JAX. While our JAX script did not show the pathological problems of DDE-T (DDE with TensorFlow backend), it is slower than DDE-T. That DDE-T itself being more efficient in higher-level form than in lower-level form makes working directly with higher-level form even more attractive in addition to the advantages mentioned further above. Since coming up with an appropriate learning-rate schedule for a good solution is more art than science, we systematically codified in detail our experience running optimization through a normalization/standardization of the network-training process so readers can reproduce our results.
翻译:本文提出了几种基于导数算子分裂构建新型物理信息神经网络(PINN)以求解偏微分代数方程的形式,并以非线性Kirchhoff杆作为演示原型。开源框架DeepXDE可能是文档最完善且包含众多示例的框架。然而,我们遇到了一些病态问题,并提出了新方法予以解决。这些新方法包括偏微分方程形式——除了从低阶形式推导外,还实现了从依赖变量较少的低阶形式向依赖变量更多的高阶形式的演进。传统上,最高阶形式(动量平衡形式)是(手动)通过繁琐(且易出错)的逐次代入过程推导最低阶形式的起点。在有限元方法中,下一步是在形成弱形式并采用适当插值函数线性化后,对最低阶形式进行离散化,随后进行代码实现与测试。通过将所提出的新型PINN直接应用于最高阶形式,可以绕过所有这些步骤中耗时的繁琐工作。我们开发了基于JAX的脚本。虽然我们的JAX脚本未出现DDE-T(基于TensorFlow后端的DDE)的病态问题,但其计算速度慢于DDE-T。值得注意的是,DDE-T本身在求解高阶形式时比低阶形式更高效,这进一步增强了直接处理高阶形式的吸引力(除前述优势外)。由于为获得良好解而设计合适的学习率调度方案更偏向艺术而非科学,我们通过神经网络训练过程的归一化/标准化处理,将优化运行经验系统化地详细编码,以确保读者可复现我们的结果。