Automatic recognition of debris rock lithology based on unsupervised semantic segmentation

Abstract

Accurate identification of lithology in debris rock is crucial for optimizing resource development in geological exploration and the oil and gas industry. The traditional approach, which depends on experts manually analyzing remote sensing images, is not only laborious but also vulnerable to subjectivity. In contrast, supervised learning, although highly automated, is limited by the need for large-scale annotated data and sample imbalance issues. In our proposed unsupervised semantic segmentation method, automatic segmentation of rock images not only improves the efficiency and accuracy of lithology recognition but also reduces human errors, providing an effective solution for automated lithology analysis. We collected a large amount of debris rock data from the Qingshuihe-Karazha using remote sensing satellites and used an improved FCN network combined with super-pixel segmentation to generate pseudo labels instead of manual labeling, achieving unsupervised segmentation. We compared this method with traditional K-Means, ISODATA, and CNN + K-Means pseudo-label generation methods. By calculating evaluation metrics named ARE, AMI, and FMI, which are used for unsupervised semantic segmentation methods, we found that our method maintains high consistency and robustness in various image sizes, especially when the size of debris rock images is large, and its stability is superior. At the same time, we addressed the boundary issues caused by the need for block division in the lithology image of ultra-large debris rocks, as well as the problem of a large number of similar blocks after block division. The efficiency and accuracy of this method in lithology identification were determined, providing more convenient and efficient data processing methods for geological researchers.