Pretrain finite element method

A pretraining and warm-start framework for PDEs via physics-informed neural operators

Verfasst von

Yizheng Wang, Zhongkai Hao, Mohammad Sadegh Eshaghi, Cosmin Anitescu, Xiaoying Zhuang, Timon Rabczuk, Yinghua Liu

Abstract

We propose a Pretrained Finite Element Method (PFEM), a physics-driven framework that bridges the efficiency of neural operator learning with the accuracy and robustness of classical finite element methods (FEM). PFEM consists of a physics-informed pretraining stage and an optional warm-start stage. In the pretraining stage, a neural operator based on the Transolver architecture is trained solely from governing partial differential equations, without relying on labeled solution data. The model operates directly on unstructured point clouds, jointly encoding geometric information, material properties, and boundary conditions, and produces physically consistent initial solutions with extremely high computational efficiency. PDE constraints are enforced through explicit finite element-based differentiation, avoiding the overhead associated with automatic differentiation. In the warm-start stage, the pretrained prediction is used as an initial guess for conventional FEM solvers, preserving their accuracy, convergence guarantees, and extrapolation capability while substantially reducing the number of iterations required to reach a prescribed tolerance. PFEM is validated on a broad range of benchmark problems, including linear elasticity and nonlinear hyperelasticity with complex geometries, heterogeneous materials, and arbitrary boundary conditions. Numerical results demonstrate strong generalization in the pretraining stage with relative errors on the order of 1%, and speedups of up to one order of magnitude in the warm-start stage compared to FEM with zero initial guesses.

Details

Organisationseinheit(en)
Institut für Photonik
Externe Organisation(en)
Tsinghua University
Bauhaus-Universität Weimar
Typ
Artikel
Journal
Journal of the Mechanics and Physics of Solids
Band
214
ISSN
0022-5096
Publikationsdatum
14.05.2026
Publikationsstatus
Elektronisch veröffentlicht (E-Pub)
Peer-reviewed
Ja
ASJC Scopus Sachgebiete
Physik der kondensierten Materie, Werkstoffmechanik, Maschinenbau
Elektronische Version(en)
https://doi.org/10.1016/j.jmps.2026.106682 (Zugang: Geschlossen )
 

Zitieren

Laden...