Optimizing Convolutional Neural Network Hyperparameters for MNIST Image Classification: Insights and Implications
The recognition of handwritten digits has been among the most enduring fundamental problems explored in the field of machine learning and computer vision. The objective of this work is to design a state-of-art Convolutional Neural Network (CNN) for the image classification on the MNIST dataset to resolve issues related to data pre-processing and parameter tuning effectively. The work starts with standardizing the image data and then using one-hot encoding for the models that will improve the generalization and representation. The CNN structure followed in this work includes the convolutional layer, batch normalization, pooling layers, and fully connected layers with an efficient integration of features. To measure such performance indicators as accuracy, the given dataset is divided for training and validation. The results are also preprocessed to control for overfitting and convergence using back propagation with early stop to stop training when significant accuracy improvement is not observed and learning rate decay to encourage the model to search for better weights in the final stages of training. The model’s performance is tested and evaluated by using different epoch intervals and learning rates, and it is shown how critical it is to select the correct default hyperparameters for achieving efficiency. This paper also presents the approach for the practical application of the preprocessing of images and the systematic assessment in enhancing CNN-based handwritten digit recognition.
Authors
- Muhammad Zulkifl Hasan (ORCID: https://orcid.org/0000-0002-2733-5527)
- Ijaz Hussain (ORCID: https://orcid.org/0000-0003-3254-1680)
- T Chinwalla Asif
- Eman Dar
- Muhammad Zunnurain Hussain (ORCID: https://orcid.org/0000-0002-6071-1029)
- Khadija Rana
- Hoor Fatima Yousaf
Institutions
- Universiti Putra Malaysia (MY)
- Information Technology University (PK)
- Bahria University (PK)
Publication Details
- Published
- 2024-11-28
- DOI
- https://doi.org/10.1109/icaiccit64383.2024.10912098
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00