| Literature DB >> 35206957 |
Walaa Gouda1, Maram Almurafeh2, Mamoona Humayun2, Noor Zaman Jhanjhi3.
Abstract
The coronavirus disease (COVID-19) is rapidly spreading around the world. Early diagnosis and isolation of COVID-19 patients has proven crucial in slowing the disease's spread. One of the best options for detecting COVID-19 reliably and easily is to use deep learning (DL) strategies. Two different DL approaches based on a pertained neural network model (ResNet-50) for COVID-19 detection using chest X-ray (CXR) images are proposed in this study. Augmenting, enhancing, normalizing, and resizing CXR images to a fixed size are all part of the preprocessing stage. This research proposes a DL method for classifying CXR images based on an ensemble employing multiple runs of a modified version of the Resnet-50. The proposed system is evaluated against two publicly available benchmark datasets that are frequently used by several researchers: COVID-19 Image Data Collection (IDC) and CXR Images (Pneumonia). The proposed system validates its dominance over existing methods such as VGG or Densnet, with values exceeding 99.63% in many metrics, such as accuracy, precision, recall, F1-score, and Area under the curve (AUC), based on the performance results obtained.Entities:
Keywords: COVID-19; chest X-ray; deep transfer learning; neural network (NN); pneumonia
Year: 2022 PMID: 35206957 PMCID: PMC8872326 DOI: 10.3390/healthcare10020343
Source DB: PubMed Journal: Healthcare (Basel) ISSN: 2227-9032
Figure 1Confirmed COVID-19 cases globally (15 January 2022) [8].
Figure 2Samples of CXR images and CT images (a), and CXR image scan (b) CT image scan.
Literature comparison of COVID-19 diagnostic methods using CXR images.
| Recent Work | Techniques Used | Number of Classes | Accuracy |
|---|---|---|---|
| Khan et al. [ | CoroNet | 4 | 89.6% |
| Ucar and Korkmaz [ | Bayes-SqueezeNet | 3 | 98.3% |
| Apostopolus et al. [ | VGG-19 | 3 | 93.48% |
| Sahinbas & Catak [ | VGG-16, VGG-19, ResNet, DenseNet, InceptionV3 | 2 | 80% |
| Jamil and Hussain [ | Deep CNN | 2 | 93% |
| Alzab et al. [ | VGG-16 | 2 | - |
| Joaquin. [ | ResNet-50 | 2 | 96.2% |
| Sethy et al. [ | ResNet-50 + SVM | 3 | 95.33% |
| Houssein et al. [ | hybrid quantum classical CNNs | 3 | 88.6% |
| Saad et al. [ | CNN, GoogleNet, ResNet-18 | 2 | 99.3% |
| Apostolopoulos, & Mpesiana [ | MobileNetV2 | 3 | 96.78% |
| Oh et al. [ | ResNet-18 | 3 | 88.9% |
| Brunese et al. [ | VGG-16 | 3 | 96% |
| slam et al. [ | CNN+LSTM | 2 | 99.4% |
| Ezzat et al. [ | DenseNet121+GSA | 2 | 98.3% |
| Sahlol et al. [ | Inception + FO-MPA | 2 | 99.6% |
| Toraman et al. [ | Capsule Network | 2 | 97.24% |
| Rajaraman, S. and Antani, S. [ | VGG16 | 2 | 93.0% |
| Afshar, P. et al. [ | capsule network | 2 | 97.2% |
| Elshennawy, N. & Ibrahim, D. [ | ResNet152V2, MobileNetV2 | 2 | 99.22% |
Figure 3A schematic methodology for the COVID19 detection system.
Figure 4CXR images from COV-PEN dataset: (a) COVID-19, (b) pneumonia, and (c) mild.
Figure 5Output of the proposed image enhancement process: (a) raw CXR image and (b) enhanced image.
Figure 6Modified versions of the proposed Resnet-50 model: (a) original pre-trained model, (b) adding one FC layer, and (c) adding two FC layers and one Sigmoid.
Average accuracy for the first version model using the COV-PEN dataset with the first 50 layers frozen, epochs = 15, optimizer = Adam, and batch size = 128.
| Learning Rate | Ensemble Using Several Runs | ||
|---|---|---|---|
| Run 1 | Run 2 | Run 3 | |
| 0.0002 | 0.949820789 | 0.924731183 | 0.955197133 |
| 0.0004 | 0.935483871 | 0.919354839 | 0.928315412 |
| 0.0006 | 0.964157706 |
| 0.935483871 |
| 0.0008 | 0.956989247 | 0.930107527 | 0.944444444 |
| 0.001 | 0.935483871 | 0.919354839 | 0.928315412 |
Average accuracy for the first version model using the COV-PEN dataset with the first 50 layers frozen, epochs = 15, optimizer = sgmd, and batch size = 128.
| Learning Rate | Ensemble Using Several Runs | ||
|---|---|---|---|
| Run 1 | Run 2 | Run 3 | |
| 0.0002 | 0.953405018 | 0.955197133 |
|
| 0.0004 | 0.931899642 | 0.910394265 | 0.949820789 |
| 0.0006 | 0.949820789 | 0.948028674 | 0.931899642 |
| 0.0008 | 0.919354839 | 0.944444444 | 0.9390681 |
| 0.001 | 0.931899642 | 0.910394265 | 0.949820789 |
Average accuracy for the second version model using the COV-PEN dataset with the first 50 layers frozen, epochs = 15, optimizer = Adam, and batch size = 128.
| Learning Rate | Ensemble Using Several Runs | ||
|---|---|---|---|
| Run 1 | Run 2 | Run 3 | |
| 0.0002 | 0.853046595 | 0.903942652 | 0.808960573 |
| 0.0004 | 0.749820789 |
| 0.892473118 |
| 0.0006 | 0.678136201 | 0.808960573 | 0.747311828 |
| 0.0008 | 0.88781362 | 0.62437276 | 0.716845878 |
| 0.001 | 0.683512545 | 0.679928315 | 0.617845867 |
Average accuracy for the second version model using the COV-PEN dataset with the first 50 layers frozen, epochs = 15, optimizer = sgmd, and batch size = 128.
| Learning Rate | Ensemble Using Several Runs | ||
|---|---|---|---|
| Run 1 | Run 2 | Run 3 | |
| 0.0002 | 0.933691756 | 0.937275986 | 0.933691756 |
| 0.0004 | 0.939068100 | 0.935483871 | 0.903942652 |
| 0.0006 |
| 0.917562724 | 0.907526882 |
| 0.0008 | 0.892473118 | 0.716845878 | 0.808960573 |
| 0.001 | 0.747311828 | 0.921146953 | 0.808960573 |
Average accuracy for the first version model using the COV-PEN dataset freeze = 0, epochs = 15, optimizer = Adam, and batch size = 128.
| Learning Rate | Ensemble Using Several Runs | ||
|---|---|---|---|
| Run 1 | Run 2 | Run 3 | |
| 0.0002 | 0.815412186 | 0.843010753 | 0.842175627 |
| 0.0004 | 0.607526882 | 0.772401434 | 0.607526882 |
| 0.0006 |
| 0.798207885 | 0.81172043 |
| 0.0008 | 0.610394265 | 0.678853047 | 0.607526882 |
| 0.001 | 0.756630824 | 0.733333333 | 0.734050179 |
Average accuracy for the first version model using the COV-PEN dataset freeze = 0, epochs = 15, optimizer = sgmd, and batch size = 128.
| Learning Rate | Ensemble Using Several Runs | ||
|---|---|---|---|
| Run 1 | Run 2 | Run 3 | |
| 0.0002 |
| 0.937275986 | 0.933691756 |
| 0.0004 | 0.933691756 | 0.908602151 | 0.919002151 |
| 0.0006 | 0.9390681 | 0.88172043 | 0.843010753 |
| 0.0008 | 0.935483871 | 0.869175627 | 0.81172043 |
| 0.001 | 0.734050179 | 0.790322581 | 0.756630824 |
Figure 7The best result’s confusion matrix for the first version model freeze = 0.
Figure 8The best result’s confusion matrix for the first version model freeze = 50.
Figure 9The best result’s confusion matrix for the second version model.
Figure 10ROC curve for the first version model freeze = 0.
Figure 11ROC curve for the first version model freeze = 50.
Figure 12ROC curve for the second version model.
Best result for all models.
| Quantitative Measures | 1st Version Model (freeze = 0) | 1st Version Model (freeze = 50) | 2nd Version Model |
|---|---|---|---|
| Overall Accuracy | 99.05 |
| 99.05 |
| Precision | 98.89 |
| 98.92 |
| Recall | 98.39 |
| 98.39 |
| F1-score | 98.63 |
| 98.65 |
| AUC | 99.99 |
| 99.98 |
Figure 13Best result for all models.
Comparison of the proposed 1st version model with existing systems.
| Reference | Dataset | Classifier | Accuracy | Precision | Recall | F1-Score | AUC |
|---|---|---|---|---|---|---|---|
| Proposed 1st version | COV-PEN | Modified version of Resnet-50 | 99.63 | 100 | 98.89 | 99.44 | 100 |
| Afshar, P. et. al. [ | - COVID-19 Image Data Collection | — | 97.2 | 97.67 | 97.5 | 97.70 | — |
| - Chest X-ray Images (Pneumonia) | |||||||
| - COVID-19 Image Data Collection | |||||||
| Rajaraman, S. and Antani, S. [ | - Pediatric CXR dataset | VGG16 | 93.0 | 93.15 | 97.53 | 94.57 | 95.0 |
| - RSNA CXR dataset | InceptionV3 | ||||||
| - CheXpert CXR dataset | Xception | ||||||
| - NIH CXR-14 dataset | Densenet121 | ||||||
| - Twitter COVID-19 CXR dataset | NASNet-Mobile | ||||||
| - COVID-19 Image Data Collection | |||||||
| Narin, A., et. al. [ | - Chest X-ray Images (Pneumonia) | ResNet-50 | 99.5 | 99.4 | 99.5 | 98.0 | 98.7 |
| - COVID-19 Image Data Collection | ResNet-101 | ||||||
| ResNet-152 | |||||||
| InceptionV3 | |||||||
| InceptionRes | |||||||
| net-V2 | |||||||
| - Chest X-ray Images (Pneumonia) | |||||||
| Hemdan, E. et. al. [ | COVID-19 Image Data Collection | DenseNet201 | 90 | 83 | — | 91.00 | — |
| Elshennawy, N. & Ibrahim, D. [ | Chest X-ray Images (Pneumonia) | ResNet152V2 | 99.22 | 99.43 | 99.44 | 99.44 | 99.77 |
| MobileNetV2 | |||||||
| Wang et al. [ | - COVID-19 Image Data Collection | VGG-19 | 93.3 | — | 91 | — | — |
| - COVID-19 Chest X-ray Dataset | Resnet-50 | ||||||
| - ActualMed COVID-19 Chest X-ray Dataset | COVID-Net | ||||||
| - RSNA Pneumonia Detection Challenge dataset | |||||||
| - COVID-19 radiography database | |||||||
| Zhang et al. [ | - COVID-19 Image Data Collection | ResNet-18 | — | — | 96 | — | 95.18 |
| - Chest X-ray Images (Pneumonia) | |||||||
| Das et al. [ | - COVID-19 Image Data Collection | extreme version of the Inception (Xception) model | 97.40 | — | 97.09 | 96.96 | — |
| - ChestX-ray8 database (Pneumonia | Normal) | ||||||
| Ozturk et. al. [ | - COVID-19 Image Data Collection | DarkNet | 98.08 | 98.3 | 95.1 | 96.5 | — |
| - ChestX-ray8 database (Pneumonia | Normal) |
Figure 14The proposed first version model performance in comparison with current systems.