Finite Element Eigenfunction Network (FEENet): A Hybrid Framework for Solving PDEs on Complex Geometries
Shiyuan Li, Hossein Salahshoor
TL;DR
FEENet addresses the challenge of solving PDEs on complex geometries by embedding the solution space in a geometry-aware spectral basis obtained from the Finite Element Method. A Branch Net learns spectral coefficients for each input, enabling a fast, resolution-independent reconstruction of the solution via $u(x)=\sum_{k=1}^M c_k \psi_k(x)$ (with time-dependent problems incorporating $e^{-\lambda_k t}$ when appropriate). Across Poisson and heat benchmarks on 2D and 3D geometries, FEENet demonstrates superior accuracy and efficiency compared to DeepONet, while offering interpretability and natural generalization to nonlocal operators through functional calculus on the spectrum. The framework decouples geometric encoding from operator learning, and the one-time eigenbasis computation can be amortized across multiple PDE problems on the same domain, enabling scalable many-query solving with physics-consistent dynamics.
Abstract
Neural operators aim to learn mappings between infinite-dimensional function spaces, but their performance often degrades on complex or irregular geometries due to the lack of geometry-aware representations. We propose the Finite Element Eigenfunction Network (FEENet), a hybrid spectral learning framework grounded in the eigenfunction theory of differential operators. For a given domain, FEENet leverages the Finite Element Method (FEM)toperformaone-timecomputationofaneigenfunctionbasisintrinsictothegeometry. PDE solutions are subsequently represented in this geometry-adapted basis, and learning is reduced to predicting the corresponding spectral coefficients. Numerical experiments conducted across a range of parameterized PDEs and complex two- and three-dimensional geometries, including benchmarks against the seminal DeepONet framework (1), demonstrate that FEENet consistently achieves superior accuracy and computational efficiency. We further highlight key advantages of the proposed approach, including resolution-independent inference, interpretability, and natural generalization to nonlocal operators defined as functions of differential operators. We envision that hybrid approaches of this form, which combine structure-preserving numerical methods with data-driven learning, offer a promising pathway toward solving real-world PDE problems on complex geometries.
