Adaptive habitat biogeography-based optimizer for optimizing deep CNN hyperparameters in image classification

Jiayun Xin, Mohammad Khishe, Diyar Qader Zeebaree, Laith Abualigah, Taher M. Ghazal

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

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.

Original languageBritish English
Article numbere28147
JournalHeliyon
Volume10
Issue number7
DOIs
StatePublished - 15 Apr 2024

Keywords

  • Adaptive habitat
  • Biogeography-based optimizer
  • Deep convolutional neural networks
  • Digital image classification

Fingerprint

Dive into the research topics of 'Adaptive habitat biogeography-based optimizer for optimizing deep CNN hyperparameters in image classification'. Together they form a unique fingerprint.

Cite this