Abstract
Convolutional Neural Networks (CNNs) have shown enormous potential for solving multi-label image classification problems. In recent years, a lot of experimentation is done with various state-of-the-art CNN architectures. The CNN architectures have evolved to become deeper and more complex in these years. These architectures are big due to a greater number of layers and trainable parameters. However, there are many real-time applications which demand fast and accurate classification. Keeping this in consideration, a simple model inspired by Inception V7 is proposed for multi-label image classification in this work. The proposed model consists of six convolution layers including three inception blocks with one million parameters approximately, which are very few as compared to many state-of-the-art CNN models. This makes the model deployable in lightweight devices for some real-time applications. The comparison experiments with other deep state-of-the-art CNNs were carried out on image datasets from multiple domains including general benchmark datasets, medical datasets, and agricultural datasets. The model exhibits better performance on many datasets making it feasible to use in various domains for multi-label image classification.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Boutell, M.R., Luo, J., Shen, X., Brown, C.M.: Learning multi-label scene classification. Pattern Recogn. 37(9), 1757–1771 (2004)
Challenge, G.: Ocular disease intelligent recognition (ODIR-2019) (2019). https://odir2019.grand-challenge.org/. Accessed 19 Feb 2020
Chu, W.T., Guo, H.J.: Movie genre classification based on poster images with deep neural networks. In: Proceedings of the Workshop on Multimodal Understanding of Social, Affective and Subjective Attributes, pp. 39–45 (2017)
Devkar, R., Shiravale, S.: A survey on multi-label classification for images. Int. J. Comput. Appl. 162(8), 39–42 (2017)
Guan, Q., Huang, Y.: Multi-label chest x-ray image classification via category-wise residual attention learning. Pattern Recogn. Lett. 130, 259–266 (2020)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Hua, Y., Mou, L., Zhu, X.X.: Relation network for multilabel aerial image classification. IEEE Trans. Geosci. Remote Sens. 58, 4558–4572 (2020)
Huang, G., Liu, Z., Van Der Maaten, L., Weinberger, K.Q.: Densely connected convolutional networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4700–4708 (2017)
Irvin, J., et al.: CheXpert: a large chest radiograph dataset with uncertainty labels and expert comparison. Proceedings of the AAAI Conference on Artificial Intelligence, vol. 33, pp. 590–597 (2019)
Jia, D., et al.: An ensemble neural network for multi-label classification of electrocardiogram. In: Liao, H., et al. (eds.) MLMECH/CVII-STENT -2019. LNCS, vol. 11794, pp. 20–27. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-33327-0_3
Li, P., Chen, P., Xie, Y., Zhang, D.: Bi-modal learning with channel-wise attention for multi-label image classification. IEEE Access 8, 9965–9977 (2020)
Mohanty, S.P., Hughes, D.P., Salathé, M.: Using deep learning for image-based plant disease detection. Front. Plant Sci. 7, 1419 (2016)
Park, J.Y., Hwang, Y., Lee, D., Kim, J.H.: MarsNet: multi-label classification network for images of various sizes. IEEE Access 8, 21832–21846 (2020)
Planet: Planet: Understanding the Amazon from Space (2017). https://www.kaggle.com/c/planet-understanding-the-amazon-from-space/data. Accessed 10 Feb 2020
Read, J., Pfahringer, B., Holmes, G., Frank, E.: Classifier chains for multi-label classification. Mach. Learn. 85(3), 333 (2011)
Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., Chen, L.C.: MobileNetV 2: Inverted residuals and linear bottlenecks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4510–4520 (2018)
Shung, K.P.: Performance metrics (2018). https://towardsdatascience.com/accuracy-precision-recall-or-f1-331fb37c5cb9. Accessed 10 Mar 2019
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556 (2014)
Szegedy, C., Ibarz, J.: Scene classification with inception-7. In: Large-Scale Scene Understanding Challenge Workshop (lSUN), p. 5. CVPR, Boston (2015)
Szegedy, C., et al.: Going deeper with convolutions. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1–9 (2015)
Szymański, P., Kajdanowicz, T.: A scikit-based Python environment for performing multi-label classification. ArXiv e-prints, February 2017
Tahir, M.A., Kittler, J., Bouridane, A.: Multilabel classification using heterogeneous ensemble of multi-label classifiers. Pattern Recogn. Lett. 33(5), 513–523 (2012)
Tsoumakas, G., Katakis, I.: Multi-label classification: an overview. Int. J. Data Warehous. Min. (IJDWM) 3(3), 1–13 (2007)
Wei, Y., et al.: HCP: a flexible CNN framework for multi-label image classification. IEEE Trans. Pattern Anal. Mach. Intell. 38(9), 1901–1907 (2015)
Wen, S., et al.: Multilabel image classification via feature/label co-projection. IEEE Trans. Syst. Man Cybern.: Syst. (2020)
Xia, G.S., et al.: DOTA: a large-scale dataset for object detection in aerial images. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3974–3983 (2018)
Zhang, L., Shah, S.K., Kakadiaris, I.A.: Hierarchical multi-label classification using fully associative ensemble learning. Pattern Recogn. 70, 89–103 (2017)
Zhu, F., Li, H., Ouyang, W., Yu, N., Wang, X.: Learning spatial regularization with image-level supervisions for multi-label image classification. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 5513–5522 (2017)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Jain, S., Thakur, P.S., Bharti, K., Khanna, P., Ojha, A. (2021). A Lightweight Multi-label Image Classification Model Based on Inception Module. In: Singh, S.K., Roy, P., Raman, B., Nagabhushan, P. (eds) Computer Vision and Image Processing. CVIP 2020. Communications in Computer and Information Science, vol 1378. Springer, Singapore. https://doi.org/10.1007/978-981-16-1103-2_20
Download citation
DOI: https://doi.org/10.1007/978-981-16-1103-2_20
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-16-1102-5
Online ISBN: 978-981-16-1103-2
eBook Packages: Computer ScienceComputer Science (R0)