| Literature DB >> 35720905 |
Amel Ksibi1, Mohammed Zakariah2, Manel Ayadi1, Hela Elmannai3, Prashant Kumar Shukla4, Halifa Awal5,6, Monia Hamdi3.
Abstract
COVID-19 has remained a threat to world life despite a recent reduction in cases. There is still a possibility that the virus will evolve and become more contagious. If such a situation occurs, the resulting calamity will be worse than in the past if we act irresponsibly. COVID-19 must be widely screened and recognized early to avert a global epidemic. Positive individuals should be quarantined immediately, as this is the only effective way to prevent a global tragedy that has occurred previously. No positive case should go unrecognized. However, current COVID-19 detection procedures require a significant amount of time during human examination based on genetic and imaging techniques. Apart from RT-PCR and antigen-based tests, CXR and CT imaging techniques aid in the rapid and cost-effective identification of COVID. However, discriminating between diseased and normal X-rays is a time-consuming and challenging task requiring an expert's skill. In such a case, the only solution was an automatic diagnosis strategy for identifying COVID-19 instances from chest X-ray images. This article utilized a deep convolutional neural network, ResNet, which has been demonstrated to be the most effective for image classification. The present model is trained using pretrained ResNet on ImageNet weights. The versions of ResNet34, ResNet50, and ResNet101 were implemented and validated against the dataset. With a more extensive network, the accuracy appeared to improve. Nonetheless, our objective was to balance accuracy and training time on a larger dataset. By comparing the prediction outcomes of the three models, we concluded that ResNet34 is a more likely candidate for COVID-19 detection from chest X-rays. The highest accuracy level reached 98.34%, which was higher than the accuracy achieved by other state-of-the-art approaches examined in earlier studies. Subsequent analysis indicated that the incorrect predictions occurred with approximately 100% certainty. This uncovered a severe weakness in CNN, particularly in the medical area, where critical decisions are made. However, this can be addressed further in a future study by developing a modified model to incorporate uncertainty into the predictions, allowing medical personnel to manually review the incorrect predictions.Entities:
Mesh:
Year: 2022 PMID: 35720905 PMCID: PMC9201714 DOI: 10.1155/2022/9414567
Source DB: PubMed Journal: Comput Intell Neurosci
List of some of the benchmark datasets used in previous works.
| Database | Previous works |
|---|---|
| COVID-19 Radiography Database | [ |
| GitHub repository by [ | [ |
| Chest X-Ray Images (Pneumonia) | [ |
Distribution of the collected images.
| Category | No. of images | |
|---|---|---|
| Train | COVID-19 | 9803 |
| Normal | 8956 | |
|
| ||
| Test | COVID-19 | 2847 |
| Normal | 2831 | |
Figure 1Samples of chest X-ray images from prepared dataset. (a) Normal. (b) COVID-19.
Figure 2Block diagram of the proposed system.
Figure 3Detailed architecture of ResNet.
Sizes of outputs and convolutional kernels for ResNet versions.
| Layer name | Output size | 34 layers | 50 layers | 101 layers |
|---|---|---|---|---|
| conv 1 | 112 × 112 | 7 × 7, 64, stride 2 | ||
| conv 2.x | 56 × 56 | 3 × 3 max pool, stride 2 | ||
|
|
|
| ||
| conv 3.x | 28 × 28 |
|
|
|
| conv 4.x | 14 × 14 |
|
|
|
| conv 4.x | 7 × 7 |
|
|
|
| 1 × 1 | Average pool, 1000-d fc, softmax | |||
| FLOPs | 3.6 × 109 | 3.8 × 109 | 7.6 × 109 | |
Figure 4Loss for the (a) ResNet34, (b) ResNet50, and (c) ResNet101.
Figure 5The flow graph. (a) Validation accuracy. (b) Error rate of the ResNet34, ResNet50, and ResNet101 models.
Validation confusion matrix.
| Models | True:COVID-19 | True: healthy | |
|---|---|---|---|
| ResNet34 | Predicted: COVID-19 | 1924 | 8 |
| Predicted: healthy | 4 | 1815 | |
|
| |||
| ResNet50 | Predicted: COVID-19 | 1923 | 11 |
| Predicted: healthy | 5 | 1812 | |
|
| |||
| ResNet101 | Predicted: COVID-19 | 1925 | 10 |
| Predicted: healthy | 3 | 1813 | |
Validation performance metric.
| Models | Accuracy (%) | Recall (%) | Precision (%) |
|
|---|---|---|---|---|
| ResNet34 |
|
| 99.78 |
|
| ResNet50 | 99.57 | 99.40 | 99.72 | 99.56 |
| ResNet101 | 99.65 | 99.45 |
| 99.64 |
Validation classification report.
| Models | Recall (%) | Precision (%) |
| |
|---|---|---|---|---|
| ResNet34 | COVID-19 | 99.79 | 99.59 | 99.69 |
| Healthy | 99.56 | 99.78 | 99.67 | |
|
| ||||
| ResNet50 | COVID-19 | 99.74 | 99.43 | 99.59 |
| Healthy | 99.40 | 99.72 | 99.56 | |
|
| ||||
| ResNet101 | COVID-19 | 99.84 | 99.48 | 99.66 |
| Healthy | 99.45 | 99.83 | 99.64 | |
Figure 6Confusion matrices: (a) ResNet34, (b) ResNet50, and (c) ResNet101 on test dataset.
Figure 7Test data performance of the models on individual classes.
Test data performance metric.
| Models | Accuracy (%) | Recall (%) | Precision (%) |
| Training time |
|---|---|---|---|---|---|
| ResNet34 |
|
| 99.81 |
|
|
| ResNet50 | 97.99 | 96.29 | 99.67 | 97.95 | 1200 minutes |
| ResNet101 | 98.45 | 97.35 |
| 98.42 | 1700 minutes |
Figure 8Some of the dataset's misclassified images (labeled as predicted class/actual class/loss value/probability of prediction).
Number of parameters in ResNet versions [56].
| ResNet version | Number of parameters (in millions) |
|---|---|
| ResNet34 | 21.8 |
| ResNet50 | 25.6 |
| ResNet101 | 44.5 |
Performance comparison with previous works.
| Works | Architecture | Accuracy (%) |
|---|---|---|
| [ | VGG19 DenseNet201 | 90 |
| [ | COVIDPEN | 96 |
| [ | DenseNet201 + ResNet50V2 + Inceptionv3 | 91.62 |
| [ | VGG19, ResNet152 Xception, DenseNet201 InceptionResNetV2 | 96 |
| [ | Customized DarkNet | 98.08 |
| Our model | ResNet34 | 98.34 |