基于敏感度的YOLO网络集成剪枝算法

作者：张江永; 徐智勇^*; 张建林; 许涛

来源：计算机工程, 2021, 47(09): 59-68.

DOI：10.19678/j.issn.1000-3428.0058784

• 卷积神经网络
• 敏感度
• 集成剪枝算法
• YOLO网络
• 重要性评价

深层卷积神经网络所需的计算量和存储空间严重制约了其在资源有限平台上的应用与部署。针对基于单一参数重要性评价或者特征重建的剪枝算法泛化能力较差的问题,提出基于敏感度的集成剪枝算法,利用BN层的缩放因子稀疏YOLO网络中卷积核个数较多的冗余层,结合3种参数重要性评价方法对卷积核做重要性排序,并根据敏感度确定每一层的剪枝比率。实验结果表明,该剪枝算法对于YOLOv3和YOLOv3-tiny网络分别缩减80.5%和92.6%的参数量,并且相比基于网络轻量化方法的剪枝算法提升了网络模型压缩后的检测精度和泛化能力。

• 单位
  中国船舶工业系统工程研究院; 中国科学院光电技术研究所; 中国科学院大学