Enhancing Accuracy for Classification Using the CNN Model and Hyperparameter Optimization Algorithm

The Convolutional Neural Network (CNN) is a widely employed deep learning model, particularly effective for image recognition and classification tasks. The performance of a CNN is influenced not only by its architecture but also critically by its hyperparameters. Consequently, optimizing hyperparameters is essential for improving CNN model performance. In this study, the authors propose leveraging optimization algorithms such as Random Search, Bayesian Optimization with Gaussian Processes, and Bayesian Optimization with Treestructured Parzen Estimators to fine-tune the hyperparameters of the CNN model. The performance of the optimized CNN is compared with traditional machine learning models, including Random Forest (RF), Support Vector Classification (SVC), and K-Nearest Neighbors (KNN). Both the MNIST and Olivetti Faces datasets are utilized in this research. In the training procedure, on the MNIST dataset, the CNN model achieved a minimum accuracy of 97.85%, surpassing traditional models, which had a maximum accuracy of 97.50% across all optimization techniques. Similarly, on the Olivetti Faces dataset, the CNN achieved a minimum accuracy of 94.96%, while traditional models achieved a maximum accuracy of 94.00%. In the training-testing procedure, the CNN demonstrated impressive results, achieving accuracy rates exceeding 99.31% on the MNIST dataset and over 98.63% on the Olivetti Faces dataset, significantly outperforming traditional models, whose maximum values were 98.69% and 97.50%, respectively. Furthermore, the study compares the performance of the CNN model with three optimization algorithms. The results show that integrating CNN with these optimization techniques significantly improves prediction accuracy compared to traditional models.

Authors

Institutions

Publication Details

Journal
Indonesian Journal of Electrical Engineering and Informatics (IJEEI)
Published
2024-09-12
DOI
https://doi.org/10.52549/ijeei.v12i3.5545
Citations
2
Primary Topic
Neural Networks and Applications
Type
article
Field-Weighted Citation Impact
0.69
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Enhancing Accuracy for Classification Using the CNN Model and Hyperparameter Optimization Algorithm

Thanh Tran, Dai Nguyen Quoc
2 citations
Indonesian Journal of Electrical Engineering and Informatics (IJEEI)
Neural Networks and Applications
0.69
article

Enhancing Accuracy for Classification Using the CNN Model and Hyperparameter Optimization Algorithm

Thanh Tran, Dai Nguyen Quoc
article en
2 citations

Abstract

The Convolutional Neural Network (CNN) is a widely employed deep learning model, particularly effective for image recognition and classification tasks. The performance of a CNN is influenced not only by its architecture but also critically by its hyperparameters. Consequently, optimizing hyperparameters is essential for improving CNN model performance. In this study, the authors propose leveraging optimization algorithms such as Random Search, Bayesian Optimization with Gaussian Processes, and Bayesian Optimization with Treestructured Parzen Estimators to fine-tune the hyperparameters of the CNN model. The performance of the optimized CNN is compared with traditional machine learning models, including Random Forest (RF), Support Vector Classification (SVC), and K-Nearest Neighbors (KNN). Both the MNIST and Olivetti Faces datasets are utilized in this research. In the training procedure, on the MNIST dataset, the CNN model achieved a minimum accuracy of 97.85%, surpassing traditional models, which had a maximum accuracy of 97.50% across all optimization techniques. Similarly, on the Olivetti Faces dataset, the CNN achieved a minimum accuracy of 94.96%, while traditional models achieved a maximum accuracy of 94.00%. In the training-testing procedure, the CNN demonstrated impressive results, achieving accuracy rates exceeding 99.31% on the MNIST dataset and over 98.63% on the Olivetti Faces dataset, significantly outperforming traditional models, whose maximum values were 98.69% and 97.50%, respectively. Furthermore, the study compares the performance of the CNN model with three optimization algorithms. The results show that integrating CNN with these optimization techniques significantly improves prediction accuracy compared to traditional models.

Indonesian Journal of Electrical Engineering and Informatics (IJEEI)Vol. 12(3)
Industrial University of Ho Chi Minh City (VN)
Openalex Percentile: Top 24%
Neural Networks and Applications
0.69
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Enhancing Accuracy for Classification Using the CNN Model and Hyperparameter Optimization Algorithm — Thanh Tran, Dai Nguyen Quoc · Indonesian Journal of Electrical Engineering and Informatics (IJEEI) (2024) | TGRS Research Map | TGRS