» Articles » PMID: 38689992

Adaptive Habitat Biogeography-based Optimizer for Optimizing Deep CNN Hyperparameters in Image Classification

Overview
Journal Heliyon
Specialty Social Sciences
Date 2024 May 1
PMID 38689992
Authors
Affiliations
Soon will be listed here.
Abstract

Deep Convolutional Neural Networks (DCNNs) have shown remarkable success in image classification tasks, but optimizing their hyperparameters can be challenging due to their complex structure. This paper develops the Adaptive Habitat Biogeography-Based Optimizer (AHBBO) for tuning the hyperparameters of DCNNs in image classification tasks. In complicated optimization problems, the BBO suffers from premature convergence and insufficient exploration. In this regard, an adaptable habitat is presented as a solution to these problems; it would permit variable habitat sizes and regulated mutation. Better optimization performance and a greater chance of finding high-quality solutions across a wide range of problem domains are the results of this modification's increased exploration and population diversity. AHBBO is tested on 53 benchmark optimization functions and demonstrates its effectiveness in improving initial stochastic solutions and converging faster to the optimum. Furthermore, DCNN-AHBBO is compared to 23 well-known image classifiers on nine challenging image classification problems and shows superior performance in reducing the error rate by up to 5.14%. Our proposed algorithm outperforms 13 benchmark classifiers in 87 out of 95 evaluations, providing a high-performance and reliable solution for optimizing DNNs in image classification tasks. This research contributes to the field of deep learning by proposing a new optimization algorithm that can improve the efficiency of deep neural networks in image classification.

Citing Articles

Optimizing VGG16 deep learning model with enhanced hunger games search for logo classification.

Hussain M, Thaher T, Almourad M, Mafarja M Sci Rep. 2024; 14(1):31759.

PMID: 39738231 PMC: 11685449. DOI: 10.1038/s41598-024-82022-5.

References
1.
Viswanathan G, Buldyrev S, Havlin S, da Luz M, Raposo E, Stanley H . Optimizing the success of random searches. Nature. 1999; 401(6756):911-4. DOI: 10.1038/44831. View

2.
Hwang J, Lustig N, Jung M, Lee J . Autoencoder and restricted Boltzmann machine for transfer learning in functional magnetic resonance imaging task classification. Heliyon. 2023; 9(7):e18086. PMC: 10372668. DOI: 10.1016/j.heliyon.2023.e18086. View

3.
Wong N, Bahmani H . A review of the current state of research on artificial blue light safety as it applies to digital devices. Heliyon. 2022; 8(8):e10282. PMC: 9420367. DOI: 10.1016/j.heliyon.2022.e10282. View

4.
Sun Y, Xue B, Zhang M, Yen G, Lv J . Automatically Designing CNN Architectures Using the Genetic Algorithm for Image Classification. IEEE Trans Cybern. 2020; 50(9):3840-3854. DOI: 10.1109/TCYB.2020.2983860. View

5.
Zheng X, Zhang Y, Hong S, Li H, Tang L, Xiong Y . Evolving Fully Automated Machine Learning via Life-Long Knowledge Anchors. IEEE Trans Pattern Anal Mach Intell. 2021; 43(9):3091-3107. DOI: 10.1109/TPAMI.2021.3069250. View