一个用于磨粒图像快速分类的轻量化CNN 模型 |
刘信良, 陈国宁, 苏化, 王静秋, 王晓雷 (2024) |
发布人:戴庆文 发布时间:2024-02-15 浏览次数:697 |
针对磨粒分析CNN 模型存在参数多、运算速度慢、难以实际应用等问题, 开展磨粒图像分类CNN 模型的轻量化研究。通过分析模型各层的参数量、运算量和剪枝敏感度, 确定卷积层4 和卷积层5 为滤波器剪枝的目标; 对卷积层4 和5 所有滤波器重要性进行计算并排序, 以75%的剪枝率去除重要性低的滤波器并重新训练, 获得轻量化模型。实验结果表明, 轻量化后的模型在保证准确率几乎不降低的情况下实现了磨粒图像的快速分类, 其理论参数量和内存占用量均减少50%以上, 运算速度提高20%以上。研究结果为CNN 模型在便携式、移动式铁谱分析设备上的应用提供参考。 |