Classification of Rock Thin Slice Images by Fusing Diffusion and Wavelet Residual Networks

Rui-shan DU , Wen-xuan MU , Ling-dong MENG

Science Technology and Engineering ›› 2026, Vol. 26 ›› Issue (13) : 5566 -5573.

PDF
Science Technology and Engineering ›› 2026, Vol. 26 ›› Issue (13) : 5566-5573 DOI: 10.12404/j.issn.1671-1815.2503159
research-article
Classification of Rock Thin Slice Images by Fusing Diffusion and Wavelet Residual Networks
Author information +
History +
PDF

Abstract

To address the classification challenge of rock thin section images due to large intra-class variations and high inter-class similarities, a dual-path classification model WD-DDPM that integrates local frequency domain features and global semantics was proposed. Based on ResNet-50, a multi-scale downsampling module (WTMDB) based on wavelet transform to replace the traditional downsampling operation was designed. By performing wavelet decomposition on the image, it retains the low-frequency contour information while enhancing the edge and texture details in the high-frequency sub-bands using an attention mechanism, effectively strengthening the model’s ability to extract local discriminative features. Additionally, a U-Net encoder from the diffusion model was introduced as a parallel path to extract global context features containing overall structure and long-range dependencies. Finally, by fusing the dual-path features of local details and global semantics, a complementary image representation was formed and sent to the classifier for lithology identification. The experimental results on a dataset containing 108 rock thin slice samples show that the accuracy of the proposed method reaches 97.5%, the recall rate is 94.7%, and the F1 score is 96.0%, which is significantly better than mainstream convolutional neural network models such as VGG16 and MobileNetV3 (under the same experimental conditions), verifying its superior performance and practical value in the complex rock image classification task.

Graphical abstract

Keywords

diffusion model / residual network / wavelet transform / rock thin section

Cite this article

Download citation ▾
Rui-shan DU, Wen-xuan MU, Ling-dong MENG. Classification of Rock Thin Slice Images by Fusing Diffusion and Wavelet Residual Networks. Science Technology and Engineering, 2026, 26(13): 5566-5573 DOI:10.12404/j.issn.1671-1815.2503159

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (17715KB)

24

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉