
Mineral image recognition based on progressive deep learning across different granularity levels
Chengzhou WAN, Xiaohui JI, Mei YANG, Mingyue HE, Zhaochong ZHANG, Shan ZENG, Yuzhu WANG
Mineral image recognition based on progressive deep learning across different granularity levels
In recent years mineral image recognition has become increasingly important for mineral identification with the use of deep learning. While such application has gained some success, further improvement is needed to enhance the identification accuracy on large-scale mineral datasets. The fine differences in morphology, texture, and color between different minerals may align with the characteristics of fine-grained recognition algorithms, yet results of fine-grained recognition for mineral identification have rarely been reported. This paper proposes a fine-grained mineral identification model based on Next-ViT, which allows precise classification of mineral images by progressive model training across different granularity levels. In this approach, Next-ViT, which combines the advantages of transformer and convolutional neural network, is utilized to extract rich image features; a random jigsaw generator is then employed to create mineral puzzles at different granularity levels encompassing various information from detailed to general. The model training involves progressive learning across multiple granularity levels. In the early stages, the model primarily focuses on fine-grained features, learning detailed information from the puzzles to differentiate between different minerals; as training progresses, model learning gradually shifts to higher granularity levels, capturing more abstract and global information. Through this approach, the model can effectively utilize information across multiple granularity levels, thereby improving the accuracy of mineral identification. Experimental results demonstrated the effectiveness of this approach, with the proposed model achieving an accuracy of 86.5% in mineral identification on a dataset on 36 common minerals.
mineral identification / deep learning / Next-ViT / fine-grained identification / progressive multi-granularity-level training
[1] |
郝慧珍, 顾庆, 胡修棉. 基于机器学习的矿物智能识别方法研究进展与展望[J]. 地球科学, 2021, 46(9): 3091-3106.
|
[2] |
|
[3] |
徐述腾, 周永章. 基于深度学习的镜下矿石矿物的智能识别实验研究[J]. 岩石学报, 2018, 34(11): 3244-3252.
|
[4] |
周永章, 左仁广, 刘刚, 等. 数学地球科学跨越发展的十年: 大数据、人工智能算法正在改变地质学[J]. 矿物岩石地球化学通报, 2021, 40(3): 556-573.
|
[5] |
周永章, 张良均, 张奥多, 等. 地球科学大数据挖掘与机器学习[M]. 广州: 中山大学出版社, 2018.
|
[6] |
|
[7] |
郭艳军, 周哲, 林贺洵, 等. 基于深度学习的智能矿物识别方法研究[J]. 地学前缘, 2020, 27(5): 39-47.
|
[8] |
|
[9] |
彭伟航, 白林, 商世为, 等. 基于改进InceptionV3模型的常见矿物智能识别[J]. 地质通报, 2019, 38(12): 2059-2066.
|
[10] |
杨彪, 马亦骥, 倪瑞璞, 等. 基于多尺度密集连接网络的矿物图像智能识别[J]. 云南大学学报(自然科学版), 2022, 44(6): 1118-1126.
|
[11] |
杨彪, 倪瑞璞, 高皓, 等. 基于多分辨率图像的矿物特征自动提取与矿物智能识别模型[J]. 有色金属工程, 2022, 12(5): 84-93.
|
[12] |
|
[13] |
矿物数据库[EB/OL]. [2024-04-24]. https://www.mindat.org/.
|
[14] |
|
[15] |
马瑶, 智敏, 殷雁君, 等. CNN和Transformer在细粒度图像识别中的应用综述[J]. 计算机工程与应用, 2022, 58(19): 53-63.
|
[16] |
李祥霞, 吉晓慧, 李彬. 细粒度图像分类的深度学习方法[J]. 计算机科学与探索, 2021, 15(10): 1830-1842.
|
[17] |
|
[18] |
|
[19] |
|
[20] |
|
[21] |
|
[22] |
|
[23] |
|
/
〈 |
|
〉 |