Design of Lightweight Convolutional Neural Network Based on Dimensionality Reduction Module

2019 The 5th International Conference on Electrical Engineering, Control and Robotics

无线电电子学;计算机科学

Zhou, Yue^1 ; Feng, Yanyan^1 ; Zeng, Shangyou^1 ; Pan, Bing^1

College of Electronic Engineering, Guangxi Normal University, Guilin

541000, China^1

关键词: Classification accuracy;    Convolution kernel;    Convolution neural network;    Convolutional neural network;    Dimension reduction;    Dimensionality reduction;    Feature acquisition;    Recognition algorithm;   

Others  :  https://iopscience.iop.org/article/10.1088/1757-899X/533/1/012045/pdf
DOI  :  10.1088/1757-899X/533/1/012045

学科分类：计算机科学（综合）

来源: IOP

【 摘 要 】

The traditional CNN extracted image features is insufficient, so the classification accuracy is not ideal. It also brought overmuch model parameters and calculation. Based on these problems, this paper proposes a dimension reduction residual module. First, reducing the dimension of the output feature map. Then, feature extraction used two different sets of convolution kernels. It can get more sufficient and different characteristic information. Two sets constitute the cascaded layer. Finally, the cascaded layer act as the input of the next layer. This module can reduce parameters. Meanwhile, it increases the depth of the network and enriches the diversity of feature acquisition. A new convolution neural network is build through this module. The performance of new network and other recognition algorithms is compared on GTSRB and 101-food datasets. The new network model is reduced to about 6.2MB, and the classification accuracy can reach 98.2% on GTSRB, 72.3% on 101-food. The experimental results show that this module can effectively improve network performance and control model size.