计算机应用研究2021,Vol.38Issue(12) :3808-3813,3830.DOI:10.19734/j.issn.1001-3695.2021.03.0128

轻量型图像分类神经网络改进研究

Research on improvement of lightweight image classification neural network

王光宇 张海涛
计算机应用研究2021,Vol.38Issue(12) :3808-3813,3830.DOI:10.19734/j.issn.1001-3695.2021.03.0128

轻量型图像分类神经网络改进研究

Research on improvement of lightweight image classification neural network

王光宇 1张海涛1
扫码查看

作者信息

  • 1. 兰州大学 信息科学与工程学院,兰州730000
  • 折叠

摘要

当前普遍使用的轻量型神经网络仍然存在计算量与参数量过大的问题,导致算力较低的廉价移动设备无法快速完成图像分类任务.针对此问题提出了一种更适合于应用在算力较低的廉价移动设备上的轻量型神经网络,引入了代价较小的线性操作与特征图合并操作用于减少神经网络的计算量与参数量,还引入了改进的残差结构、注意力机制和标签平滑技术用于提高结果判断的准确率.基于PD-38数据集的实验表明,该神经网络相比传统的轻量型神经网络使用较小的计算量与参数量可以达到较高的分类准确率.在公共数据集CIFAR-10上的实验进一步表明该神经网络具有通用性.

关键词

卷积神经网络/注意力机制/图像分类/残差网络

引用本文复制引用

出版年

2021
计算机应用研究
四川省电子计算机应用研究中心

计算机应用研究

CSTPCDCSCD北大核心
影响因子:0.93
ISSN:1001-3695
被引量2
参考文献量2
段落导航相关论文