Computer Science and Information Technologies
Vol 6, No 2: July 2025

Optimizing EfficientNet for imbalanced medical image classification using grey wolf optimization

Khotimah, Khusnul (Unknown)
Surono, Sugiyarto (Unknown)
Thobirin, Aris (Unknown)



Article Info

Publish Date
01 Jul 2025

Abstract

The advancement of deep learning in computer vision has result in substantial progress, particularly in image classification tasks. However, challenges arise when the model is applied to small and unbalanced datasets, such as X-ray data in medical applications. This study aims to improve the classification performance of fracture X-ray images using the EfficientNet architecture optimized with grey wolf optimization (GWO). EfficientNet was chosen for its efficiency in handling small datasets, while GWO was applied to optimize hyperparameters, including learning rate, weight decay, and dropout to improve model accuracy. Random cropping, rotation, flipping, color jittering, and random erasing, were used to expand the diversity of the dataset, and class weighting is applied to overcome class imbalance. The evaluation uses accuracy, precision, recall, and F1-score metrics. The combination of EfficientNetB0 and GWO resulted in an average 4.5% improvement in model performance over baseline methods. This approach provides benefits in developing deep learning methods for medical image classification, especially in dealing with small and imbalanced datasets.

Copyrights © 2025






Journal Info

Abbrev

csit

Publisher

Subject

Computer Science & IT Engineering

Description

Computer Science and Information Technologies ISSN 2722-323X, e-ISSN 2722-3221 is an open access, peer-reviewed international journal that publish original research article, review papers, short communications that will have an immediate impact on the ongoing research in all areas of Computer ...