用神经网络学习非线性微分方程的有限元解算器,提升求解效率。
Learning Nonlinear Finite Element Solution Operators using Multilayer Perceptrons and Energy Minimization
- 用多层感知机拟合有限元解算器,输入参数输出解逼近值。
- 基于能量泛函损失函数,局部组装能量实现并行训练。
- 适用于需要快速求解的工程模拟,可与传统有限元结合使用。
我们开发并评估了一种学习由偏微分方程(PDEs)控制的非线性问题解算器的方法。该方法基于有限元离散化,通过多层感知机(MLP)表示解算器:以问题数据变量为输入,输出有限元解的预测结果。这些变量通常对应边界条件、系数和右端项等输入参数的参数化形式。输出为相应有限元解的近似,从而在理论上和实践中都能与标准有限元法(FEM)协同支持与增强。损失函数通常为能量泛函,并提出了基于每个单元局部组装能量的高效可并行训练算法。对于大规模问题,可通过每轮迭代仅随机选取少量网格单元来提升学习效率。该方法在多个典型测试案例中表现良好,学习得到的解算器不仅独立有效,还能与传统有限元理论和软件结合使用,显著提升求解性能。
原文摘要 · Abstract (English)
We develop and evaluate a method for learning solution operators to nonlinear problems governed by partial differential equations (PDEs). The approach is based on a finite element discretization and aims at representing the solution operator by a multilayer perceptron (MLP) that takes problem data variables as input and gives a prediction of the finite element solution as output. The variables will typically correspond to parameters in a parametrization of input data such as boundary conditions, coefficients, and right-hand sides. The output will be an approximation of the corresponding finite element solution, thus enabling support and enhancement by the standard finite element method (FEM) both theoretically and practically. The loss function is most often an energy functional and we formulate efficient parallelizable training algorithms based on assembling the energy locally on each element. For large problems, the learning process can be made more efficient by using only a small fraction of randomly chosen elements in the mesh in each iteration. The approach is evaluated on several relevant test cases, where learning the finite element solution operator turns out to be beneficial, both in its own right but also by combination with standard FEM theory and software.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。