arrow
Return

Neural network-augmented differentiable finite element method for boundary value problems

delete2025-01-01
delete3
PRE
AI
X
Xi Wang
Z
Zhen‐Yu Yin *
武
武威 (Wei Wu)
H
Hehua Zhu
DOI:10.1016/j.ijmecsci.2024.109783delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Classical numerical methods such as finite element method (FEM) face limitations due to their low efficiency when addressing large-scale problems. As a novel paradigm, the physics-informed neural network (PINN) has demonstrated significant potential to solve partial differential equations. However, conventional PINNs utilize meshless control at discrete sampling points, which limits their ability to effectively handle complex boundaries. Moreover, catastrophic failure may occur in the deep energy method (DEM, a specific type of PINN). To handle these challenges, this study proposes a Neural Network-augmented Differentiable Finite Element Method (NNDFEM) by combining PINN and finite element approximation. In NNDFEM, the neural network backend solely predicts nodal variables. Derivatives and complex boundary conditions can be well handled by the finite element frontend. The governing equation over the domain, Dirichlet, and Neumann boundary conditions are directly enforced on the finite element frontend. Thus, losses of boundary conditions in PINN are rendered unnecessary. The overfitting problem in DEM is also significantly mitigated. Fully connected neural network (FCNN), modified FCNN, and graph-convolutional network are tested as backends. NNDFEM circumvents nodal force calculation and matrix assembly in FEM. Functional losses of linear elasticity, finite strain nonlinear elasticity, heat conduction, and flow in porous media are validated. A systematic exploration unveils the role of 3D finite element mesh. For large-scale problems, a multi-fidelity learning strategy is employed. Thus, the threedimensional case with over three million degrees of freedom trains well in two minutes. Benefiting from the fast inference of the neural network backend, the forward pass is 8,550 times faster than FEM.
Keywords:
Finite element method
Physics-informed neural network
Differentiable numerical analysis
Fully connected neural network
Graph convolutional network
Multi-fidelity learning

Journal

International Journal of Mechanical Sciences cover
International Journal of Mechanical Sciences
IF:
9.4
Papers:
1.0W
Citations:
4.5W

Organization

H
hong kong polytechnic university
Scholars:
3.0W
Papers: 4.1W
Citations: 921
T
tongji university
Scholars:
7.9W
Papers: 6.0W
Citations: 98
Cited Papers

Cited Papers

Thermodynamics-based Artificial Neural Networks for constitutive modeling
err2021-02-01
err208
errOAAI
errMasi, Filippo; Stefanou, Ioannis; Vannucci, Paolo; Maffi-Berthier, Victor
errShare
errSave
errShare
errSave
The role of computational science in digital twins
err2024-03-26
err8
errOAAI
errWillcox, Karen; Segundo, Brittany
errShare
errSave
Convolution Hierarchical Deep-learning Neural Networks (C-HiDeNN): finite elements, isogeometric analysis, tensor decomposition, and beyond
err2023-05-16
err19
PREAI
errLu, Ye; Li, Hengyang; Zhang, Lei; Park, Chanwook; Mojumder, Satyajit; Knapik, Stefan; Sang, Zhongsheng; Tang, Shaoqiang; Apley, Daniel W.; Wagner, Gregory J.; Liu, Wing Kam
errShare
errSave
errShare
errSave
researcher View more