润滑与密封

作者：刘信良，陈国宁，苏化，王静秋，王晓雷

单位：南京航空航天大学,直升机动力学全国重点实验室,,南京航空航天大学,直升机动力学全国重点实验室,,南京航空航天大学,直升机动力学全国重点实验室,,南京航空航天大学,直升机动力学全国重点实验室,,南京航空航天大学,直升机动力学全国重点实验室,

关键词：铁谱分析;卷积神经网络;磨粒图像分类;轻量化

基金：直升机传动技术重点实验室基金项目（HTL-A-21G03）

针对磨粒分析CNN模型存在参数多、运算速度慢、难以实际应用等问题，开展磨粒图像分类CNN模型的轻量化研究。通过分析模型各层的参数量、运算量和剪枝敏感度，确定卷积层4和卷积层5为滤波器剪枝的目标;对卷积层4和5所有滤波器重要性进行计算并排序，以75%的剪枝率去除重要性低的滤波器并重新训练，获得轻量化模型。实验结果表明，轻量化后的模型在保证准确率几乎不降低的情况下实现了磨粒图像的快速分类，其理论参数量和内存占用量均减少50%以上，运算速度提高20%以上。研究结果为CNN模型在便携式、移动式铁谱分析设备上的应用提供参考。

来源：2024年第10期

《润滑与密封》期刊编辑部

查看润滑与密封杂志2024年第10期