Novel Rock Image Classification: The Proposal and Implementation of HKUDES_Net

Rock classification provides vital information to geosciences and geological engineering practices. Reaping the benefits of the advances of computer vision-based deep learning artificial intelligence (AI) technology, this study aims to develop a next-generation convolutional neural network (CNN) to perform automatic rock classification. Two major challenging issues have been particularly addressed. First, most of the previous rock classifications are simply transfer learning of CNNs that are trained by life-like scenarios. Second, classifying rock types with similar textures leads to severe overfitting of CNNs. In this study, a novel CNN called HKUDES_Net is proposed and implemented to classify seven common Hong Kong rock types, namely fine-grained granite, medium-grained granite, coarse-grained granite, coarse ash tuff, fine ash tuff, feldsparphyric rhyolite, and granodiorite. With the aid of dynamic expansion, squeeze and excitation, and other strategies, HKUDES_Net can classify rock types with similar texture patterns/colors but different grain sizes. As compared with the other ten landmark CNNs and seven feature-based algorithms, HKUDES_Net has the best performance in precision (90.9%), recall (90.4%), and f1-score (90.5%). By implementing the alerting level, which restricts the training loss hovering above a small constant and prevents the validation loss from rising, the overfitting has been efficiently eliminated. The proposal and implementation of HKUDES_Net highlight the value of interdisciplinary research and will continuously pave the way for better coupling AI and geosciences.