Structure Preservation for the Deep Neural Network Multigrid Solver

Abstract

The simulation of partial differential equations is a central subject of numerical analysis and an indispensable tool in science, engineering and related fields. Existing approaches, such as finite elements, provide (highly) efficient tools but deep neural network-based techniques emerged in the last few years as an alternative with very promising results. We investigate the combination of both approaches for the approximation of the Navier-Stokes equations and to what extent structural properties such as divergence freedom can and should be respected. Our work is based on DNN-MG, a deep neural network multigrid technique, that we introduced recently and which uses a neural network to represent fine grid fluctuations not resolved by a geometric multigrid finite element solver. Although DNN-MG provides solutions with very good accuracy and is computationally highly efficient, we noticed that the neural network-based corrections substantially violate the divergence freedom of the velocity vector field. In this contribution, we discuss these findings and analyze three approaches to address the problem: a penalty term to encourage divergence freedom of the network output; a penalty term for the corrected velocity field; and a network that learns the stream function, i.e. the scalar potential of the divergence free velocity vector field and which hence yields by construction divergence free corrections. Our experimental results show that the third approach based on the stream function outperforms the other two and not only improves the divergence freedom but in particular also the overall fidelity of the simulation.

References in corpus (11)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators
• Hidden Physics Models: Machine Learning of Nonlinear Partial Differential Equations
• Fourier Neural Operator for Parametric Partial Differential Equations
• Neural Operator: Graph Kernel Network for Partial Differential Equations
• Symplectic Recurrent Neural Networks
• Lagrangian Neural Networks
• A neural network multigrid solver for the Navier-Stokes equations
• FiniteNet: A Fully Convolutional LSTM Network Architecture for Time-Dependent Partial Differential Equations
• A Conservative Finite Element Method for the Incompressible Euler Equations with Variable Density
• Neural-PDE: A RNN based neural network for solving time dependent PDEs

Cited by in corpus (4)

• A priori and a posteriori error estimates for the Deep Ritz method applied to the Laplace and Stokes problem
• DNN-MG: A Hybrid Neural Network/Finite Element Method with Applications to 3D Simulations of the Navier-Stokes Equations
• Deep neural networks for geometric multigrid methods
• A robust and stable hybrid neural network/finite element method for 2D flows that generalizes to different geometries