Journal of Jilin University(Earth Science Edition) ›› 2021, Vol. 51 ›› Issue (5): 1597-1604.doi: 10.13278/j.cnki.jjuese.20200291

Previous Articles     Next Articles

Automatic Lithology Classification Method Based on Deep Learning of Rock Sample Meso-Image

Xiong Yuehan1, Liu Dongyan1, Liu Dongsheng2, Wang Yanlei2, Tang Xiaoshan3   

  1. 1. School of Civil Engineering, Chongqing University, Chongqing 400045, China;
    2. Chongqing Bureau of Geology and Minerals Exploration, Chongqing 401121, China;
    3. School of Civil Engineering and Architecture, Chongqing University of Science and Technology, Chongqing 401331, China
  • Received:2020-12-03 Online:2021-09-26 Published:2021-09-29
  • Supported by:
    Supported by the Natural Science Foundation of Chongqing (Postdoctoral Fund) (cstc2020jcyj-bsh0137) and the Scientific Research Project of Chongqing Bureau of Geology and Minerals Exploration (DKJ-2020DZJ-A-015)

Abstract: In the current geotechnical engineering, manual identification methods are usually used to distinguish the types of rock samples, which are highly specialized, time-consuming,and susceptible to subjective factors, resulting in low accuracy. With the development of computer technology, machine learning is gradually applied to the automatic lithology identification, which opens up a new path for rock sample classification. Based on this, the authors took the meso-images of the four typical rock samples (mudstone, sandy mudstone, argillaceous sandstone, and sandstone) in the main urban area of Chongqing as the research object. Based on the Inception V3 convolutional network model and migration learning algorithm, a deep learning model of rock sample mesoscopic images was eatablished, and training and learning were completed. The results show that after 1 000 times training, the classification accuracy rate in the training set reaches 92.77%, and the classification accuracy rate in the verification set is 76.31%. Among them, the sandstone recognition accuracy rate is 97.28%, the mudstone recognition accuracy rate in the verification set is 81.85%, the argillaceous sandstone recognition accuracy rate is 72.59%, and the sandy mudstone recognition accuracy rate is 72.35%. Compared with the existing machine learning methods, this recognition model can automatically recognize the rock samples with very similar lithology, and has higher recognition accuracy, robustness, and generalization ability.

Key words: rock sample meso-image, deep learning, lithology identification, convolutional neural network, automatic classification

CLC Number: 

  • P588
[1] 陈建平, 李婧, 崔宁, 等.大数据背景下地质云的构建与应用[J]. 地质通报, 2015, 34(7):1260-1265. Chen Jianping, Li Jing, Cui Ning, et al. The Construction and Application of Geological Cloud Under the Big Data Background[J]. Geological Bulletin of China, 2015, 34(7):1260-1265.
[2] 肖克炎, 孙莉, 李楠, 等.大数据思维下的矿产资源评价[J]. 地质通报, 2015, 34(7):1266-1272. Xiao Keyan, Sun Li, Li Nan, et al. Mineral Resources Assessment Under the Thought of Big Data[J]. Geological Bulletin of China, 2015, 34(7):1266-1272.
[3] 严光生, 薛群威, 肖克炎, 等.地质调查大数据研究的主要问题分析[J]. 地质通报, 2015, 34(7):1273-1279. Yan Guangsheng, Xue Qunwei, Xiao Keyan, et al. An Analysis of Major Problems in Geological Survey Big Data[J]. Geological Bulletin of China, 2015, 34(7):1273-1279.
[4] 赵鹏大.大数据时代数字找矿与定量评价[J]. 地质通报, 2015, 34(7):1255-1259. Zhao Pengda. Digital Mineral Exploration and Quantitative Evaluation in the Big Data Age[J]. Geological Bulletin of China, 2015, 34(7):1255-1259.
[5] 刘延保, 曹树刚, 刘玉成.基于LS-SVM的岩石细观图像分析方法探讨[J]. 岩石力学与工程学报, 2008, 27(5):1059-1065. Liu Yanbao, Cao Shugang, Liu Yucheng. Discussion on Analytical Method for LS-SVM Based Mesoscopic Rock Images[J]. Chinese Journal of Rock Mechanics and Engineering, 2008, 27(5):1059-1065.
[6] Singh N, Singh T N, Tiwary A, et al. Textural Identification of Basaltic Rock Mass Using Image Processing and Neural Network[J]. Computational Geosciences, 2010, 14(2):301-310.
[7] Młynarczuk M, Górszczyk A, Ślipek B. The Application of Pattern Recognition in the Automatic Classification of Microscopic Rock Images[J]. Computers & Geosciences, 2013, 60(5):126-133.
[8] 郭超, 刘烨.多色彩空间下的岩石图像识别研究[J]. 科学技术与工程, 2014, 14(18):247-251, 255. Guo Chao, Liu Ye. Recognition of Rock Images Based on Multiple Color Spaces[J]. Science Technology and Engineering, 2014, 14(18):247-251, 255.
[9] 程国建, 杨静, 黄全舟, 等.基于概率神经网络的岩石薄片图像分类识别研究[J]. 科学技术与工程, 2013, 13(31):9231-9235. Cheng Guojian, Yang Jing, Huang Quanzhou, et al. Rock Image Classification Recognition Based on Probabilistic Neural Networks[J]. Science Technology and Engineering, 2013, 13(31):9231-9235.
[10] 刘烨, 程国建, 马微, 等.基于铸体薄片图像颜色空间与形态学梯度的岩石分类[J]. 中南大学学报(自然科学版), 2016, 47(7):2375-2382. Liu Ye, Cheng Guojian, Ma Wei, et al. Rock Classification Based on Features Form Color Space and Morphological Gradient of Rock Thin Section Image[J]. Journal of Central South University(Science and Technology), 2016, 47(7):2375-2382.
[11] 程国建, 刘丽婷.深度学习算法应用于岩石图像处理的可行性研究[J]. 软件导刊, 2016, 15(9):163-166. Cheng Guojian, Liu Liting. Feasibility Study of Deep Learning Algorithm Applied to Rock Image Processing[J]. Software Guide, 2016, 15(9):163-166.
[12] 张野, 李明超, 韩帅.基于岩石图像深度学习的岩性自动识别与分类方法[J]. 岩石学报, 2018, 34(2):333-342. Zhang Ye, Li Mingchao, Han Shuai. Automatic Identification and Classification in Lithology Based on Deep Learning in Rock Images[J]. Acta Petrologica Sinica, 2018, 34(2):333-342.
[13] 白林, 姚钰, 李双涛, 等.基于深度学习特征提取的岩石图像矿物成分分析[J]. 中国矿业, 2018, 27(7):178-182. Bai Lin, Yao Yu, Li Shuangtao, et al. Mineral Composition Analysis of Rock Image Based on Deep Learning Feature Extraction[J]. China Mining Magazine, 2018, 27(7):178-182.
[14] 白林, 魏昕, 刘禹, 等.基于VGG模型的岩石薄片图像识别[J]. 地质通报, 2019, 38(12):2053-2058. Bai Lin, Wei Xin, Liu Yu, et al. Rock Thin Section Image Recognition and Classification Based on VGG Model[J]. Geological Bulletin of China, 2019, 38(12):2053-2058.
[15] 王恒, 姜亚楠, 张欣, 等.基于梯度提升算法的岩性识别方法[J]. 吉林大学学报(地球科学版), 2021, 51(3):940-950. Wang Heng, Jiang Yanan, Zhang Xin, et al. Lithology Identification Method Based on Gradient Boosting Algorithm[J]. Journal of Jilin University(Earth Science Edition), 2021, 51(3):940-950.
[16] 刘云鹏, 郭春影, 秦明宽, 等.基于PCA-SVM算法对稀土元素与稀土判别指标耦合数据集的铀矿床分类[J]. 吉林大学学报(地球科学版), 2021, 51(3):723-733. Liu Yunpeng, Guo Chunying, Qin Mingkuan, et al. Classfication of Uranium Deposits Based on PCA-SVM Algorithm for Coupling Data Set of Rare Earth Elements and Rare Earth Discrimination Indexes[J]. Journal of Jilin University(Earth Science Edition), 2021, 51(3):723-733.
[17] Szegedy C, Liu W, Jia Y, et al. Going Deeper with Convolutions[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Boston:Hynes Convention Center, 2015:1-9.
[18] Szegedy C, Vanhoucke V, Ioffe S, et al. Rethinking the Inception Architecture for Computer Vision[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas:Caesars Palace, 2016:2818-2826.
[19] Lin M, Chen Q, Yan S C. Network in Network[EB/OL]. ArXiv Preprint, 2013:(2013-12-16)[2020-11-21]. arxiv.org/abs/1312.4400.
[20] 程国建, 范鹏召.基于深度信念网络的岩石粒度分类[J]. 西安石油大学学报(自然科学版), 2018, 33(3):107-112. Cheng Guojian, Fan Pengzhao. Analysis of Rock Granularity by Deep Belief Network[J]. Journal of Xi'an Shiyou University(Natural Science Edition), 2018, 33(3):107-112.
[21] 程国建, 马微, 魏新善, 等.基于图像处理与神经网络的岩石组构识别[J]. 西安石油大学学报(自然科学版), 2013, 28(5):105-110. Cheng Guojian, Ma Wei, Wei Xinshan, et al. Research of Rock Texture Identification Based on Image Processing and Neural Network[J]. Journal of Xi'an Shiyou University(Natural Science Edition), 2013, 28(5):105-110.
[1] Chen Yingxian, Zhu Zhe, Fu Jiepeng, Ma Huiru. Deep Hydrogeological Profile Generation Method Based on Conditional Generation Adversarial Network [J]. Journal of Jilin University(Earth Science Edition), 2026, 56(3): 975-985.
[2] Li Haigang, Wang Tao, Yang Yanwei, Dong Xuezheng, Liao Liyong, Fu Xiaodong, Liu Shuolei, Ni Yumiao. Inversion  of Ground Penetrating Radar Data for Underground Pipelines Based on Deep Learning [J]. Journal of Jilin University(Earth Science Edition), 2026, 56(3): 1026-1037.
[3] Liu Hongxue, Yang Huachao, Bian Hefang, Li Bin, Li Lei, Wang Sen.  Intelligent Extraction of Remote Sensing Image Change Patches Based on Deep Learning and Human-Computer Collaboration in Coal Mine Surface Areas [J]. Journal of Jilin University(Earth Science Edition), 2026, 56(3): 1076-1087.
[4] Hu Feiyue, Xu Haoxiang, Deng Chengjian, Yang Shuduo. Susceptibility Evaluation of Karst Collapse Based on Deep Learning Models: A Case Study of  Guangzhou-Foshan-Zhaoqing Area [J]. Journal of Jilin University(Earth Science Edition), 2026, 56(2): 584-597.
[5] Liu Tao, Liu Zongbao, Zhang Kejia, Zhang Yan, Zhang Ruixue, Liu Xiaowen, Xu Cuiyun. Thin Section Image Generation and Recognition Method of Tight Sandstone Reservoir Based on Deep Learning [J]. Journal of Jilin University(Earth Science Edition), 2026, 56(2): 724-738.
[6] Gao Quanming, Shang Fuqing, Chai Jin, Wang Yi, Sun Wei, Zhao Jing. Real-Time Localization and Tracking of Magnetic Object Motion Trajectory Based on MagTrack-CNN [J]. Journal of Jilin University(Earth Science Edition), 2025, 55(6): 2088-2099.
[7] Feng Yuanyuan, Li Chaokui, Liu Songhuan, Tian Qin. SSC-SeNet: A Semantic Segmentation Algorithm for Buildings in Surface Mining Areas by Fusing Point Cloud and Image Data [J]. Journal of Jilin University(Earth Science Edition), 2025, 55(5): 1757-1773.
[8] Qi Jiao, Cao Siyuan. Surface-Related Multiple Attenuation Based on Deep Learning with Prior Knowledge [J]. Journal of Jilin University(Earth Science Edition), 2025, 55(5): 1702-1714.
[9] Yang Xiaotian, Tan Jinlin, Yu Xin, Zhao Junzhe, Liu Ming. Ship Target Tracking Based on GAM-YOLOv8 Remote Sensing Images [J]. Journal of Jilin University(Earth Science Edition), 2025, 55(1): 328-339.
[10] Gao Kangzhe, Wang Fengyan, Liu Ziwei, Wang Mingchang. Semantic Segmentation of Remote Sensing Images Based on Improved U-Net [J]. Journal of Jilin University(Earth Science Edition), 2024, 54(5): 1752-1763.
[11] Wang Wei, Xiong Yizhou, Wang Xin. NHNet: A Novel Hierarchical Semantic Segmentation Network for Remote Sensing Images [J]. Journal of Jilin University(Earth Science Edition), 2024, 54(5): 1764-1772.
[12] Wang Tingting, Huang Zhixian, Wang Hongtao, Yang Minghao, Zhao Wanchun. Rock Thin Slice Lithology Identification Based on MobileNetV2 [J]. Journal of Jilin University(Earth Science Edition), 2024, 54(4): 1432-1442.
[13] Pang Zhichao, Xiao Hua, Mao Chenfei, Chen Guojun, Liang Wankun, Gao Ming, Zhang Xiao. Lithological Characteristics and Logging Identification Methods of Gypsum-Bearing Strata in Southern Margin Area of  Junggar Basin [J]. Journal of Jilin University(Earth Science Edition), 2024, 54(4): 1419-1431.
[14] Lin Yuzhun, Liu Zhi, Wang Shuxiang, Rui Jie, Jin Fei .  Research Progress of Road Extraction Method for Optical Remote Sensing Images Based on Convolutional Neural Network [J]. Journal of Jilin University(Earth Science Edition), 2024, 54(3): 1068-1080.
[15] Wang Xinling, Zhu Xinyi, Zhang Hongbing, Sun Bo, Xu Kexin. Lithology Identification Method for Logging While Drilling Based on Random Tree Embedding [J]. Journal of Jilin University(Earth Science Edition), 2024, 54(2): 701-708.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
[1] You Minxin,Liu Jianmin. Application Status and Progress of Isotopic Geochemistry in Research of Emeishan Large Igneous Province (ELIP)[J]. Journal of Jilin University(Earth Science Edition), 2014, 44(4): 1231 -1243 .
[2] YANG Chun-mei, LI Hong-qi,LU Da-wei,ZHANG Fang-li,GAO Yuan,SHAO Ying-chao. The Relationship Between Rock Resistivity and Water Saturation in WaterDriveOil and OilDriveWater Process[J]. J4, 2005, 35(05): 667 -671 .
[3] ZHU Hong-chen,ZHANG Jiong-fei,QUAN Heng. Two Stages of Mesozoic Lithogenesis and Mineralization in Daxing’anling Mountains[J]. J4, 2005, 35(04): 436 -0442 .
[4] ZHU Jian-wei, ZHAO Gang, LIU Bo, GUO Wei, CHENG Jun. Identification Technology and It’s Application of Well-Logging About Oil Shale[J]. J4, 2012, 42(2): 289 -295 .
[5] CHEN Li, LIANG Hai-an,ZHANG Wen-juan,RONG Fan. Application of Fuzzy Mathematics in Division of Engineering Geo-Environment of Cities-An Example of Fushun City[J]. J4, 2008, 38(5): 837 -0840 .
[6] GAO Gui-mei,SU Ke,WANG Wen-ying,GAN Shu-cai,LIU Zhao-jun. Study on Rare Earth and Trace Elements in Oil Shale Samples, Huadian, Jilin Province[J]. J4, 2006, 36(6): 974 -0979 .
[7] WU Kong-yun,JIANG Zhong-cheng,YE Ye. Influence of Different Plant Communities On Erosion Rates of Limestone Rock Blocks[J]. J4, 2007, 37(5): 967 -0971 .
[8] ZHOU Yan-zhang, CHI Bao-ming, LIU Zhong-pei. Mode of Structural Control about Groundwater Pour-in Mechanism in Xiadian Gold Deposits in Shandong Province[J]. J4, 2008, 38(2): 255 -0260 .
[9] ZHANG Yuan, LIU Lian-deng,SUN Jing-gui,CHEN Guo-hua,ZHANG Hong-xi,YAN Fu-chuan, YANG Kai-chun. Two Phase Overprinting Mineralization in Huangbuling Gold Deposit in Northwestern Jiaodong Peninsula[J]. J4, 2008, 38(1): 21 -0026 .
[10] LU Cheng-peng, SHU Long-cang, YUAN Li-bo, ZHANG Rong-rong, HUANG Bi-juan, WANG Bin-bin. Determination of Hydrogeologic Parameters of Karst Aquifer Based on Tracer Test[J]. J4, 2009, 39(4): 717 -721 .