Learning optimal multigrid smoothers via neural networks

02/24/2021
by   Ru Huang, et al.
0

Multigrid methods are one of the most efficient techniques for solving linear systems arising from Partial Differential Equations (PDEs) and graph Laplacians from machine learning applications. One of the key components of multigrid is smoothing, which aims at reducing high-frequency errors on each grid level. However, finding optimal smoothing algorithms is problem-dependent and can impose challenges for many problems. In this paper, we propose an efficient adaptive framework for learning optimized smoothers from operator stencils in the form of convolutional neural networks (CNNs). The CNNs are trained on small-scale problems from a given type of PDEs based on a supervised loss function derived from multigrid convergence theories, and can be applied to large-scale problems of the same class of PDEs. Numerical results on anisotropic rotated Laplacian problems demonstrate improved convergence rates and solution time compared with classical hand-crafted relaxation methods.

READ FULL TEXT
research
02/25/2019

Learning to Optimize Multigrid PDE Solvers

Constructing fast numerical solvers for partial differential equations (...
research
02/05/2019

Some machine learning schemes for high-dimensional nonlinear PDEs

We propose new machine learning schemes for solving high dimensional non...
research
04/16/2021

Finite Difference Nets: A Deep Recurrent Framework for Solving Evolution PDEs

There has been an arising trend of adopting deep learning methods to stu...
research
05/19/2022

Learning Interface Conditions in Domain Decomposition Solvers

Domain decomposition methods are widely used and effective in the approx...
research
06/01/2022

The effect of time discretization on the solution of parabolic PDEs with ANNs

We investigate the resolution of parabolic PDEs via Extreme Learning Mac...
research
02/02/2023

An Enhanced V-cycle MgNet Model for Operator Learning in Numerical Partial Differential Equations

This study used a multigrid-based convolutional neural network architect...
research
09/20/2018

Machine Learning for semi linear PDEs

Recent machine learning algorithms dedicated to solving semi-linear PDEs...

Please sign up or login with your details

Forgot password? Click here to reset