Hybrid Deep Learning for 3D Reconstruction of Multi-Mineral Porous Media: Integrating U-Net and GAN for Enhanced Segmentation and Texture Preservation

Keshavarz, Bahareh; Masihi, Mohsen; Hajipour Shirazi Fard, Mastaneh; Biniaz Delijani, Ebrahim

doi:10.24200/sci.2025.66376.10008

Hybrid Deep Learning for 3D Reconstruction of Multi-Mineral Porous Media: Integrating U-Net and GAN for Enhanced Segmentation and Texture Preservation

Articles in Press

Document Type : Research Article

Authors

Bahareh Keshavarz ¹

Mohsen Masihi ²

Mastaneh Hajipour Shirazi Fard ¹

Ebrahim Biniaz Delijani ³

¹ Department of Petroleum Engineering, Science and Research Branch, Islamic Azad University, Tehran, Iran

² Department of Chemical & Petroleum Engineering, Sharif University of Technology Sharif University of Technology, Tehran, Iran

³ Department of Petroleum Engineering, Science and Research Branch, Islamic Azad University, Tehran, Iran.

10.24200/sci.2025.66376.10008

Abstract

This study proposes a hybrid deep learning approach combining U-Net and Generative Adversarial Network (GAN) architectures for the segmentation and texture-based reconstruction of 3D multi-mineral porous media images. The dataset consists of high-resolution 3D Leopard sandstone images, segmented into four key mineral classes: macro-pores, clay, quartz, and high-density minerals. Our approach leverages the feature extraction capabilities of a ResNet-18 backbone within U-Net, pre-trained specifically for multi-mineral segmentation, which then feeds these detailed features into a GAN framework for image reconstruction. The model effectively bridges segmentation and reconstruction, achieving superior image quality and structural fidelity compared to standalone GAN models by preserving intricate textures and maintaining macroscopic rock structures. Quantitative assessments reveal that the hybrid model yields porosity and absolute permeability values with minimal discrepancies (2.25% and 1.54% error, respectively) compared to actual data. These findings highlight the model's ability to replicate critical geophysical metrics and generate accurate 3D representations. Unlike traditional methods that either focused solely on segmentation or reconstruction, our model uniquely integrates segmentation-driven texture data for image reconstruction, offering a novel solution for geoscientific applications in hydrogeology, petroleum engineering, and environmental science

Keywords

Deep Learning

Image Reconstruction

Multi Mineral Segmentation

GAN (Generative Adversarial Network)

UNet (U-shaped Convolutional Neural Network)

Subjects