Rock thin sections identification based on improved squeeze-and-Excitation Networks model

Abstract

Rock thin section recognition provides geological information, which is crucial in petroleum geology, exploration, and mining research as a kind of fundamental work. Although many machine learning methods solve this research, there are still problems of data hierarchy and model pertinence. We use the hierarchical classification method to divide the dataset into sedimentary, metamorphic, and igneous rock as first-level, and to subdivide a total of 105 s-level further from the three categories. We propose the MaSE-ResNeXt model based on the fundamental work in the SeNet that can enhance the feature connection between different channels. The MaSE-ResNeXt adopted hierarchical filter groups, bottleneck stacking, and other strategies to enhance the representational capability of the model which advances the solving ability in rock recognition. Six data enhancement methods and other techniques are used to improve the robustness and effectiveness. The accuracy in the test set was 90.89% and 81.97% for the first and second level, respectively, with the inference duration is only 0.0357s. This study also designs a degeneration experiment and model comparison to demonstrate the model's effectiveness. Future research can employ this model as the fundamental base of transfer learning in geology to save time in the training of study.