计算机工程

作者：张江永, 徐智勇, 张建林, 许涛

单位：1. 中国科学院光电技术研究所, 成都 610209;2. 中国科学院大学 电子电气与通信工程学院, 北京 100049;3. 中国船舶工业系统工程研究院, 北京 100036

关键词：卷积神经网络,敏感度,集成剪枝算法,YOLO网络,重要性评价

基金：国家重点研发计划（G158207）。

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

来源：2021年第9期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第9期