自学习稀疏密集连接卷积神经网络图像分类方法

• 作者: 吴鹏 林国强 赵振兵 郭玉荣
• 来源期刊: 信号处理
• 学科: 工学
• 关键词: 剪枝冗余通道 自学习 稀疏化密集连接卷积神经网络 图像分类
• 年，卷（期）: 2019,（10）
• 所属期刊栏目: 应用
• 页码范围: 1747-1752
• 页数: 6页
• 分类号: TN911.73
• 字数: 3547字
• 语种: 中文
• DOI: 10.16798/j.issn.1003-0530.2019.10.016

摘要

通道剪枝是深度模型压缩的主要方法之一.针对密集连接卷积神经网络中,每一层都接收其前部所有卷积层的输出特征图作为输入,但并非每个后部层都需要所有先前层的特征,网络中存在很大冗余的缺点.本文提出一种自学习剪枝密集连接网络中冗余通道的方法,得到稀疏密集连接卷积神经网络.首先,提出了一种衡量每个卷积层中每个输入特征图对输出特征图贡献度大小的方法,贡献度小的输入特征图即为冗余特征图;其次,介绍了通过自学习,网络分阶段剪枝冗余通道的训练过程,得到了稀疏密集连接卷积神经网络,该网络剪枝了密集连接网络中的冗余通道,减少了网络参数,降低了存储和计算量;最后,为了验证本文方法的有效性,在图像分类数据集CIFAR-10/100上进行了实验,在不牺牲准确率的前提下减小了模型冗余.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	赵振兵	华北电力大学电子与通信工程系	54	713	15.0	25.0
2	郭玉荣	华北电力大学电子与通信工程系	3	3	1.0	1.0