基于联合剪枝深度模型压缩的种子分选方法研究
董燕
李环宇
李卫杰
李春雷
刘洲峰
中原工学院电子信息学院,河南郑州450007
摘要：现有基于深度学习的种子分级分选方法已取得了理想的识别效果,但由于分选方法的性能依赖于模型的宽度和深度,导致模型参数量显著增加,使其难以部署在实际应用中资源受限的边缘设备上.为此,提出了 一种通道和卷积层联合剪枝的深度模型压缩方法,在通道剪枝阶段,将稀疏正则化训练的BN层参数作为衡量通道重要性的指标,在不损失模型精度的前提下,实现最大限度的通道剪枝压缩.然后提出基于线性探针的层剪枝方法,在压缩模型的同时减少内存访问,从而提升模型推理速度.最后采用知识蒸馏技术对剪枝网络进行知识迁移,补偿网络因剪枝而造成的精度损失.结果表明,在红芸豆和玉米种子数据集上所提出的方法使模型计算量减少86.55％和91.55％情况下,分别实现了实际推理速度2.1倍和2.8倍的提升,且仍保持较好的识别准确度(97.38％和96.56％),为模型在实际种子分选系统的部署提供技术支撑.
关键词：种子分级分选计算机视觉深度学习模型压缩层剪枝知识蒸馏
分类号：S126(农业物理学)
资助基金：国家自然科学基金(U1804157)国家自然科学基金(2072489)国家自然科学基金(61772576)河南省教育厅科技创新团队项目(21IRTSTHN013)
论文发表日期：2022-01-15
在线出版日期：2026-09-12（本平台首次上网日期，不代表文献的发表时间）
页数：9( 162-170 )
英文信息