arXiv:2409.02810math.NAcs.AI2024-09被引 3

用神经网络生成空间基函数,结合有限元解时变偏微分方程。

A hybrid FEM-PINN method for time-dependent partial differential equations

  • 时间方向用有限元基函数,空间系数由神经网络输出。
  • 自适应采样提升训练效率,解决高维与低光滑性问题。
  • 适合求解复杂时变方程,尤其对数据稀疏场景有效。

本文提出一种混合数值方法,用于求解演化型偏微分方程(PDEs),将时间有限元法与深度神经网络相结合。与传统基于深度学习的模型不同,本方法在时间方向使用有限元基函数,而空间依赖系数由神经网络输出。通过在时间方向施加Galerkin或配点投影,得到一组关于空间系数的PDE,并在PINN框架中进行逼近。该混合策略具有双重优势:避免了时间积分中的统计误差,且神经网络输出可视为一组降维后的空间基函数。为缓解高维与低正则性带来的困难,我们设计了一种自适应采样策略:利用显式密度模型近似由PDE残差诱导的分布,并据此生成新的时间相关随机样本以扩充训练集。一系列数值实验验证了该方法的有效性与高效性。

原文摘要 · Abstract (English)

In this work, we present a hybrid numerical method for solving evolution partial differential equations (PDEs) by merging the time finite element method with deep neural networks. In contrast to the conventional deep learning-based formulation where the neural network is defined on a spatiotemporal domain, our methodology utilizes finite element basis functions in the time direction where the space-dependent coefficients are defined as the output of a neural network. We then apply the Galerkin or collocation projection in the time direction to obtain a system of PDEs for the space-dependent coefficients which is approximated in the framework of PINN. The advantages of such a hybrid formulation are twofold: statistical errors are avoided for the integral in the time direction, and the neural network's output can be regarded as a set of reduced spatial basis functions. To further alleviate the difficulties from high dimensionality and low regularity, we have developed an adaptive sampling strategy that refines the training set. More specifically, we use an explicit density model to approximate the distribution induced by the PDE residual and then augment the training set with new time-dependent random samples given by the learned density model. The effectiveness and efficiency of our proposed method have been demonstrated through a series of numerical experiments.

偏微分方程神经网络有限元自适应采样

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。