ETRI Journal (Aug 2023)

A novel MobileNet with selective depth multiplier to compromise complexity and accuracy

  • Chan Yung Kim,
  • Kwi Seob Um,
  • Seo Weon Heo

DOI
https://doi.org/10.4218/etrij.2022-0103
Journal volume & issue
Vol. 45, no. 4
pp. 666 – 677

Abstract

Read online

In the last few years, convolutional neural networks (CNNs) have demonstrated good performance while solving various computer vision problems. However, since CNNs exhibit high computational complexity, signal processing is performed on the server side. To reduce the computational complexity of CNNs for edge computing, a lightweight algorithm, such as a MobileNet, is proposed. Although MobileNet is lighter than other CNN models, it commonly achieves lower classification accuracy. Hence, to find a balance between complexity and accuracy, additional hyperparameters for adjusting the size of the model have recently been proposed. However, significantly increasing the number of parameters makes models dense and unsuitable for devices with limited computational resources. In this study, we propose a novel MobileNet architecture, in which the number of parameters is adaptively increased according to the importance of feature maps. We show that our proposed network achieves better classification accuracy with fewer parameters than the conventional MobileNet.

Keywords