Abstract
Convolutional neural networks (CNN) are a boon to image classification algorithms as it can learn highly abstract features and work with less parameter. Overfitting, exploding gradient, and class imbalance are the major challenges while training the model using CNN. These issues can diminish the performance of the model. Proper understanding and use of corrective measures can substantially prevent the model from these issues and can increase the efficiency of the model. In this paper the conceptual understanding of the basic CNN model along with its key layers is provided. The paper summarizes the results of training the deep learning model using CNN on publicly available datasets of cats and dogs. Finally the paper discusses various methods such as data augmentation, regularization, dropout, etc. to prevent the CNN model from overfitting problem. The paper will also help beginners to have a broad comprehension of CNN and motivate them to venture in this field.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Mohamed, A.-R., Dahl, G.E., Hinton, G.: Acoustic modeling using deep belief networks. IEEE Trans. Audio Speech Lang. Process. 20(1), 14–22 (2012)
LeCun, Y., Bengio, Y., Hinton, G.: Deep learning. Nature 521(7553), 436–444 (2015)
Hu, G., Wang, K., Peng, Y., Qiu, M., Shi, J., Liu, L.: Deep learning methods for underwater target feature extraction and recognition. Comput. Intell. Neurosci. 2018, 10 (2018). Article ID 1214301
http://ufldl.stanford.edu/tutorial/supervised/ConvolutionalNeuralNetwork/
https://medium.com/@udemeudofia01/basic-overview-of-convolutional-neural-network-cnn-4fcc7dbb4f17
https://machinelearningmastery.com/dropout-regularization-deep-learning-models-keras/
Jaswal, D., Vishvanathan, S., Soman K.P.: Image classification using convolutional neural networks. Int. J. Adv. Res. Technol. 3(6) (2014)
Goodfellow, I., Bengio, Y., Courville, A.: Deep Learning (Adaptive Computation and Machine Learning) (2016)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Joshi, S., Verma, D.K., Saxena, G., Paraye, A. (2019). Issues in Training a Convolutional Neural Network Model for Image Classification. In: Singh, M., Gupta, P., Tyagi, V., Flusser, J., Ören, T., Kashyap, R. (eds) Advances in Computing and Data Sciences. ICACDS 2019. Communications in Computer and Information Science, vol 1046. Springer, Singapore. https://doi.org/10.1007/978-981-13-9942-8_27
Download citation
DOI: https://doi.org/10.1007/978-981-13-9942-8_27
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-13-9941-1
Online ISBN: 978-981-13-9942-8
eBook Packages: Computer ScienceComputer Science (R0)