A Multigrid Graph U-Net Framework for Simulating Multiphase Flow in Heterogeneous Porous Media

Jiamin Jiang,Jingrun Chen,Zhouwang Yang
2024-12-17
Abstract:Numerical simulation of multi-phase fluid dynamics in porous media is critical to a variety of geoscience applications. Data-driven surrogate models using Convolutional Neural Networks (CNNs) have shown promise but are constrained to regular Cartesian grids and struggle with unstructured meshes necessary for accurately modeling complex geological features in subsurface simulations. To tackle this difficulty, we build surrogate models based on Graph Neural Networks (GNNs) to approximate space-time solutions of multi-phase flow and transport processes. Particularly, a novel Graph U-Net framework, referred to as AMG-GU, is developed to enable hierarchical graph learning for the parabolic pressure component of the coupled partial differential equation (PDE) system. Drawing inspiration from aggregation-type Algebraic Multigrid (AMG), we propose a graph coarsening strategy adapted to heterogeneous PDE coefficients, achieving an effective graph pooling operation. Results of three-dimensional heterogeneous test cases demonstrate that the multi-level surrogates predict pressure and saturation dynamics with high accuracy, significantly outperforming the single-level baseline. Our Graph U-Net model exhibits great generalization capability to unseen model configurations.
Computational Physics
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: **How to construct an efficient surrogate model in the numerical simulation of multiphase fluid dynamics to accurately predict the pressure and saturation dynamics in complex geological features, especially when dealing with multiphase flow problems in heterogeneous porous media on unstructured grids**. Specifically, although traditional numerical methods (such as the finite - volume method) can accurately solve partial differential equations (PDEs), the computational cost is relatively high, especially in cases of high - dimensional and complex geometries. And data - driven surrogate models based on convolutional neural networks (CNNs) can accelerate prediction, but are limited to regular Cartesian grids and have difficulty handling unstructured grids. Therefore, this paper proposes a new surrogate model based on graph neural networks (GNNs) - AMG - GU (Algebraic Multigrid Graph U - Net) to overcome these limitations. The following are the core objectives and problem decomposition of the paper: 1. **Problem Background**: - Numerical simulation of multiphase fluid dynamics in porous media is crucial for many geoscience applications, such as hydrocarbon recovery, groundwater management, and geological carbon dioxide sequestration. - Due to the highly nonlinear and complex parabolic - hyperbolic characteristics of the multiphase flow PDE system, it is especially important to develop efficient and accurate simulation techniques. 2. **Limitations of Existing Methods**: - CNNs are suitable for regular grid data, but perform poorly when dealing with unstructured grids. - Sub - surface simulations usually require corner - point grids or unstructured grids to describe complex geological features, which further limits the application of CNNs. 3. **Solutions Proposed in the Paper**: - Construct a surrogate model based on GNNs to approximate the spatio - temporal solutions of multiphase flow and transport processes in porous media. - Propose a novel Graph U - Net framework (AMG - GU), which realizes effective hierarchical graph learning through an algebraic multigrid (AMG) - inspired graph coarsening strategy. - Design a specialized network architecture for the parabolic pressure component to capture multi - scale features and reduce computational complexity. 4. **Specific Objectives**: - Develop a graph - learning model that can handle heterogeneous PDE coefficients, which is an unresolved problem in previous work. - Verify the model performance in three - dimensional heterogeneous test cases, prove its high accuracy in predicting pressure and saturation dynamics, and significantly outperform single - level baseline models. - Demonstrate the good generalization ability of the model on unseen model configurations. In summary, the goal of this paper is to construct an efficient, accurate, and surrogate model suitable for complex geological features by combining graph neural networks and algebraic multigrid methods to solve the computational challenges in multiphase fluid dynamics simulations.