李鹏, 蒋品群, 曾上游, 夏海英, 廖志贤, 范瑞. 基于分组残差结构的轻量级卷积神经网络设计[J]. 微电子学与计算机, 2019, 36(7): 43-47.
引用本文: 李鹏, 蒋品群, 曾上游, 夏海英, 廖志贤, 范瑞. 基于分组残差结构的轻量级卷积神经网络设计[J]. 微电子学与计算机, 2019, 36(7): 43-47.
LI Peng, JIANG Pin-qun, ZENG Shang-you, XIA Hai-Ying, LIAO Zhi-xian, FAN Rui. Design of lightweight convolution neural network based on group residual structure[J]. Microelectronics & Computer, 2019, 36(7): 43-47.
Citation: LI Peng, JIANG Pin-qun, ZENG Shang-you, XIA Hai-Ying, LIAO Zhi-xian, FAN Rui. Design of lightweight convolution neural network based on group residual structure[J]. Microelectronics & Computer, 2019, 36(7): 43-47.

基于分组残差结构的轻量级卷积神经网络设计

Design of lightweight convolution neural network based on group residual structure

  • 摘要: 针对传统深度卷积神经网络参数数量过多, 很难在移动设备上应用的问题, 提出基于分组残差结构的轻量级卷积神经网络架构GResNets.利用三个卷积层的瓶颈结构将上层输出特征图分为数量相等的四组, 根据组内的瓶颈模块加入恒等映射的方式和组外相邻模块是否加入残差学习, 设计了三种轻量级卷积神经网络架构.试验阶段, 在Caltech-256, Food-101和GTSRB图像分类数据集上评测了三种网络架构的性能.实验结果表明, 与传统深度卷积神经网络相比, GResNets能在网络参数较少的情况下, 具有同样、甚至更优越的分类性能, 适合在移动设备上应用.

     

    Abstract: In order to solve the problem that the parameters of traditional deep convolution neural network are too large to be used in mobile devices, a lightweight convolution neural network architecture GResNets based on group residual structure is proposed. Using the bottleneck structure with three convolution layers, the output feature maps of the previous layer are divided into four equal groups. Three lightweight convolution neural network architectures are designed according to the way of adding identical mapping to the bottleneck module in the group and whether or not the adjacent module outside the group joins residual learning. In the experimental stage, the performance of the three network architectures was evaluated on three image recognition datasets, Caltech-256, Food-101 and GTSRB. The experimental results show that GResNets has the same or even better classification performance than traditional deep convolution neural networks under the condition of fewer network parameters, and is suitable for mobile devices.

     

/

返回文章
返回