Research of Lightweight Convolution Neural Network Based on Feature Expansion Convolution

被引:0
|
作者
Xin-Zheng X.U. [1 ,2 ]
Shan L.I. [1 ]
机构
[1] School of Computer Science and Technology, China University of Mining and Technology, Jiangsu, Xuzhou
[2] Engineering Research Center of Mining Digital Ministry of Education, Jiangsu, Xuzhou
来源
Tien Tzu Hsueh Pao/Acta Electronica Sinica | 2023年 / 51卷 / 02期
基金
中国国家自然科学基金;
关键词
convolutional neural network; deep learning; feature expansion convolution; feature reuse; image classification; lightweight;
D O I
10.12263/DZXB.20210559
中图分类号
学科分类号
摘要
This paper starts with the network structure of convolution neural network model, and uses the idea of fea⁃ ture reuse to design an efficient feature expansion convolution module. The module reduces the number of output channels of standard convolution module and introduces multi branch structure. Through the cheap operation on each branch, the out⁃ put feature map of standard convolution operation is transformed and fused to generate a new feature map. The final output of the module is obtained by merging the feature graphs generated on each branch. The feature expansion convolution mod⁃ ule uses the idea of feature reuse to reuse the features in the model, which not only reduces the calculation of the model, but also enriches the hidden information of the feature graph and improves the performance of the model. Finally, the feature ex⁃ pansion convolution module is used to replace the standard convolution module, and the lightweight VGG16 (Visual Geom⁃ etry Group 16-Layer) model and residual structure are designed, and good classification results are achieved on CIFAR and ILSVRC2012 (ImageNet Large Scale Visual Recognition Challenge 2012) datasets. © 2023 Chinese Institute of Electronics. All rights reserved.
引用
收藏
页码:355 / 364
页数:9
相关论文
共 36 条
  • [1] ZHOU F Y, JIN L P, DONG J., Review of convolutional neural network, Chinese Journal of Computers, 40, 6, pp. 1229-1251, (2017)
  • [2] LU H T, ZHANG Q C., Applications of deep convolutional neural network in computer vision, Journal of Data Ac⁃ quisition and Processing, 31, 1, pp. 1-17, (2016)
  • [3] KRIZHEVSKY A, SUTSKEVER I, HINTON G E., Ima⁃ geNet classification with deep convolutional neural net⁃ works, Communications of the ACM, 60, 6, pp. 84-90, (2017)
  • [4] QUAN Y, LI Z X, ZHANG C L, Et al., Fusing deep dilated convolutions network and light-weight network for object detection, Acta Electronica Sinica, 48, 2, pp. 390-397, (2020)
  • [5] LAI X B, XU M S, XU X M., Glioblastoma multiforme multi-modal MR images segmentation using multi-class CNN, Acta Electronica Sinica, 47, 8, pp. 1738-1747, (2019)
  • [6] WANG Q, TAN J., Optical ultra-precision detection tech⁃ nology based on artificial intelligence technology, Laser Journal, 42, 2, pp. 156-160, (2021)
  • [7] ZHAO Y X, WU T, HAN Y., Identifying the correctness of fit of internal components based on a convolutional neural network, Acta Electronica Sinica, 46, 8, pp. 1983-1988, (2018)
  • [8] GUO M, ZHANG J Y., Computation offloading policy for machine learning in mobile edge computing environments, Journal of Computer Applications, 41, 9, pp. 2639-2645, (2021)
  • [9] VICENTE S, CARREIRA J, AGAPITO L, Et al., Recon⁃ structing pascal voc, 2014 IEEE Conference on Com⁃ puter Vision and Pattern Recognition, pp. 41-48, (2014)
  • [10] LIN T Y, MAIRE M, BELONGIE S, Et al., Microsoft CO⁃ CO: Common objects in context, European Confer⁃ ence on Computer Vision, pp. 740-755, (2014)