|
|
|
|
|
|
| Intelligent Lithology Identification Based on Transfer Learning of Rock Hyperspectral Images |
| LI Shan1, 2, 3, LIN Peng1, 2, 3, XU Zhen-hao1, 2, 3*, XIANG Hang1, 2, 3, LI Qian-ji1, 2, 3 |
1. School of Qilu Transportation, Shandong University, Jinan 250061, China
2. State Key Laboratory for Tunnel Engineering, Jinan 250061, China
3. Institute of Geotechnical and Underground Engineering, Shandong University, Jinan 250061, China
|
|
|
|
|
Abstract The rapid identification of lithology holds significant fundamental geological research significance as well as engineering application value. Traditional lithology recognition primarily depends on the image features of rocks. However, confusion tends to arise when identifying rocks with similar appearances. Consequently, relevant studies further utilize spectral features to reflect the compositional information of rocks. Nevertheless, spectral testing usually demands sample preparation and belongs to the category of destructive testing. This article proposes an intelligent lithology recognition method based on transfer learning of rock hyperspectral images, taking advantage of the integrated imaging hyperspectral technology and the non-destructive, non-contact imaging characteristics. Firstly, the hyperspectral data of the rock region of interest are normalized, and dimensionality reduction is performed to reduce the redundancy of spectral data. Then, a rock hyperspectral image transfer learning model is constructed using a 3D ResNet network, and three-dimensional information is extracted through a residual network. The transfer learning method is reused to train the model by loading pre-trained weights, thereby achieving intelligent recognition of lithology. In this article, the confusion matrix, accuracy (ACC), precision (P), recall (R), and F1 values (F1) are used as evaluation indicators for model accuracy. A comparative analysis is conducted on ResNet101 and ResNet18/34/50 models. The results indicate that the ResNet-101 migration model has the highest accuracy in the test set, reaching 98.29%. The highest P can reach 98.32%, the highest R can reach 98.29%, and the highest F1 can reach 98.31%. The accuracy of ResNet-101 in identifying rock spectral data is over 90% (except for chlorite schist), and most results can even reach 100%. Compared to ResNet18/34/50, ResNet101 has higher recognition accuracy and better stability for identifying each type of rock. In addition, this method was employed to predict the lithology of sampled tunnel site rocks pixel by pixel, verifying the good robustness and generalization performance of the proposed lithology intelligent identification method, which can be used for rapid and intelligent lithology identification in engineering fields like geology, logging, transportation, and water conservancy.
|
|
Received: 2024-11-11
Accepted: 2025-03-24
|
|
|
|
Corresponding Authors:
XU Zhen-hao
E-mail: zhenhao_xu@sdu.edu.cn
|
|
[1] Shi H, Xu Z H, Lin P, et al. Geoenergy Science and Engineering, 2023, 231: 212382.
[2] Liang H B, Chen H F, Guo J H, et al. Expert Systems with Applications, 2022, 189: 116142.
[3] Song L, Yin X Y, Yin L J. IEEE Geoscience and Remote Sensing Letters, 2023, 20: 7503105.
[4] XU Zhen-hao, MA Wen, LI Shu-cai, et al（许振浩，马 文，李术才，等）. Geological Review（地质论评）, 2022, 68(6): 2290.
[5] Xu Z H, Ma W, Lin P, et al. Journal of Rock Mechanics and Geotechnical Engineering, 2022, 14(4): 1140.
[6] LI Juan, SUN Hui-lan, HOU Qing-xiang（李 娟，孙惠兰，侯庆香）. China Petrochem（中国石油石化）, 2017, 10: 61.
[7] Hou S K, Shi H Y, Cao X H, et al. IEEE Transactions on Geoscience and Remote Sensing, 2022, 60: 5521213.
[8] Farmonov N, Amankulova K, Szatmári J, et al. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2023, 16: 1576.
[9] ZHONG Jia-ping, LI Yun-song, XIE Wei-ying, et al（钟佳平，李云松，谢卫莹，等）. Acta Electronica Sinica（电子学报）, 2024, 52(5): 1716.
[10] YAN Shou-xun, ZHANG Bing, ZHAO Yong-chao, et al(燕守勋, 张 兵, 赵永超, 等). Remote Sensing Technology and Applications（遥感技术与应用）, 2003, 18(4): 191.
[11] WU Meng-juan, JIN Jia, WANG Jin-lin, et al(吴梦娟，靳 佳，王金林，等). Acta Geologica Sinica(地质学报)，2024，98(1)：314.
[12] WANG Jun-jie, YUAN Xi-ping, GAN Shu, et al(王俊杰, 袁希平, 甘 淑, 等). Journal of Lanzhou University (Natural Science Edition)[兰州大学学报(自然科学版)], 2023, 59(6): 786.
[13] Lin N, Fu J W, Jiang R Z, et al. Remote Sensing, 2023, 15(15): 3764.
[14] LI Lian-jie, FAN Shu-xiang, WANG Xue-wen, et al(李廉洁, 樊书祥, 王学文, 等). Spectroscopy and Spectral Analysis(光谱学与光谱分析), 2022, 42(4): 1250.
[15] Xu Y H, Wang H, Zhou F, et al. IEEE Transactions on Geoscience and Remote Sensing, 2025, 63: 5511216.
[16] He L, Li J, Liu C Y, et al. IEEE Transactions on Geoscience and Remote Sensing, 2018, 56(3): 1579.
[17] Li Z Y, Xue Z H, Xu Q, et al. IEEE Transactions on Geoscience and Remote Sensing, 2024, 62: 5502019.
[18] ZHAO Xin, MA Jing-yi, CHEN Han, et al（赵 昕，马竞一，陈 晗，等）. Transactions of the Chinese Society for Agricultural Machinery（农业机械学报）, 2024, 55(4): 368.
[19] ZHOU Fei-xiang, JIANG Hong, GUO Bao-lin, et al（周飞翔，姜 红，郭宝林，等）. China Journal of Chinese Materia Medica（中国中药杂志）, 2024, 49(24): 6660.
[20] Okada N, Maekawa Y, Owada N, et al. Minerals, 2020, 10(9): 809.
[21] Afjal M I, Mondal M N I, Mamun M A. Journal of Spatial Science, 2024, 69(3): 821.
[22] Mei S H, Ji J Y, Hou J H, et al. IEEE Transactions on Geoscience and Remote Sensing, 2017, 55(8): 4520.
[23] Ma X T, Man Q X, Yang X M, et al. Remote Sensing, 2023, 15(4): 992.
[24] LIU Ye, HAN Yu-bo, ZHU Wen-rui（刘 烨，韩雨伯，朱文瑞）. Earth Science Fronties（地学前缘）, 2024, 31(4): 95.
[25] Liu L F, Ji M, Buchroithner M. Sensors, 2018, 18(9): 3169.
[26] Galdames F J, Perez C A, Estevez P A, et al. Chemometrics and Intelligent Laboratory Systems, 2022, 224: 104538.
[27] Yang K, Zhao M, Argyropoulos D. Postharvest Biology and Technology, 2025, 219: 113247.
[28] Xu Z H, Shi H, Lin P, et al. International Journal of Rock Mechanics and Mining Sciences, 2024, 180: 105814.
[29] Xu Z H, Yu T F, Lin P, et al. Engineering Geology, 2023, 325: 107279.
|
| [1] |
LI Yun-peng, WANG Hong-wei, DAI Xue-jing, WU Lian-quan, HU Wei-cheng, ZHANG Yan-chun. Fast Visual Identification Method of Pyrotechnic Composition Based on Hyperspectral Imaging[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(08): 2183-2189. |
| [2] |
LENG Yu-peng1, LI Chun-jiang2, YANG Wen-huan2, LI Wei-ping2, TANG Shi-ke2, LI Zhi-jun1*. Spectroscopic Character of River Ice Surface and Spectral Analysis of
Dissolved Organic Matter in Ice in Inner Mongolia Section of
Yellow River[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(08): 2273-2280. |
| [3] |
TANG Bin1, LUO Xi-ling1, WANG Jian-xu1, FAN Wen-qi2*, SUN Yu-yu1, LIU Jia-lu2, TANG Huan2, ZHAO Ya3*, ZHONG Nian-bing1. Hyperspectral Image Classification of Pigments Based on Multiscale
Spatial-Spectral Feature Extraction[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(08): 2364-2372. |
| [4] |
LI Ze-qi1, YANG Zheng1, ZHOU Zhuang-fei1, PENG Ji-yu1, ZHU Feng-le1*, HE Qing-hai2. Detection of Moisture Content in Strawberry Leaves Using Hyperspectral and Broad Learning System[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(06): 1534-1542. |
| [5] |
WU Yun-long1, XING Li-teng2, 3, SUN Ai-yun1, CHENG Can3, 4, YIN Li-lei1, JIA Wen-bao1*. A Method for Predicting the Emission Spectrum of High-Power X-Ray Tubes Based on Neural Network Models[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(06): 1657-1662. |
| [6] |
PENG Jian-heng1, HU Xin-jun1, 2*, ZHANG Jia-hong1, TIAN Jian-ping1, CHEN Man-jiao1, HUANG Dan2, LUO Hui-bo2. Identification and Adulteration Detection of Lotus Root Starch Using
Hyperspectral Imaging Technology Combined With Deep Learning[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(06): 1759-1767. |
| [7] |
DENG Ying-jiao1, CHEN Jun2, WANG Jian-sheng1, HU Liu-ping3, ZHANG Qing1, DU Yu-zhen3, WANG Yan1, LI Qing-li1*. Analysis of Urine Sediment Samples Based on Microscopy Hyperspectral Imaging Technology[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(05): 1243-1250. |
| [8] |
WU Nian-yi1, CANG Hao1, GAO Xiu-wen1, LI Yong-quan1, TAN Fei1, DI Ruo-yu1, RUAN Shi-wei1, GAO Pan1*, LÜ Xin2*. Cotton Verticillium Wilt Severity Detection Based on Hyperspectral
Imaging and SSFNet[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(05): 1300-1309. |
| [9] |
WANG Jin-zhi1, ZHOU Guang-sheng2*, LÜ Xiao-min2, REN Hong-rui1. Research on Remote Sensing Extraction of Artificial Surface in Lhasa City Based on Spectral Features[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(04): 1061-1070. |
| [10] |
ZHANG Fu1, WANG Meng-yao1, YAN Bao-ping1, ZHANG Fang-yuan1, YUAN Ye1, ZHANG Ya-kun1, FU San-ling2*. Hyperspectral Imaging Combined With ELM for Eggs Variety
Identification[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(03): 836-841. |
| [11] |
ZHAO Xin1, 4, SHI Yu-na1, LIU Yi-tong1, JIANG Hong-zhe2, CHU Xuan3, ZHAO Zhi-lei1, 4, WANG Bao-jun1, 4*, CHEN Han1. Key Feature Analysis in Identification and Authenticity of Ziziphi Spinosae Semen by Using Hyperspectral Images Based on 1DCNN and PLSDA[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(03): 869-877. |
| [12] |
WU Chang-yu1, DAI Jing-jing1, 2*, SONG Yang1, 2, CHEN Wei1, 2, LIU Zhi-bo1, LIU Hong-cheng3, BAI Long-yang1. Study on Core Hyperspectral Imaging and Its Significance in Exploration: A Case Study of Bangpu Large Polymetallic Deposit in Tibet[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(01): 38-44. |
| [13] |
XU Yang1, MAO Yi-lin1, LI He1, WANG Yu1, WANG Shuang-shuang2, QIAN Wen-jun1, DING Zhao-tang2*, FAN Kai1*. Multispectral and Hyperspectral Prediction Models of REC, SPAD and MDA in Overwintered Tea Plant[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2025, 45(01): 256-263. |
| [14] |
LI Ling-qiao1, WANG Zhuo-jian1, CHEN Jiang-hai1, LU Feng1, HUANG Dian-gui2, YANG Hui-hua3, LI Quan2*. A Model Transfer Method Based on Transfer Component Analysis and
Direct Correction[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2024, 44(12): 3399-3405. |
| [15] |
CHEN Cheng1, YAN Bing1, YIN Zuo-wei1*, CAO Wei-yu2, WANG Wen-jing1. Study on the Spectrum and Visualization of “Trapiche” Tourmaline Based on Hyperspectral Imaging Technology[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2024, 44(12): 3429-3434. |
|
|
|
|
