Geo-guided deep learning for spatial downscaling of solute transport in heterogeneous porous media

Abstract

Resolving solute transport in heterogeneous porous media is a complex task, because of the sparse experimental data and the high computational cost of numerical simulations. This work proposes a unique two-stage deep learning architecture comprising a dual-branch autoencoder and a geo-guided super-resolution generative adversarial network (Gg-SRGAN) to address this dual challenge. The dual-branch autoencoder addresses the issue of sparsity by constructing a continuous, but coarse representation of concentration and pressure profiles from a sparse, discontinuous profile with up to 85% missing data points. The Gg-SRGAN is then employed to generate a finer representation of field variables from the outputs generated by the dual-branch autoencoder (i.e., downscaling). We train and test our framework using six solute transport cases with varying levels of heterogeneity and compare the results with standalone methods, namely the vanilla autoencoder and vanilla SRGAN, in addition to ground truth profiles generated by the finite element method (FEM). The comparisons are performed based on several statistical metrics, such as absolute point error (APE), mean squared error (MSE), peak signal-to-noise ratio (PSNR), structural similarity index (SSIM), and learned perceptual image patch similarity (LPIPS). The first four cases are used for training, evaluation, and testing. The last two cases are utilized for blind testing to determine the generalizability of the framework. Our results show that the dual-branch autoencoder outperforms the vanilla autoencoder, and the Gg-SRGAN outperforms the SRGAN during both the training and evaluation phases. Moreover, the proposed framework can successfully construct the fine representation of concentration profiles, compared to FEM, using the coarse representation of the pressure, concentration, and domain permeability fields. When tested using the two blind test cases, the proposed dual-branch autoencoder and Gg-SRGAN exhibit superior performance compared to their counterparts in terms of all evaluation metrics.