Deep Reinforcement Learning for Adaptive Mesh Refinement

by   Corbin Foucart, et al.

Finite element discretizations of problems in computational physics often rely on adaptive mesh refinement (AMR) to preferentially resolve regions containing important features during simulation. However, these spatial refinement strategies are often heuristic and rely on domain-specific knowledge or trial-and-error. We treat the process of adaptive mesh refinement as a local, sequential decision-making problem under incomplete information, formulating AMR as a partially observable Markov decision process. Using a deep reinforcement learning approach, we train policy networks for AMR strategy directly from numerical simulation. The training process does not require an exact solution or a high-fidelity ground truth to the partial differential equation at hand, nor does it require a pre-computed training dataset. The local nature of our reinforcement learning formulation allows the policy network to be trained inexpensively on much smaller problems than those on which they are deployed. The methodology is not specific to any particular partial differential equation, problem dimension, or numerical discretization, and can flexibly incorporate diverse problem physics. To that end, we apply the approach to a diverse set of partial differential equations, using a variety of high-order discontinuous Galerkin and hybridizable discontinuous Galerkin finite element discretizations. We show that the resultant deep reinforcement learning policies are competitive with common AMR heuristics, generalize well across problem classes, and strike a favorable balance between accuracy and cost such that they often lead to a higher accuracy per problem degree of freedom.


page 15

page 19

page 21

page 22

page 23


Reinforcement Learning for Adaptive Mesh Refinement

Large-scale finite element simulations of complex physical systems gover...

Numerical approximation of partial differential equations with MFEM library

We revise the finite element formulation for Lagrange, Raviart- Thomas, ...

Learning robust marking policies for adaptive mesh refinement

In this work, we revisit the marking decisions made in the standard adap...

Multi-Agent Reinforcement Learning for Adaptive Mesh Refinement

Adaptive mesh refinement (AMR) is necessary for efficient finite element...

Adaptive Mesh Refinement in Analog Mesh Computers

The call for efficient computer architectures has introduced a variety o...

Data-based Adaptive Refinement of Finite Element Thin Plate Spline

The thin plate spline, as introduced by Duchon, interpolates a smooth su...

Neural Networks to solve Partial Differential Equations: a Comparison with Finite Elements

We compare the Finite Element Method (FEM) simulation of a standard Part...

Please sign up or login with your details

Forgot password? Click here to reset