Enhancing image classification accuracy using convolutional neural network on CIFAR-10 dataset
| dc.contributor.author | Ghafouri, Sara | |
| dc.contributor.supervisor | Baniasadi, Amirali | |
| dc.date.accessioned | 2024-09-05T21:54:27Z | |
| dc.date.available | 2024-09-05T21:54:27Z | |
| dc.date.issued | 2024 | |
| dc.degree.department | Department of Electrical and Computer Engineering | |
| dc.degree.level | Master of Engineering MEng | |
| dc.description.abstract | This project explores the application of Convolutional Neural Networks (CNNs) for image classification on the CIFAR-10 dataset, a widely recognized benchmark in computer vision. The CIFAR-10 dataset comprises 60,000 32x32 color images across 10 distinct classes. This study aims to build and optimize a deep learning model to achieve high classification accuracy on this dataset. A CNN with multiple convolutional, pooling, and dropout layers was implemented, enhanced with batch normalization to prevent overfitting. Data augmentation techniques were applied to improve the model's generalization capabilities. The model was trained using the Adam optimizer, with callbacks for learning rate reduction and early stopping to fine-tune the training process.[1] The results demonstrate the effectiveness of deep learning techniques in achieving substantial performance on the CIFAR-10 dataset, with detailed analysis of training and validation metrics. The model was evaluated on selected data from the CIFAR-10 dataset, showcasing its predictive capabilities. The findings provide insights into the design and optimization of CNNs for practical image classification tasks. | |
| dc.description.scholarlevel | Graduate | |
| dc.identifier.uri | https://hdl.handle.net/1828/20380 | |
| dc.language.iso | en | |
| dc.rights | Available to the World Wide Web | |
| dc.subject | image classification | |
| dc.subject | CNN | |
| dc.subject | CIFAR-10 | |
| dc.subject | computer vision | |
| dc.title | Enhancing image classification accuracy using convolutional neural network on CIFAR-10 dataset | |
| dc.type | project |