Download pdf
Application of convolutional neural networks in image classification and applications of improved convolutional neural networks
- 1 School of Computer Science and Engineering, Central South University, Changsha, China
* Author to whom correspondence should be addressed.
Abstract
This paper reviews the application and improvement of convolutional neural networks (CNNs) in image classification. Firstly, a shallow CNN for interstitial lung disease image classification is presented. This model suppresses overfitting through a unique network architecture and optimisation algorithm. Next, the improved VGG16 architecture and MIDNet18 model are discussed and their superior performance in brain tumour image classification is demonstrated. Subsequently, a CNN-CapsNet model for cervical cancer image classification and its improvement are presented and the customised model is compared with the conventional VGG-16 CNN architecture in the paper. Next, the application of sparse convolutional kernels and hybrid sparse convolutional kernels (HDCs) in solving the problem of computational resource consumption is presented. Subsequently, methods for solving the problem of limited training data through transfer learning and network data augmentation techniques are discussed, as well as GAN-generated datasets for solving the overfitting problem. Finally, the effect of degraded images on the classification effectiveness of CNNs is explored. The results show that the improved CNN architecture and algorithms have significant effects in solving the problems of overfitting and computational resource consumption, and can significantly improve the accuracy and efficiency of image classification. And degraded images do adversely affect the accuracy of CNN for image classification.
Keywords
Convolutional neural networks, improved VGG16 architecture, MIDNet18 model, CNN-CapsNet, GAN.
[1]. Li Q, Cai W, Wang X, Zhou Y, Feng D D, Chen M 2014 In: 13th International Conference on Control Automation Robotics & Vision (ICARCV) Singapore 844-848
[2]. Mohan R, Ganapathy K, A R 2022 Brain tumour classification of magnetic resonance images using a novel CNN-based medical image analysis and detection network in comparison to VGG16 J. Popul. Ther. Clin. Pharmacol. 28 113-125
[3]. Ramya M, Kirupa G, Rama A 2022 Brain tumor classification of magnetic resonance images using a novel CNN-based medical image analysis and detection network in comparison with AlexNet J. Popul. Ther. Clin. Pharmacol. 29 97-108
[4]. Cibi A, Rose R J 2023 Classification of stages in cervical cancer MRI by customized CNN and transfer learning Cogn. Neurodyn. 17 1261-1269
[5]. Lei X, Pan H, Huang X 2019 A Dilated CNN Model for Image Classification IEEE Access 7 124087-124095
[6]. Han D, Liu Q, Fan W 2018 A new image classification method using CNN transfer learning and web data augmentation Expert Syst. Appl. 95 43-56
[7]. Lee H, Kwon H 2017 Going Deeper With Contextual CNN for Hyperspectral Image Classification IEEE Trans. Image Process. 26 4843-4855
[8]. Gulakala R, Markert B, Stoffel M 2023 Rapid diagnosis of Covid-19 infections by a progressively growing GAN and CNN optimisation Comput. Methods Programs Biomed. 229 107262
[9]. Pei Y, Huang Y, Zou Q, Zhang X, Wang S 2021 Effects of Image Degradation and Degradation Removal to CNN-Based Image Classification IEEE Trans. Pattern Anal. Mach. Intell. 43 1239-1253
Cite this article
Liu,T. (2024). Application of convolutional neural networks in image classification and applications of improved convolutional neural networks. Applied and Computational Engineering,81,56-62.
Data availability
The datasets used and/or analyzed during the current study will be available from the authors upon reasonable request.
Disclaimer/Publisher's Note
The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of EWA Publishing and/or the editor(s). EWA Publishing and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.
About volume
Volume title: Proceedings of the 2nd International Conference on Machine Learning and Automation
© 2024 by the author(s). Licensee EWA Publishing, Oxford, UK. This article is an open access article distributed under the terms and
conditions of the Creative Commons Attribution (CC BY) license. Authors who
publish this series agree to the following terms:
1. Authors retain copyright and grant the series right of first publication with the work simultaneously licensed under a Creative Commons
Attribution License that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this
series.
2. Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the series's published
version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgment of its initial
publication in this series.
3. Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and
during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See
Open access policy for details).