arXiv:2604.15398cs.LGcs.NA2026-04

用离散弱形式和神经网络求解偏微分方程,提升数值稳定性。

Python library supporting Discrete Variational Formulations and training solutions with Collocation-based Robust Variational Physics Informed Neural Networks (DVF-CRVPINN)

论文配图:Python library supporting Discrete Variational Formulations and training solutions with Collocation-based Robust Variational Physics Informed Neural Networks (DVF-CRVPINN)
图 1 · 摘自论文原文
  • 基于离散点集定义弱形式,用克罗内克函数作测试函数
  • 采用离散差分导数自动微分,训练神经网络解二维斯托克斯方程
  • 损失函数与真实误差相关,可有效控制训练过程中的数值误差

本文探索利用离散弱形式求解偏微分方程(PDEs)的可能性。提出一个编程环境,用于定义离散计算域、在点集上定义离散函数、构建离散内积,并引入基于克罗内克δ函数的离散弱形式。在此基础上,提出一种离散神经网络表示法:在离散点集上定义解函数,使用离散有限差分导数进行自动微分。以二维斯托克斯方程为例,采用离散弱残差作为损失函数,结合Adamax优化器和离散自动微分进行训练。此外,还提供了基于离散弱形式的严格数学推导,证明了损失函数的适定性和鲁棒性。解由神经网络训练得到,损失函数与真实误差相关,实现对训练过程中数值误差的稳健控制。除斯托克斯问题外,还以拉普拉斯问题为例说明该库的功能。

原文摘要 · Abstract (English)

We explore the possibility of solving Partial Differential Equations (PDEs) using discrete weak formulations. We propose a programming environment for defining a discrete computational domain, introducing discrete functions defined over a set of points, constructing discrete inner products, and introducing discrete weak formulations employing Kronecker delta test functions. Building on this setup, we propose a discrete neural network representation, training the solution function defined over a discrete set of points and employing discrete finite difference derivatives in the automatic differentiation procedures. As a challenging computational model example, we focus on Stokes equations in two-dimensions, defined over a discrete set of points. We train the solution using the discrete weak residual and the Adamax algorithm with discrete automatic differentiation of the discrete gradients. Despite introducing the python environment, we also provide a rigorous mathematical formulation based on discrete weak formulations, proving the well-posedness and robustness of the loss function. The solution of the discrete weak formulations is based on neural network training employing a robust loss function that is related to the true error. In this way, we have a robust control of the numerical error during the training of the neural networks. Besides the Stokes formulation, we also explain the functionality of the proposed library using the Laplace problem formulation.

PDE求解神经网络弱形式数值稳定

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