UGrid: An Efficient-And-Rigorous Neural Multigrid Solver for Linear PDEs

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Han, Xi, Hou, Fei, Qin, Hong
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909282651340800
author Han, Xi
Hou, Fei
Qin, Hong
author_facet Han, Xi
Hou, Fei
Qin, Hong
contents Numerical solvers of Partial Differential Equations (PDEs) are of fundamental significance to science and engineering. To date, the historical reliance on legacy techniques has circumscribed possible integration of big data knowledge and exhibits sub-optimal efficiency for certain PDE formulations, while data-driven neural methods typically lack mathematical guarantee of convergence and correctness. This paper articulates a mathematically rigorous neural solver for linear PDEs. The proposed UGrid solver, built upon the principled integration of U-Net and MultiGrid, manifests a mathematically rigorous proof of both convergence and correctness, and showcases high numerical accuracy, as well as strong generalization power to various input geometry/values and multiple PDE formulations. In addition, we devise a new residual loss metric, which enables unsupervised training and affords more stability and a larger solution space over the legacy losses.
format Preprint
id arxiv_https___arxiv_org_abs_2408_04846
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle UGrid: An Efficient-And-Rigorous Neural Multigrid Solver for Linear PDEs
Han, Xi
Hou, Fei
Qin, Hong
Numerical Analysis
Artificial Intelligence
Machine Learning
Mathematical Software
Numerical solvers of Partial Differential Equations (PDEs) are of fundamental significance to science and engineering. To date, the historical reliance on legacy techniques has circumscribed possible integration of big data knowledge and exhibits sub-optimal efficiency for certain PDE formulations, while data-driven neural methods typically lack mathematical guarantee of convergence and correctness. This paper articulates a mathematically rigorous neural solver for linear PDEs. The proposed UGrid solver, built upon the principled integration of U-Net and MultiGrid, manifests a mathematically rigorous proof of both convergence and correctness, and showcases high numerical accuracy, as well as strong generalization power to various input geometry/values and multiple PDE formulations. In addition, we devise a new residual loss metric, which enables unsupervised training and affords more stability and a larger solution space over the legacy losses.
title UGrid: An Efficient-And-Rigorous Neural Multigrid Solver for Linear PDEs
topic Numerical Analysis
Artificial Intelligence
Machine Learning
Mathematical Software
url https://arxiv.org/abs/2408.04846