A New Image Classification Approach via Improved MobileNet Models with Local Receptive Field Expansion in Shallow Layers.

Journal: Computational intelligence and neuroscience
Published Date:

Abstract

Because deep neural networks (DNNs) are both memory-intensive and computation-intensive, they are difficult to apply to embedded systems with limited hardware resources. Therefore, DNN models need to be compressed and accelerated. By applying depthwise separable convolutions, MobileNet can decrease the number of parameters and computational complexity with less loss of classification precision. Based on MobileNet, 3 improved MobileNet models with local receptive field expansion in shallow layers, also called Dilated-MobileNet (Dilated Convolution MobileNet) models, are proposed, in which dilated convolutions are introduced into a specific convolutional layer of the MobileNet model. Without increasing the number of parameters, dilated convolutions are used to increase the receptive field of the convolution filters to obtain better classification accuracy. The experiments were performed on the Caltech-101, Caltech-256, and Tubingen animals with attribute datasets, respectively. The results show that Dilated-MobileNets can obtain up to 2% higher classification accuracy than MobileNet.

Authors

  • Wei Wang
    State Key Laboratory of Quality Research in Chinese Medicine, Institute of Chinese Medical Sciences, University of Macau, Macau 999078, China.
  • Yiyang Hu
    College of Computer and Communication Engineering, Changsha University of Science and Technology, Changsha 410114, China.
  • Ting Zou
    Yiyang Branch, China Telecom Co., Ltd., Yiyang 413000, China.
  • Hongmei Liu
    Clinical Research Center & Division of Mood Disorders, Shanghai Mental Health Center, Shanghai Jiao Tong University School of Medicine, Shanghai 200030, China.
  • Jin Wang
    Cells Vision (Guangzhou) Medical Technology Inc., Guangzhou, China. Electronic address: wangjin@cellsvision.com.
  • Xin Wang
    Key Laboratory of Bio-based Material Science & Technology (Northeast Forestry University), Ministry of Education, Harbin 150040, China.