Li, Kunyao, Li, Haijiang  ORCID: https://orcid.org/0000-0001-6326-8133, Khudhair, Ali ORCID: https://orcid.org/0000-0002-5062-7448, Yan, Jun and Wang, Bin
2025.
Efficient generation of digital rock CT images using LoRA-enhanced stable diffusion models.
Intelligent Geoengineering
2
(2)
, pp. 96-108.
10.1016/j.ige.2025.06.002
|
Preview |
PDF
- Published Version
Available under License Creative Commons Attribution Non-commercial No Derivatives. Download (12MB) | Preview |
Abstract
Digital rock analysis (DRA) is fundamental for geo-energy research, enabling the characterisation of microstructures for applications like hydrocarbon recovery, carbon storage, and groundwater modelling. Although 2D CT images provide valuable pore-scale data, the scarcity of real-world datasets limits the effectiveness of advanced analysis. Generative AI presents a promising approach for synthesizing high-quality rock images but faces key challenges, including high computational demands, insufficient evaluation metrics, and the trade-off between image fidelity and diversity. To address these limitations, this study proposes the use of Low-Rank Adaptation (LoRA) for fine-tuning stable diffusion models, significantly reducing computational requirements while maintaining image quality. A systematic investigation was conducted to evaluate the influence of LoRA training parameters, including rank and learning rate, on the quality of generated images. Image outputs were assessed using both standard generative metrics, such as Kernel Inception Distance (KID), and domain-specific metrics, including porosity, pore count, and pore area distributions. The optimised LoRA-enhanced diffusion model achieved a 92.6 % reduction in KID relative to baseline models, while also improving inference speed. Building on these advancements, this study demonstrates that the LoRA-enhanced diffusion model significantly improves neural network extrapolation in incomplete data scenarios through statistically consistent synthetic generation. Despite control challenges, this approach reduces costs and enables diverse applications, bridging fundamental rock physics with practical energy research.
| Item Type: | Article |
|---|---|
| Date Type: | Publication |
| Status: | Published |
| Schools: | Schools > Engineering |
| Publisher: | Elsevier |
| ISSN: | 3050-6190 |
| Date of First Compliant Deposit: | 19 June 2025 |
| Date of Acceptance: | 9 June 2025 |
| Last Modified: | 24 Jun 2025 09:45 |
| URI: | https://orca.cardiff.ac.uk/id/eprint/179213 |
Actions (repository staff only)
 |
Edit Item |
Dimensions
Dimensions