Abstract
Deep Learning is one of the machine learning area, which is widely used in recent research fields. In this, the work exhibits about working of the Convolutional Neural Networks (CNNs) for image classification. Deep learning approaches are better than the traditional learning algorithms when the data size is large because every day, a vast volume of data is accumulated everywhere. In deep learning, Convolutional Neural Network is one of the leading architecture. Convolutional Neural Network contains pre-trained models to transfer knowledge for learning the features, and such models are LeNet, AlexNet, GoogleNet, VGG16, VGG19, Resnet50, etc. These architectures are trained with a large ImageNet dataset, which contains millions of images. Moreover, these trained networks are also used to do new tasks. Among these pre-trained models, GoogleNet has less number of parameters, and this causes to reduce the computation complexity. We propose a deep network with Dimensionality Reduction Module (DRM), which works on less training data, and produce more accurate classification with minimum processing time and also a minimum number of parameters with regularization. The performance of classification, as well as training time and classification time of the proposed architecture, is measured with popular datasets such as ORL, Adience face dataset, Caltech101, and CIFAR10. The proposed architecture achieves better performance with less time when compared with the state of the work.
Similar content being viewed by others
References
Alom MZ, Taha TM, Yakopcic C, Westberg S, Sidike P, Nasrin MS, Asari VK (2018) The history began from alexnet: A comprehensive survey on deep learning approaches. arXiv preprint arXiv:1803.01164.
Anantha NL, Bathula BP (2018) Comparative study on traditional recommender systems and deep learning based recommender systems. Adv Modelling Anal A. 61:64–69. https://doi.org/https://doi.org/10.18280/ama_b.610202
Bai J, Jiang H, Li S, Ma X (2019) Nhl pathological image classification based on hierarchical local information and googlenet-based representations. BioMed Res Int. 2019. https://doi.org/https://doi.org/10.1155/2019/1065652
Bartle, A., Zheng, J. (2015). Gender classification with deep learning. In: Technical report. The Stanford NLP Group. http://cs224d.stanford.edu/reports/BartleAric.pdf
Bunrit S, Kerdprasop N, Kerdprasop K (2019) Evaluating on the transfer learning of CNN architectures to a construction material image classification task. Int J Mach Learn Comput 9:201–207. https://doi.org/10.18178/ijmlc.2019.9.2.787
Eidinger E, Enbar R, Hassner T (2014) Age and gender estimation of unfiltered faces. IEEE Trans Inf Forensics Secur 9:2170–2179. https://doi.org/10.1109/TIFS.2014.2359646
Fukushima K (1988) Neocognitron: a hierarchical neural network capable of visual pattern recognition. Neural Netw 1:119–130. https://doi.org/10.1016/0893-6080(88)90014-7
Gupta BB, Agrawal DP, Yamaguchi S (2019) Deep learning models for human centered computing in fog and mobile edge networks. J Ambient Intell Human Comput Springer 10:2907–2911. https://doi.org/10.1007/s12652-018-0919-8
He K, Zhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. In: Proceedings of the IEEE conference on computer vision and pattern recognition. 770–778. https://openaccess.thecvf.com/content_cvpr_2016/papers/He_Deep_Residual_Learning_CVPR_2016_paper.pdf
Hussain M, Bird JJ, Faria DR (2018) A study on cnn transfer learning for image classification. In: UK workshop on computational intelligence, 840:191–202. Springer, Cham. https://doi.org/https://doi.org/10.1007/978-3-319-97982-3_16
Kasikrit (2019) [Online AT & T database] https://www.kaggle.com/kasikrit/att-database-of-faces/activity
Khan A, Sohail A, Zahoora U, Qureshi AS (2019) A survey of the recent architectures of deep convolutional neural networks. arXiv 2019. arXiv preprint arXiv:1901.06032.
Krishna ST, Kalluri HK (2019) Deep learning and transfer learning approaches for image classification. Int J Recent Technol Eng (IJRTE), 7:427–432. https://www.ijrte.org/wp-content/uploads/papers/v7i5s4/E10900275S419.pdf
Alex Krizhevsky (2009) CIFAR10 and CIFAR100 data sets: https://www.cs.toronto.edu/~kriz/cifar.html
Krizhevsky, A., Sutskever, I., Hinton, G. E. (2012). Imagenet classification with deep convolutional neural networks. In: Advances in neural information processing systems, 1097–1105. http://papers.nips.cc/paper/4824-imagenet-classification-with-deep-convolutional-neural-networks.pdf
Vihar Kurama (2020) [GoogleNet Architecture] https://blog.paperspace.com/popular-deep-learning-architectures-alexnet-vgg-googlenet/
LeCun Y, Bottou L, Bengio Y, Haffner P (1998) Gradient-based learning applied to document recognition. Proc IEEE 86:2278–2324. https://doi.org/10.1109/5.726791
Levi, G., Hassner, T. (2015). Age and gender classification using convolutional neural networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition workshops, 34–42.
Li Y, Zhang H, Xue X, Jiang Y, Shen Q (2018) Deep learning for remote sensing image classification: a survey. Wiley Interdisciplinary Reviews 8:e1264. https://doi.org/10.1002/widm.1264
Lin, M., Chen, Q., Yan, S. (2013). Network in network. 1–10. arXiv preprint arXiv:1312.4400.
Nilsback ME, Zisserman A (2008) Automated flower classification over a large number of classes. In 2008 sixth Indian conference on computer vision, graphics & image processing, 722–729. IEEE. doi: https://doi.org/10.1109/ICVGIP.2008.47
Pan SJ, Yang Q (2010) A survey on transfer learning. IEEE Trans Knowl Data Eng 22:1345–1359. https://doi.org/10.1109/TKDE.2009.191
Prabhu (2018), Architecture of Convolutional Neural Network, https://medium.com/@RaghavPrabhu/understanding-of-convolutional-neural-network-cnn-deep-learning-99760835f148
Sajja TK, Devarapalli RM, Kalluri HK (2019) Lung cancer detection based on CT scan images by using deep transfer learning. Traitement du Signal 36:339–344. https://doi.org/10.18280/ts.360406
Sharma N, Jain V, Mishra A (2018) An analysis of convolutional neural networks for image classification. Proc Comput Sci 132:377–384. https://doi.org/10.1016/j.procs.2018.05.198
Simonyan K, Zisserman A (2014) Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556.
Sudha KK, Sujatha P (2019) A qualitative analysis of googlenet and alexnet for fabric defect detection. Int J Recent Technol Eng 8:86–92
Sultana F, Sufian A, Dutta P (2018) Advancements in image classification using convolutional neural network. In: 2018 fourth international conference on research in computational intelligence and communication networks (ICRCICN), IEEE. 122–129. doi: https://doi.org/10.1109/ICRCICN.2018.8718718.
Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, Rabinovich A (2015) Going deeper with convolutions. In: Proceedings of the IEEE conference on computer vision and pattern recognition, 1–9.
Szegedy C, Ioffe S, Vanhoucke V, Alemi A (2016) Inception-v4, inception-resnet and the impact of residual connections on learning. In: Association for the advancement of artificial intelligence (AAAI), 4:12. arXiv preprint arXiv:1602.07261
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Sajja, T.K., Kalluri, H.K. Image classification using regularized convolutional neural network design with dimensionality reduction modules: RCNN–DRM. J Ambient Intell Human Comput 12, 9423–9434 (2021). https://doi.org/10.1007/s12652-020-02663-y
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s12652-020-02663-y