Convolutional Neural Network Approach to Classifying the CIFAR-10 Dataset
How can supervised machine learning be applied as a technique on a convolutional neural network to solve the image classification problem of recognizing and classifying images in the CIFAR-10 dataset?
DOI:
https://doi.org/10.47611/jsrhs.v12i2.4388Keywords:
Artificial Intelligence, Machine Learning, Image Classification, Convolutional Neural Network, Supervised Learning, CIFAR-10Abstract
Convolutional neural network (CNN) is a powerful tool that can be used in many applications of machine learning. This paper demonstrates the effectiveness of using a CNN to classify images in the CIFAR-10 dataset. The model achieved an accuracy of 0.6276 and a loss of 1.116452 on the validation set. It was observed that the accuracy of predictions varied from class to class, and this paper discusses the potential causes for this variation, such as similar classes sharing common features. Further research in this field could lead to improvement in driving assistance technology and eventually automated driving.
Downloads
References or Bibliography
Baeldung. (2022, June 20). Epoch in neural networks. Baeldung on Computer Science. Retrieved October 2022, from https://www.baeldung.com/cs/epoch-neural-networks
Baeldung. (2022, June 20). The difference between epoch and iteration in Neural Networks. Baeldung on Computer Science. Retrieved October 2022, from https://www.baeldung.com/cs/neural-networks-epoch-vs-iteration
Brownlee, J. (2021, January 26). A gentle introduction to object recognition with deep learning. Machine Learning Mastery. Retrieved September 2022, from https://machinelearningmastery.com/object-recognition-with-deep-learning/
Burns, E. (2021, March 30). What is machine learning and why is it important? SearchEnterpriseAI. Retrieved September 2022, from https://www.techtarget.com/searchenterpriseai/definition/machine-learning-ML
Krizhevsky, A. (2009). Learning Multiple Layers of Features from Tiny Images. , , 32--33.
Markham, K. (2020, February 3). Simple guide to confusion matrix terminology. Data School. Retrieved September 2022, from https://www.dataschool.io/simple-guide-to-confusion-matrix-terminology/
Selig, J. (2022, July 4). What is machine learning? A definition. expert.ai. Retrieved September 2022, from https://www.expert.ai/blog/machine-learning-definition/
Team, K. (n.d.). Keras Documentation: CIFAR10 small images classification dataset. Keras. Retrieved September 2022, from https://keras.io/api/datasets/cifar10/
Wikimedia Foundation. (2019, July 30). Contextual image classification. Wikipedia. Retrieved September 2022, from https://en.wikipedia.org/wiki/Contextual_image_classification
Published
How to Cite
Issue
Section
Copyright (c) 2023 Chaoyi Jiang; Guillermo Goldsztein
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
Copyright holder(s) granted JSR a perpetual, non-exclusive license to distriute & display this article.
