Browse > Article
http://dx.doi.org/10.46670/JSST.2021.30.5.286

Lightweight image classifier for CIFAR-10  

Sharma, Akshay Kumar (Department of Electronic Engineering, Daegu Universtiy)
Rana, Amrita (Department of Electronic Engineering, Daegu Universtiy)
Kim, Kyung Ki (Department of Electronic Engineering, Daegu Universtiy)
Publication Information
Journal of Sensor Science and Technology / v.30, no.5, 2021 , pp. 286-289 More about this Journal
Abstract
Image classification is one of the fundamental applications of computer vision. It enables a system to identify an object in an image. Recently, image classification applications have broadened their scope from computer applications to edge devices. The convolutional neural network (CNN) is the main class of deep learning neural networks that are widely used in computer tasks, and it delivers high accuracy. However, CNN algorithms use a large number of parameters and incur high computational costs, which hinder their implementation in edge hardware devices. To address this issue, this paper proposes a lightweight image classifier that provides good accuracy while using fewer parameters. The proposed image classifier diverts the input into three paths and utilizes different scales of receptive fields to extract more feature maps while using fewer parameters at the time of training. This results in the development of a model of small size. This model is tested on the CIFAR-10 dataset and achieves an accuracy of 90% using .26M parameters. This is better than the state-of-the-art models, and it can be implemented on edge devices.
Keywords
Computer vision; Convolutional neural networks; Image classification; Lightweight CNN;
Citations & Related Records
연도 인용수 순위
  • Reference
1 X. Zhang, X. Zhou, M. Lin, and J. Sun. "ShuffleNet: An extremely efficient convolutional neural network for mobile devices", Proc. of IEEE Conf. on Comput. Vis. Pattern Recognit., pp. 6848-6856, Salt Lake City, Utah, 2018.
2 G. Huang, Z. Liu, L. Van Der Maaten, and K. Q. Weinberger, "Densely connected convolutional networks", Proc. of IEEE Conf. on Comput. Vis. Pattern Recognit., pp. 4700-4708, Honolulu, Hawaii, 2017.
3 W. Sun, X. Zhang, and X. He, "Lightweight image classifier using dilated and depthwise separable convolutions", J. Cloud Comp., Vol. 9, No. 1, pp. 1-12, 2020.   DOI
4 F. Sultana, A. Sufian, and P. Dutta, "Advancements in image classification using convolutional neural network", Proc. of IEEE 2018 Fourth Int. Conf. on Res. Comput. Intell. Commun. Netw., pp. 122-129, 2018.
5 A. G. Howard, M. Zhu, B. Chen, D. Kalenichenko, W. Wang, T. Weyand, M. Andreetto, H. Adam, "Mobilenets: Efficient convolutional neural networks for mobile vision applications", Proc. Conf. on Comput. Vis. Pattern Recognit., pp. 1704.04861(1)-1704.04861(9), Honolulu, Hawaii 2017.
6 R. Hecht-Nielsen, "Theory of the backpropagation neural network", Proc. of IEEE IJCNN, pp. 593-605, San Diego, CA, 1989.
7 D. H. Hubel and T. N. Wiesel, "Receptive fields and functional architecture of monkey striate cortex", J. Physiol., Vol. 195, No. 1, pp. 215-243, 1968.   DOI
8 https://www.cs.toronto.edu/~kriz/cifar.html (retrieved on Aug. 25, 2021).
9 Y. Lecun, L. Bottou, Y. Bengio, and P. Haffner, "Gradient-based learning applied to document recognition", Proc. of IEEE, Vol. 86, No. 11, pp. 2278-2324, 1998.   DOI
10 K. Fukushima, "Neocognitron: A self-organizing neural network model for a mechanism of pattern recognition unaffected by shift in position", Biol. Cybern., Vol. 36, No. 4, pp. 193-202, 1980.   DOI
11 J. Deng, W. Dong, R. Socher, L. J. Li, K. Li, and F. F. Li, "Imagenet: A large-scale hierarchical image database", Proc. of IEEE Conf. on Comput. Vis. Pattern Recognit., pp. 248-255, Miami, Florida, 2009.
12 I. N. Junejo and N. Ahmed, "Depthwise separable convolutional neural networks for pedestrian attribute recognition", SN Comput. Sci., Vol. 2, No. 2, pp. 1-11, 2021.   DOI