The Effectiveness of Data Augmentation in Image Classification Using Whale Optimization Algorithm
Abstract
Imbalanced datasets significantly degrade the performance of image classification systems, particularly in application domains where data collection is difficult, such as medical diagnosis and plant disease analysis. Data augmentation is a commonly used strategy to alleviate this problem by increasing the number of samples in minority classes. In this study, a novel image augmentation approach based on optimized edge detection, named Whale Optimization Algorithm–based Image Augmentation (WOA-IA), is proposed. Unlike conventional augmentation techniques, the proposed method generates synthetic samples by combining original images with optimally extracted edge representations. Edge detection is performed using a multi-layer filter structure whose parameters are optimized via the whale optimization algorithm by maximizing similarity with reference edge images. The resulting edge images are then integrated into an angular placement-based augmentation model. The effectiveness of the proposed WOA-IA method is evaluated on five widely used imbalanced datasets—Plant Leaves, Flowers102, Cassava, Plant Village, and HAM10000—using pre-trained CNN models including DenseNet201, MobileNet, and VGG16. Experimental results demonstrate that the proposed method consistently improves classification performance compared to both imbalanced datasets and standard edge-based augmentation techniques, particularly for datasets with high imbalance ratios.
Keywords
Image augmentation, Edge detection, Whale optimization algorithm, Imbalanced datasets
Supporting Institution
Ethical Statement
Thanks
References
- Abdel-Gawad, A. H., Said, L. A., & Radwan, A. G. (2020). Optimized edge detection technique for brain tumor detection in MR images. IEEE Access, 8, 136243–136259. https://doi.org/10.1109/ACCESS.2020.3009898
- Abramović, M., Tomasović, Ž., Saulig, N., & Marasović, I. (2023). Edge detection using vector quantization and local entropy measures applied to spectrogram component extraction. In 2023 8th International Conference on Smart and Sustainable Technologies (SpliTech) (pp. 1–4). https://doi.org/10.23919/SPLITECH58164.2023.10193655
- Ahmed, A. A., Altameemi, H. G. A., Azeez Asmael, A. A., & Al-Obaidi, M. A. (2023). An effective multiclass human skin lesion diagnosis system based on convolutional neural networks. Automatic Control and Computer Sciences, 57(2), 135–142. https://doi.org/10.3103/S0146411623020025
- Al Sahili, Z., & Awad, M. (2022). The power of transfer learning in agricultural applications: AgriNet. Frontiers in Plant Science, 13, Article 992700. https://doi.org/10.3389/fpls.2022.992700
- Chawla, N. V., Bowyer, K. W., Hall, L. O., & Kegelmeyer, W. P. (2002). SMOTE: Synthetic minority over-sampling technique. Journal of Artificial Intelligence Research, 16, 321–357. https://doi.org/10.1613/jair.953
- Chen, W., Yang, K., Yu, Z., Shi, Y., & Chen, C. L. P. (2024). A survey on imbalanced learning: Latest research, applications and future directions. Artificial Intelligence Review, 57, Article 137. https://doi.org/10.1007/s10462-024-10759-6
- Dablain, D., Krawczyk, B., & Chawla, N. V. (2022). DeepSMOTE: Fusing deep learning and SMOTE for imbalanced data. IEEE Transactions on Neural Networks and Learning Systems, 34(9), 6390–6404. https://doi.org/10.1109/TNNLS.2021.3136503
- Dakhli, R., & Barhoumi, W. (2023). A skin lesion classification method based on expanding the surrounding lesion-shaped border for an end-to-end Inception-ResNet-v2 classifier. Signal, Image and Video Processing, 17(7), 3525–3533. https://doi.org/10.1007/S11760-023-02577-0
- David, Edy Victor Haryanto, S., & Febriana. (2022). Modified local updates of the ant colony optimization algorithm for image edge detection. In 2022 10th International Conference on Cyber and IT Service Management (CITSM). https://doi.org/10.1109/CITSM56380.2022.9935991
- Delnevo, G., Girau, R., Ceccarini, C., & Prandi, C. (2022). A deep learning and social iot approach for plants disease prediction toward a sustainable agriculture. IEEE Internet of Things Journal, 9(10), 7243–7250. https://doi.org/10.1109/JIOT.2021.3097379