Meta-MgNet: Meta Multigrid Networks for Solving Parameterized Partial Differential Equations

10/27/2020
by   Yuyan Chen, et al.
0

This paper studies numerical solutions for parameterized partial differential equations (P-PDEs) with deep learning (DL). P-PDEs arise in many important application areas and the computational cost using traditional numerical schemes can be exorbitant, especially when the parameters fall into a particular range and the underlying PDE is required to be solved with high accuracy. Recently, solving PDEs with DL has become an emerging field. Existing works demonstrate great potentials of the DL based approach in speeding up numerical solutions of PDEs. However, there is still limited research on the DL approach for P-PDEs. If we directly apply existing supervised learning models to P-PDEs, the models need to be constantly fine-tuned or retrained when the parameters change. This drastically limits the applicability and utility of these models in practice. To resolve this issue, we propose a meta-learning-based method that can efficiently solve P-PDEs with a wide range of parameters without retraining. Our key observation is to regard training a solver for the P-PDE with a given set of parameters as a learning task. Then, training a solver for the P-PDEs with varied parameters can be viewed as a multi-task learning problem, to which meta-learning is one of the most effective approaches. This new perspective can be applied to many existing PDE solvers. As an example, we adopt the Multigrid Network (MgNet) as the base solver. To achieve multi-task learning, we introduce a new hypernetwork, called Meta-NN, in MgNet and refer to the entire network as the Meta-MgNet. Meta-NN takes the differential operators and the right-hand-side of the underlying P-PDEs as inputs and generates appropriate smoothers for MgNet which can significantly affect the convergent speed. Finally, extensive numerical experiments demonstrate that Meta-MgNet is more efficient in solving P-PDEs than the MG methods and MgNet.

READ FULL TEXT

page 25

page 26

page 27

research
11/03/2022

Meta-PDE: Learning to Solve PDEs Quickly Without a Mesh

Partial differential equations (PDEs) are often computationally challeng...
research
06/18/2023

Meta-Learning for Airflow Simulations with Graph Neural Networks

The field of numerical simulation is of significant importance for the d...
research
01/25/2021

A new efficient approximation scheme for solving high-dimensional semilinear PDEs: control variate method for Deep BSDE solver

This paper introduces a new approximation scheme for solving high-dimens...
research
02/16/2023

Meta-Auto-Decoder: A Meta-Learning Based Reduced Order Model for Solving Parametric Partial Differential Equations

Many important problems in science and engineering require solving the s...
research
04/29/2021

Transfer Learning on Multi-Fidelity Data

Neural networks (NNs) are often used as surrogates or emulators of parti...
research
10/21/2019

Learning and Meta-Learning of Stochastic Advection-Diffusion-Reaction Systems from Sparse Measurements

Physics-informed neural networks (PINNs) were recently proposed in [1] a...
research
08/14/2023

Efficient Neural PDE-Solvers using Quantization Aware Training

In the past years, the application of neural networks as an alternative ...

Please sign up or login with your details

Forgot password? Click here to reset