Sliced Wasserstein Distance-Guided Three-Dimensional Porous Media Reconstruction Based on Cycle-Consistent Adversarial Network and Few-Shot Learning

Abstract

Numerical simulation studies of water–rock interaction mechanisms and pore-scale multiphase flow properties often require high computational efficiency and realistic geometries to enable a fast and accurate description of hydrodynamic behavior. In this paper, we have chosen to use deep learning models to achieve these requirements, firstly by using encoder structures to refine the image segmentation of void-solid structures on complex geometries of scanning electron microscopy (SEM) images of porous media through few-shot learning (FSL), not only obtaining an accuracy of 0.97, but also reducing the amount of annotation work. We then focus on pore-scale three-dimensional (3D) structural reconstruction using the unpaired image-to-image translation method, optimizing the cycle-consistent adversarial network (cycle-GAN) model via sliced Wasserstein distance (SWD) to transfer marine sedimentary sandstone features to the initial image, and the geometric stochastic reconstruction problems are transformed into optimization problems. Subsequently, the computational efficiency was improved by a factor of 21 by implementing the lattice Boltzmann simulation method (LBM) accelerated by GPU through compute-unified device architecture (CUDA). The flow field distribution and absolute permeability of the extracted 2D samples and the reconstructed 3D porous media structure were simulated. The results showed that our method could rapidly and accurately reconstruct the 3D structures of a given feature, ensuring statistical equivalence between the 3D reconstructed structures and 2D samples. We solve the problem of extrapolation-based 3D reconstruction of porous media and significantly reduce the time spent on structure extraction and numerical calculations.