Learning the boundary-to-domain mapping using Lifting Product Fourier Neural Operators for partial differential equations

Aditya Kashi,Arka Daw,Muralikrishnan Gopalakrishnan Meena,Hao Lu
2024-07-01
Abstract:Neural operators such as the Fourier Neural Operator (FNO) have been shown to provide resolution-independent deep learning models that can learn mappings between function spaces. For example, an initial condition can be mapped to the solution of a partial differential equation (PDE) at a future time-step using a neural operator. Despite the popularity of neural operators, their use to predict solution functions over a domain given only data over the boundary (such as a spatially varying Dirichlet boundary condition) remains unexplored. In this paper, we refer to such problems as boundary-to-domain problems; they have a wide range of applications in areas such as fluid mechanics, solid mechanics, heat transfer etc. We present a novel FNO-based architecture, named Lifting Product FNO (or LP-FNO) which can map arbitrary boundary functions defined on the lower-dimensional boundary to a solution in the entire domain. Specifically, two FNOs defined on the lower-dimensional boundary are lifted into the higher dimensional domain using our proposed lifting product layer. We demonstrate the efficacy and resolution independence of the proposed LP-FNO for the 2D Poisson equation.
Machine Learning,Numerical Analysis
What problem does this paper attempt to address?
The problem this paper attempts to address is how to predict the solution of partial differential equations (PDEs) within a domain from boundary conditions (such as spatially varying Dirichlet boundary conditions). Specifically, the paper focuses on how to map arbitrary functions defined on low-dimensional boundaries to solutions over the entire domain, a problem known as the Boundary-to-Domain (B2D) problem. This type of problem has wide applications in fields such as fluid mechanics, solid mechanics, and heat transfer. To tackle this challenge, the authors propose a new architecture based on the Fourier Neural Operator (FNO), called the Lifting Product Fourier Neural Operator (LP-FNO). This architecture generates feature representations through two FNO blocks defined on low-dimensional boundaries, and then uses a novel lifting product layer to lift these feature representations into the high-dimensional domain. For simple 1D boundary to 2D domain problems, this lifting product operation is equivalent to an outer product. The paper uses the 2D Poisson equation as an example to demonstrate the effectiveness and resolution independence of LP-FNO in handling non-uniform spatially varying Dirichlet boundary conditions.