PoreDiT: A Scalable Generative Model for Large-Scale Digital Rock Reconstruction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Huang, Yizhuo, Sun, Baoquan, Huang, Haibo
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917400456200192
author Huang, Yizhuo
Sun, Baoquan
Huang, Haibo
author_facet Huang, Yizhuo
Sun, Baoquan
Huang, Haibo
contents This manuscript presents PoreDiT, a novel generative model designed for high-efficiency digital rock reconstruction at gigavoxel scales. Addressing the significant challenges in digital rock physics (DRP), particularly the trade-off between resolution and field-of-view (FOV), and the computational bottlenecks associated with traditional deep learning architectures, PoreDiT leverages a three-dimensional (3D) Swin Transformer to break through these limitations. By directly predicting the binary probability field of pore spaces instead of grayscale intensities, the model preserves key topological features critical for pore-scale fluid flow and transport simulations. This approach enhances computational efficiency, enabling the generation of ultra-large-scale ($1024^3$ voxels) digital rock samples on consumer-grade hardware. Furthermore, PoreDiT achieves physical fidelity comparable to previous state-of-the-art methods, including accurate porosity, pore-scale permeability, and Euler characteristics. The model's ability to scale efficiently opens new avenues for large-domain hydrodynamic simulations and provides practical solutions for researchers in pore-scale fluid mechanics, reservoir characterization, and carbon sequestration.
format Preprint
id arxiv_https___arxiv_org_abs_2604_10171
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle PoreDiT: A Scalable Generative Model for Large-Scale Digital Rock Reconstruction
Huang, Yizhuo
Sun, Baoquan
Huang, Haibo
Artificial Intelligence
Applied Physics
This manuscript presents PoreDiT, a novel generative model designed for high-efficiency digital rock reconstruction at gigavoxel scales. Addressing the significant challenges in digital rock physics (DRP), particularly the trade-off between resolution and field-of-view (FOV), and the computational bottlenecks associated with traditional deep learning architectures, PoreDiT leverages a three-dimensional (3D) Swin Transformer to break through these limitations. By directly predicting the binary probability field of pore spaces instead of grayscale intensities, the model preserves key topological features critical for pore-scale fluid flow and transport simulations. This approach enhances computational efficiency, enabling the generation of ultra-large-scale ($1024^3$ voxels) digital rock samples on consumer-grade hardware. Furthermore, PoreDiT achieves physical fidelity comparable to previous state-of-the-art methods, including accurate porosity, pore-scale permeability, and Euler characteristics. The model's ability to scale efficiently opens new avenues for large-domain hydrodynamic simulations and provides practical solutions for researchers in pore-scale fluid mechanics, reservoir characterization, and carbon sequestration.
title PoreDiT: A Scalable Generative Model for Large-Scale Digital Rock Reconstruction
topic Artificial Intelligence
Applied Physics
url https://arxiv.org/abs/2604.10171