R-adaptive DeepONet: Learning Solution Operators for PDEs with Discontinuous Solutions Using an R-adaptive Strategy

Yameng Zhu,Jingrun Chen,Weibing Deng
2024-08-08
Abstract:DeepONet has recently been proposed as a representative framework for learning nonlinear mappings between function spaces. However, when it comes to approximating solution operators of partial differential equations (PDEs) with discontinuous solutions, DeepONet poses a foundational approximation lower bound due to its linear reconstruction property. Inspired by the moving mesh (R-adaptive) method, we propose an R-adaptive DeepONet method, which contains the following components: (1) the output data representation is transformed from the physical domain to the computational domain using the equidistribution principle; (2) the maps from input parameters to the solution and the coordinate transformation function over the computational domain are learned using DeepONets separately; (3) the solution over the physical domain is obtained via post-processing methods such as the (linear) interpolation method. Additionally, we introduce a solution-dependent weighting strategy in the training process to reduce the final error. We establish an upper bound for the reconstruction error based on piecewise linear interpolation and show that the introduced R-adaptive DeepONet can reduce this bound. Moreover, for two prototypical PDEs with sharp gradients or discontinuities, we prove that the approximation error decays at a superlinear rate with respect to the trunk basis size, unlike the linear decay observed in vanilla DeepONets. Therefore, the R-adaptive DeepONet overcomes the limitations of DeepONet, and can reduce the approximation error for problems with discontinuous solutions. Numerical experiments on PDEs with discontinuous solutions, including the linear advection equation, the Burgers' equation with low viscosity, and the compressible Euler equations of gas dynamics, are conducted to verify the advantages of the R-adaptive DeepONet over available variants of DeepONet.
Numerical Analysis
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the fundamental approximate lower - bound problem encountered when the existing DeepONet approximates the solution operators of partial differential equations (PDEs) with discontinuous solutions. Specifically, due to the linear reconstruction characteristics of DeepONet, it performs poorly when dealing with PDEs with discontinuous solutions or sharp gradients, which limits its performance in some practical applications. To overcome this limitation, the authors propose a DeepONet method based on the R - adaptive strategy, namely R - adaptive DeepONet. ### Main problems 1. **Limitations of linear reconstruction**: When dealing with PDEs with discontinuous solutions, the existing DeepONet has a fundamental approximate error lower - bound due to its linear reconstruction characteristics. This means that for PDEs with discontinuous solutions, the approximate error of DeepONet cannot be effectively reduced. 2. **The need to improve approximation accuracy**: In order to improve the approximation accuracy of PDEs with discontinuous solutions, a new method is required to overcome the limitations of linear reconstruction. ### Solutions The R - adaptive DeepONet method proposed by the authors solves the above problems through the following steps: 1. **Coordinate transformation**: Transform the output data from the physical domain to the computational domain, using the equidistribution principle for the transformation. 2. **Separate learning**: Learn the mappings from input parameters to the coordinate transformation function and the solution on the computational domain separately. 3. **Post - processing**: Recover the solution on the physical domain through the interpolation method. 4. **Weighted training**: Introduce weights related to the solution during the training process to reduce the final error. ### Theoretical analysis 1. **Upper bound of reconstruction error**: The authors establish the upper bound of the reconstruction error based on piecewise linear interpolation and prove that the proposed R - adaptive DeepONet can reduce this upper bound. 2. **Super - linear convergence**: For two typical PDEs with discontinuous solutions, the authors prove that the approximate error decays at a super - linear rate as the number of basis functions in the trunk network increases, which is better than the linear decay rate of the traditional DeepONet. ### Experimental verification The authors verify the effectiveness of R - adaptive DeepONet through numerical experiments, including the linear advection equation, the Burgers equation with low viscosity, and the compressible Euler equation in gas dynamics. The experimental results show that when dealing with PDEs with discontinuous solutions, R - adaptive DeepONet has higher approximation accuracy and lower computational cost compared with the traditional DeepONet and other variant models. ### Summary This paper proposes a new R - adaptive DeepONet method. By introducing the coordinate transformation and weighted training strategy, it effectively solves the limitations of the existing DeepONet when dealing with PDEs with discontinuous solutions, and improves the approximation accuracy and computational efficiency.