Improved recurrent generative model for reconstructing large-size porous media from two-dimensional images

Phys Rev E. 2022 Aug;106(2-2):025310. doi: 10.1103/PhysRevE.106.025310.

Abstract

Modeling the three-dimensional (3D) structure from a given 2D image is of great importance for analyzing and studying the physical properties of porous media. As an intractable inverse problem, many methods have been developed to address this fundamental problems over the past decades. Among many methods, the deep learning-(DL) based methods show great advantages in terms of accuracy, diversity, and efficiency. Usually, the 3D reconstruction from the 2D slice with a larger field-of-view is more conducive to simulate and analyze the physical properties of porous media accurately. However, due to the limitation of reconstruction ability, the reconstruction size of most widely used generative adversarial network-based model is constrained to 64^{3} or 128^{3}. Recently, a 3D porous media recurrent neural network based method (namely, 3D-PMRNN) (namely 3D-PMRNN) has been proposed to improve the reconstruction ability, and thus the reconstruction size is expanded to 256^{3}. Nevertheless, in order to train these models, the existed DL-based methods need to down-sample the original computed tomography (CT) image first so that the convolutional kernel can capture the morphological features of training images. Thus, the detailed information of the original CT image will be lost. Besides, the 3D reconstruction from a optical thin section is not available because of the large size of the cutting slice. In this paper, we proposed an improved recurrent generative model to further enhance the reconstruction ability (512^{3}). Benefiting from the RNN-based architecture, the proposed model requires only one 3D training sample at least and generates the 3D structures layer by layer. There are three more improvements: First, a hybrid receptive field for the kernel of convolutional neural network is adopted. Second, an attention-based module is merged into the proposed model. Finally, a useful section loss is proposed to enhance the continuity along the Z direction. Three experiments are carried out to verify the effectiveness of the proposed model. Experimental results indicate the good reconstruction ability of proposed model in terms of accuracy, diversity, and generalization. And the effectiveness of section loss is also proved from the perspective of visual inspection and statistical comparison.