A Novel Image Classification Approach via Dense-MobileNet Models

被引:154
作者
Wang, Wei [1 ]
Li, Yutao [1 ]
Zou, Ting [2 ]
Wang, Xin [1 ]
You, Jieyu [3 ]
Luo, Yanhong [1 ]
机构
[1] Changsha Univ Sci & Technol, Sch Comp & Commun Engn, Changsha 410114, Peoples R China
[2] China Telecom Co Ltd, Yiyang Branch, Yiyang 413000, Peoples R China
[3] Hunan Childrens Hosp, Changsha 410000, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolution;
D O I
10.1155/2020/7602384
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
As a lightweight deep neural network, MobileNet has fewer parameters and higher classification accuracy. In order to further reduce the number of network parameters and improve the classification accuracy, dense blocks that are proposed in DenseNets are introduced into MobileNet. In Dense-MobileNet models, convolution layers with the same size of input feature maps in MobileNet models are taken as dense blocks, and dense connections are carried out within the dense blocks. The new network structure can make full use of the output feature maps generated by the previous convolution layers in dense blocks, so as to generate a large number of feature maps with fewer convolution cores and repeatedly use the features. By setting a small growth rate, the network further reduces the parameters and the computation cost. Two Dense-MobileNet models, Dense1-MobileNet and Dense2-MobileNet, are designed. Experiments show that Dense2-MobileNet can achieve higher recognition accuracy than MobileNet, while only with fewer parameters and computation cost.
引用
收藏
页数:8
相关论文
共 19 条
  • [1] Chen WL, 2015, PR MACH LEARN RES, V37, P2285
  • [2] Denil Misha, 2013, NIPS, P2148
  • [3] Denton E, 2014, ADV NEUR IN, V27
  • [4] Licensed-Assisted Access for LTE in Unlicensed Spectrum: A MAC Protocol Design
    Han, Shiying
    Liang, Ying-Chang
    Chen, Qian
    Soong, Boon-Hee
    [J]. 2016 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2016,
  • [5] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [6] Hinton Geoffrey, 2015, ARXIV
  • [7] Howard A. G., 2017, ABS170404861 CORR, DOI DOI 10.48550/ARXIV.1704.04861
  • [8] Densely Connected Convolutional Networks
    Huang, Gao
    Liu, Zhuang
    van der Maaten, Laurens
    Weinberger, Kilian Q.
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 2261 - 2269
  • [9] Iandola F. N., 2016, ARXIV
  • [10] ImageNet Classification with Deep Convolutional Neural Networks
    Krizhevsky, Alex
    Sutskever, Ilya
    Hinton, Geoffrey E.
    [J]. COMMUNICATIONS OF THE ACM, 2017, 60 (06) : 84 - 90