Hybrid Aquila optimizer-Harris Hawks optimization for CNN hyperparameter tuning in brain tumor classification

用于脑肿瘤分类的 CNN 超参数调优的混合 Aquila 优化器-Harris Hawks 优化器

阅读:1

Abstract

Magnetic resonance imaging (MRI) is hard to categorize properly in terms of interclass similarity, there is data imbalance, and sensitive clinical decision-making: but the performance of convolutional neural networks (CNNs) highly relies on effective, yet computationally costly, hyperparameter tuning. To find solutions to such issues, the given paper proposes a hybrid solution to the problems of the Aquila Optimizer and Harris Hawks Optimization, i.e., Aquila Optimizer-Harris Hawks Optimization (AO-HHO) framework, to integrate the positive qualities of extremely good global exploration of the Aquila Optimizer and the good local exploitation process of a Harris Hawks Optimization to achieve balanced and robust CNN hyperparameter optimization. On a publicly accessible dataset of 7, 023 brain MRI images divided into glioma, meningioma, pituitary tumor, and non-tumor, the proposed algorithm has been tested on with fine-tuning critical hyperparameters, such as learning rate, batch size, number of filters, dropout rate, and optimizer type. The rate of accuracy, precision, recall and F1-score of the AO-HHO-tuned CNN is invariably high than the conventional metaheuristic algorithms, including the Particle Swarm Optimization (PSO), Genetic Algorithm (GA), and Whale Optimization Algorithm (WOA) that are approximately 78-83. The proposed method also helps in reducing the cost of computing. It takes only 77.85 s to train, while the baseline optimizers take more than 300 s. This shows that AO-HHO is a reliable, accurate, and computationally efficient framework that can be used for medical imaging decision-support applications that need to be done in real time and with limited resources.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。