An Evolutionary Deep Learning Method Based on Improved Heap-Based Optimization for Medical Image Classification and Diagnosis

Lin Zhang,Zenglin Qiao,Lina Li
DOI: https://doi.org/10.1109/access.2024.3433483
IF: 3.9
2024-08-03
IEEE Access
Abstract:With the continuous advancement of deep neural networks, leveraging deep learning for the classification and diagnosis of medical images to aid physicians in patient diagnosis has gained significant traction. However, the efficacy of deep learning networks is not solely contingent upon the network architecture, it also relies on the appropriate setting of hyperparameters. Properly tuning hyperparameters can markedly enhance the performance of deep neural networks. To propose an efficient deep learning method for medical image classification, this study introduces a deep residual neural network optimized using an enhanced Heap-based Optimization (HBO) algorithm. The enhanced HBO algorithm integrates elements grey wolf mechanism and orthogonal learning mechanism to augment algorithm convergence and bolster the equilibrium between global exploration and exploitation. The proposed algorithm is applied to optimize the hyperparameters of the deep residual network, thereby enhancing accuracy and reliability in medical image classification. To evaluate the efficacy of the proposed model, this study conducts various experiments on the CEC 2017 benchmark functions for the updated HBO algorithm, encompassing validation of strategies, scalability assessments, balance analyses, and comparisons with state-of-the-art similar algorithms. The performance of the proposed evolutionary deep learning method is assessed across three medical image datasets. The three medical image datasets encompass the colorectal cancer dataset, brain tumor dataset, and chest X-ray image dataset. These datasets represent tissue images, magnetic resonance imaging (MRI), and computed tomography (CT) images, respectively. The classification accuracies on the three datasets reached 95.6%, 97.64%, and 96.48%, respectively, outperforming other competing algorithms.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?