JRMGE / Vol 14 / Issue 4

Article

Deep learning of rock microscopic images for intelligent lithology identification: Neural network comparison and selection

Zhenhao Xu, Wen Ma, Peng Lin, Yilei Hua

Show More

a Geotechnical and Structural Engineering Research Center, Shandong University, Jinan, 250061, China
b School of Qilu Transportation, Shandong University, Jinan, 250061, China
c Institute of Robotics and Intelligent Systems, Wuhan University of Science and Technology, Wuhan, 430000, China


2022, 14(4): 1140-1152. doi:10.1016/j.jrmge.2022.05.009


Received: 2021-12-20 / Revised: 2022-04-08 / Accepted: 2022-05-15 / Available online: 2022-06-11

2022, 14(4): 1140-1152.

doi:10.1016/j.jrmge.2022.05.009


Received: 2021-12-20

Revised: 2022-04-08

Accepted: 2022-05-15

Available online: 2022-06-11


Abstract:

An intelligent lithology identification method is proposed based on deep learning of the rock microscopic images. Based on the characteristics of rock images in the dataset, we used Xception, MobileNet_v2, Inception_ResNet_v2, Inception_v3, Densenet121, ResNet101_v2, and ResNet-101 to develop microscopic image classification models, and then the network structures of seven different convolutional neural networks (CNNs) were compared. It shows that the multi-layer representation of rock features can be represented through convolution structures, thus better feature robustness can be achieved. For the loss function, cross-entropy is used to back propagate the weight parameters layer by layer, and the accuracy of the network is improved by frequent iterative training. We expanded a self-built dataset by using transfer learning and data augmentation. Next, accuracy (acc) and frames per second (fps) were used as the evaluation indexes to assess the accuracy and speed of model identification. The results show that the Xception-based model has the optimum performance, with an accuracy of 97.66% in the training dataset and 98.65% in the testing dataset. Furthermore, the fps of the model is 50.76, and the model is feasible to deploy under different hardware conditions and meets the requirements of rapid lithology identification. This proposed method is proved to be robust and versatile in generalization performance, and it is suitable for both geologists and engineers to identify lithology quickly.

Download PDF:


Keywords: Deep learning, Rock microscopic images, Automatic classification, Lithology identification

Show Figure(s)


Share and Cite

Zhenhao Xu, Wen Ma, Peng Lin, Yilei Hua, 2022. Deep learning of rock microscopic images for intelligent lithology identification: Neural network comparison and selection. J. Rock Mech. Geotech. Eng. 14 (4), 1140-1152.

Author(s) Information

Zhenhao Xu

✉️ zhenhao_xu@sdu.edu.cn

Prof. Zhenhao Xu received his BSc degree in Civil Engineering from China University of Geosciences, and a PhD degree in Geological Engineering from Shandong University and the University of Western Australia. After that, he conducted postdoctoral research in Geological Resources and Geological Engineering at Chinese Academy of Geological Sciences, China. He is now a professor at Shandong University, China. His research interests include (1) intelligent identification of lithology and adverse geology, disaster prevention and control in tunneling and underground engineering; (2) numerical simulation of groundwater and grouting process; (3) advanced geological prediction, and water and mud inrush in tunnels; and (4) urban underground space development and utilization.