为提高煤岩镜质组显微组分的识别准确率和识别模型的性能，减少识别模型训练中的人工干预，以轻量级网络模型ShuffleNet V2作为主干网络，提出一种嵌入改进注意力机制的煤岩镜质组显微亚组分识别轻量级深度学习网络模型。首先针对镜质组样本数据量较少的问题，采用随机裁剪、旋转、镜像及加噪等方法对初始样本数据进行增强，提高训练模型泛化能力；然后以大型数据集ImageNet上完成预训练的ShuffleNet V2模型为基础，采用迁移学习的方法对模型进行微调，即在本文增强的镜质组显微图像训练样本集上将预训练模型中靠近输入的若干层冻结，对靠近输出的网络层权值进行微调，生成深层特征提取层；最后在ShuffleNet V2模型的输出部分嵌入通道注意力机制ECA，并根据煤岩镜质组不同显微亚组分图像具有明显纹理差异的特点对注意力机制进行改进，构建端到端的轻量级网络识别模型，实现对煤岩镜质组7类显微亚组分的自动识别。实验结果表明：采用本文方法对煤岩镜质组显微亚组分进行识别，其平均准确率可达97.85%，较之ShuffleNet V2原模型可提升5.71%；与经典的神经网络相比，本文模型具有较高的识别准确率、较少的网络参数和计算量及较快的收敛速度；与其它轻量级网络及嵌入其它注意力机制相比，该网络在保持较少参数量的同时，识别的准确率有明显的提高。

1 数据集
1.1样本特征分析
1.2样本数据及增强
2.网络结构及算法
2.1 基础网络结构
2.2 迁移学习
2.3 注意力机制
3 实验与结果分析
3.1环境配置及参数设置
3.2评价指标
3.3实验结果分析
3.4 实验对比与分析
3.4.1 不同网络模型结果对比
3.4.2不同注意力机制对网络模型整体性能的影响
4 结论

王培珍,杨志豪,汪正才,薛子邯,刘林,张代林.嵌入改进注意力机制的镜质组显微亚组分轻量级网络识别模型[J/OL].煤炭学报:1-8[2023-06-24].DOI:10.13225/j.cnki.jccs.2022.1452.