Diffeomorphism Neural Operator for various domains and parameters of partial differential equations

Zhiwei Zhao,Changqing Liu,Yingguang Li,Zhibin Chen,Xu Liu
2024-06-20
Abstract:In scientific and engineering applications, solving partial differential equations (PDEs) across various parameters and domains normally relies on resource-intensive numerical methods. Neural operators based on deep learning offered a promising alternative to PDEs solving by directly learning physical laws from data. However, the current neural operator methods were limited to solve PDEs on fixed domains. Expanding neural operators to solve PDEs on various domains hold significant promise in medical imaging, engineering design and manufacturing applications, where geometric and parameter changes are essential. This paper presents a novel neural operator learning framework for solving PDEs with various domains and parameters defined for physical systems, named diffeomorphism neural operator (DNO). The main idea is that a neural operator learns in a generic domain which is diffeomorphically mapped from various physics domains expressed by the same PDE. In this way, the challenge of operator learning on various domains is transformed into operator learning on the generic domain. The generalization performance of DNO on different domains can be assessed by a proposed method which evaluates the geometric similarity between a new domain and the domains of training dataset after diffeomorphism. Experiments on Darcy flow, pipe flow, airfoil flow and mechanics were carried out, where harmonic and volume parameterization were used as the diffeomorphism for 2D and 3D domains. The DNO framework demonstrated robust learning capabilities and strong generalization performance across various domains and parameters.
Numerical Analysis,Machine Learning
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper aims to address the problem of solving partial differential equations (PDEs) in different geometric domains and under varying parameters. Traditional numerical methods require substantial computational resources and time when dealing with different geometric domains and parameter changes, which poses a significant challenge in scientific research and engineering applications. Although existing neural operator methods perform well in solving PDEs on fixed domains, they still require retraining when handling different geometric domains, leading to high time and training costs. To this end, the paper proposes a new neural operator framework—**Diffeomorphism Neural Operator (DNO)**, which addresses this issue by mapping different physical domains to a common domain. Specifically, the core idea of the DNO framework is to learn the neural operator on the common domain, thereby transforming the operator learning problem on different geometric domains into an operator learning problem on the common domain. This approach not only enhances the model's generalization ability but also accurately predicts the solution function on new domains of different shapes and sizes. ### Main Contributions 1. **Concept of Common Domain**: Introduced a new concept—the common domain, which solves the operator learning problem on different geometric domains by mapping different physical domains to the same common domain. 2. **Diffeomorphism Mapping**: Utilized Diffeomorphism (differentiable homeomorphism) techniques to map different physical domains to the common domain, ensuring the continuity and reversibility of the mapping. 3. **Generalization Performance Evaluation**: Proposed a method to evaluate the geometric similarity between new domains and domains in the training dataset to assess the generalization performance of DNO. 4. **Experimental Validation**: Validated the effectiveness and robustness of the DNO framework through multiple practical cases (such as Darcy flow, pipe flow, airfoil flow, and mechanical deformation). ### Application Scenarios - **Medical Imaging**: Processing images of different tissue structures to improve imaging quality and efficiency. - **Materials Science**: Simulating the mechanical behavior of different material structures to optimize material design. - **Engineering Simulation**: Quickly generating flow fields and stress fields under different geometric shapes and parameter conditions to accelerate product design and optimization processes. Through these contributions, the DNO framework provides an efficient and general solution for solving PDEs in different geometric domains and under varying parameters.