兵器装备工程学报

文章详情

稿件标题: 基于GA的卷积神经网络结构化剪枝算法
稿件作者: 贺天宇,田宗浩,张航
DOI: 10.11809/bqzbgcxb2022.02.026
科学编辑: 田鹏辉 博士(西安工业大学副教授)
栏目名称: 信息科学与控制工程
关键词: 弹载图像;卷积神经网络;剪枝;遗传算法;目标检测
文章摘要: 为解决传统神经网络模型过参数化问题,提高深度学习模型工程化实现的效率,提出了基于GA的结构化模型压缩算法。该方法从全局搜索空间对模型各层卷积核进行结构化剪枝,并以“高检测精度,低网络规模”为准则建立适应度函数,解决传统权重剪枝易陷入局部最优和剪枝结果对硬件平台不友好问题,获得精度损失低、模型压缩率高的轻量化模型结构。
稿件基金: 军队“十三五”预研基金项目
引用本文格式: 贺天宇,田宗浩,张航.基于GA的卷积神经网络结构化剪枝算法[J].兵器装备工程学报,2022,43(02):170-175.
HE Tianyu, TIAN Zonghao, ZHANG Hang.Structured pruning algorithm of CNN based on genetic algorithm[J].Journal of Ordnance Equipment Engineering,2022,43(02):170-175.
刊期名称: 2022年02期
出版时间: 2022年2月
上线时间: 2022年2月28日
浏览次数: 2566
下载次数: 870
免费阅读PDF 在线阅读 下载本期目录 下载本期封面