基于MobileViT的岩石薄片图像岩性识别方法研究

作者：王琼; 杨杰; 霍凤财^*; 董宏丽; 任伟建; 于涛

来源：地质通报, 2023, 1-11.

• 岩石薄片
• 岩性识别
• MobileViT
• 余弦退火
• 轻量化

岩石薄片图像中包含了大量肉眼无法到观察的地质特征信息，对岩石薄片图像的岩性识别结果为后续的石油勘探和开发奠定了基石。针对岩性识别数据集不均衡、识别模型参数多等问题，提出一种改进的轻量化MobileViT模型，该模型针对涵盖了90%以上常见岩性的岩石薄片图像进行建模分析。首先，为使模型更好地学习到每类岩石薄片图像中所包含的独特特征，对数据集进行数字增加。其次，使用GELU替换MobileViT中MV2模块中常规ReLU6，从而作为该模块的激活函数，有效解决神经元死亡的问题，提升模型的收敛速度。最后，划分训练集和测试集，使用余弦退火算法自动更新学习率，以迁移学习加速训练过程，实现岩石薄片图像中针对岩性的自动识别。实验结果表明，改进后的MobileViT对岩性识别的准确率达82.9%，模型的参数仅为7.66M，通过实例验证该算法具有较好的鲁棒性。

• 单位
  东北石油大学