CovH2SD: A COVID-19 detection approach based on Harris Hawks Optimization and stacked deep learning

https://doi.org/10.1016/j.eswa.2021.115805Get rights and content

Highlights

  • Proposing HHO deep learning approach for COVID-19 detection using chest CT images.

  • Applying transfer learning using 9 pre-trained convolutional neural network models.

  • Injecting HHO in learning process to select optimal configurations for each model.

  • Presenting a stacking mechanism from the optimized models.

  • The proposed approach is benchmarked against other state-of-the-art models.

Abstract

Starting from Wuhan in China at the end of 2019, coronavirus disease (COVID-19) has propagated fast all over the world, affecting the lives of billions of people and increasing the mortality rate worldwide in few months. The golden treatment against the invasive spread of COVID-19 is done by identifying and isolating the infected patients, and as a result, fast diagnosis of COVID-19 is a critical issue. The common laboratory test for confirming the infection of COVID-19 is Reverse Transcription Polymerase Chain Reaction (RT-PCR). However, these tests suffer from some problems in time, accuracy, and availability. Chest images have proven to be a powerful tool in the early detection of COVID-19. In the current study, a hybrid learning and optimization approach named CovH2SD is proposed for the COVID-19 detection from the Chest Computed Tomography (CT) images. CovH2SD uses deep learning and pre-trained models to extract the features from the CT images and learn from them. It uses Harris Hawks Optimization (HHO) algorithm to optimize the hyperparameters. Transfer learning is applied using nine pre-trained convolutional neural networks (i.e. ResNet50, ResNet101, VGG16, VGG19, Xception, MobileNetV1, MobileNetV2, DenseNet121, and DenseNet169). Fast Classification Stage (FCS) and Compact Stacking Stage (CSS) are suggested to stack the best models into a single one. Nine experiments are applied and results are reported based on the Loss, Accuracy, Precision, Recall, F1-Score, and Area Under Curve (AUC) performance metrics. The comparison between combinations is applied using the Weighted Sum Method (WSM). Six experiments report a WSM value above 96.5%. The top WSM and accuracy reported values are 99.31% and 99.33% respectively which are higher than the eleven compared state-of-the-art studies

Keywords

Computed Tomography (CT)
Convolutional Neural Network (CNN)
COVID-19
Data Augmentation (DA)
Harris Hawks Optimization (HHO)
Transfer Learning (TL)

Cited by (0)

1

This work has been done by equal efforts of all authors.

View Abstract