SHN: rock image classification and feature visualization using multiple granularity spatial disorder hierarchical residual network

Abstract

The automated classification of rock images is of paramount importance in geological analysis, as it serves as the foundational criterion for the categorization of rock lithology. Despite recent advancements in leveraging deep learning technologies to enhance the efficiency and precision of image classification, a crucial aspect has been overlooked: these methods face a performance bottleneck when attempting to apply it directly to rock classification methods. To address this limitation, we propose a multiple granularity Spatial disorder Hierarchical residual Network (SHN). This approach involves learning from objects annotated at different levels, thereby facilitating the transfer of hierarchical knowledge across levels. By enabling lower-level classes to inherit pertinent attributes from higher-level superclasses, our method aims to capture the intricate hierarchical relationships among different rock types. Especially, we introduce a multi-granularity spatial disorder module to aid neural networks in discerning discriminative details across various scales. This module enables processed images to exhibit region independence, compelling the network to adeptly identify discriminative local regions at diverse granularity levels and extract pertinent features. Furthermore, in light of the absence of a comprehensive rock dataset, this study amassed 4,227 rock images of diverse compositions from various places, culminating in the creation of a robust rock dataset for classification. Rigorous experimentation on this dataset yielded highly promising results, demonstrating the effectiveness of our proposed method in addressing the challenges of rock image classification.