Super-resolution of digital rock images with hybrid attention multi-branch neural network

Abstract

High-resolution digital rock images are vital in studying microstructure and fluid flow characterization of oil and gas reservoirs. Computed tomography (CT) is currently the primary tool to acquire digital rock images because of its non-destructiveness and three-dimensional imaging capability. However, the restriction of imaging capability of CT sets a compromise between image resolution and field of view (FOV). The Hybrid Attention Multi-branch Super-Resolution Network (HAMSR) proposed in this study aims to remedy this deficiency. HAMSR employs a limited discrete distribution to capture all possible high-order information and generate reliable high-resolution images. The network is built to prioritize the recovery of pores, cracks, and textures in digital rock images by introducing channel and spatial attention mechanisms. The structure together with modules of HAMSR have been designed and optimized to yield superior performance with fewer parameters and less training time. HAMSR and several contemporary advanced models are trained on the DeepRock-SR 2D dataset which contains lots of sandstone and carbonate images. It is evident that HAMSR-generated images have minimal noise, clearer edges, and more micropores. Quantitative calculations show that compared to traditional bicubic interpolation, HAMSR acquires a 2.669 dB and 1.376 dB increase in peak signal-to-noise ratio (PSNR) respectively on the test sets (27%–38% reduction in relative error). Furthermore, the high-resolution (HR) images reconstructed by HAMSR are generally consistent with real HR images concerning statistical characteristics such as porosity, pore size distribution, shape factor distribution, and two-point correlation function. This work provides a new super-resolution model to generate high-quality digital rock images for further analysis.