| Literature DB >> 33967405 |
Adi Alhudhaif1, Kemal Polat2, Onur Karaman3.
Abstract
X-ray units have become one of the most advantageous candidates for triaging the new Coronavirus disease COVID-19 infected patients thanks to its relatively low radiation dose, ease of access, practical, reduced prices, and quick imaging process. This research intended to develop a reliable convolutional-neural-network (CNN) model for the classification of COVID-19 from chest X-ray views. Moreover, it is aimed to prevent bias issues due to the database. Transfer learning-based CNN model was developed by using a sum of 1,218 chest X-ray images (CXIs) consisting of 368 COVID-19 pneumonia and 850 other pneumonia cases by pre-trained architectures, including DenseNet-201, ResNet-18, and SqueezeNet. The chest X-ray images were acquired from publicly available databases, and each individual image was carefully selected to prevent any bias problem. A stratified 5-fold cross-validation approach was utilized with a ratio of 90% for training and 10% for the testing (unseen folds), in which 20% of training data was used as a validation set to prevent overfitting problems. The binary classification performances of the proposed CNN models were evaluated by the testing data. The activation mapping approach was implemented to improve the causality and visuality of the radiograph. The outcomes demonstrated that the proposed CNN model built on DenseNet-201 architecture outperformed amongst the others with the highest accuracy, precision, recall, and F1-scores of 94.96%, 89.74%, 94.59%, and 92.11%, respectively. The results indicated that the reliable diagnosis of COVID-19 pneumonia from CXIs based on the CNN model opens the door to accelerate triage, save critical time, and prioritize resources besides assisting the radiologists.Entities:
Keywords: Chest X-ray images; Convolutional Neural Network (CNN); Corona Virus (COVID-19); Deep learning
Year: 2021 PMID: 33967405 PMCID: PMC8093008 DOI: 10.1016/j.eswa.2021.115141
Source DB: PubMed Journal: Expert Syst Appl ISSN: 0957-4174 Impact factor: 6.954
Fig. 1Representative chest X-ray images for other pneumonia class (upper row) and COVID-19 pneumonia class (bottom row).
Description of the total number of CXIs per class and per fold.
| Class | Total Number of CXIs/Class | Training Set/Fold | Augmented Image/Fold | Validation Set/Fold | Testing |
|---|---|---|---|---|---|
| COVID-19 pneumonia | 265 | 1,855 | 66 | 37 | |
| Other Pneumonia | 612 | 1,836 | 153 | 85 |
Fig. 2The schematic illustration of the developed CNN model.
Scheme 1Representative flowchart of proposed CNN approach.
Fig. 3Schematic illustration of 5-fold cross-validation approach.
Fig. 4The comparison of performance metrics for DenseNet-201, ResNet-18, SqueezeNet architectures obtained in Fold-1 for the CNN model.
Fig. 5The overlapped confusion matrices of validation data for (A) DenseNet-201, (B) ResNet-18, and (C) SqueezeNet architectures.
The calculated performance metrics of the proposed model.
| TP | FP | FN | TN | Accuracy(%) | Precision (%) | Recall(%) | F1-score | |
|---|---|---|---|---|---|---|---|---|
| Fold-1 | 61 | 7 | 5 | 146 | 94.52 | 89.71 | 92.42 | 91.04 |
| Fold-2 | 58 | 3 | 8 | 150 | 94.98 | 95.08 | 87.88 | 91.34 |
| Fold-3 | 60 | 2 | 6 | 151 | 96.35 | 96.77 | 90.91 | 93.75 |
| Fold-4 | 61 | 5 | 5 | 148 | 95.43 | 92.42 | 92.42 | 92.42 |
| Fold-5 | 62 | 3 | 4 | 150 | 96.80 | 95.38 | 93.94 | 94.66 |
| Total | 302 | 20 | 28 | 745 | 95.62 | 93.79 | 91.52 | 92.64 |
| Fold-1 | 58 | 9 | 8 | 144 | 92.24 | 86.57 | 87.88 | 87.22 |
| Fold-2 | 59 | 13 | 7 | 140 | 90.87 | 81.94 | 89.39 | 85.51 |
| Fold-3 | 61 | 8 | 5 | 145 | 94.06 | 88.41 | 92.42 | 90.37 |
| Fold-4 | 61 | 11 | 5 | 142 | 92.69 | 84.72 | 92.42 | 88.41 |
| Fold-5 | 60 | 3 | 6 | 150 | 95.89 | 95.24 | 90.91 | 93.02 |
| Total | 299 | 44 | 31 | 721 | 93.15 | 87.17 | 90.61 | 88.86 |
| Fold-1 | 58 | 11 | 8 | 142 | 91.32 | 84.06 | 87.88 | 85.93 |
| Fold-2 | 60 | 9 | 6 | 144 | 93.15 | 86.96 | 90.91 | 88.89 |
| Fold-3 | 59 | 10 | 7 | 143 | 92.24 | 85.51 | 89.39 | 87.41 |
| Fold-4 | 61 | 12 | 5 | 141 | 92.24 | 83.56 | 92.42 | 87.77 |
| Fold-5 | 62 | 5 | 4 | 148 | 95.89 | 92.54 | 93.94 | 93.23 |
| Total | 300 | 47 | 30 | 718 | 92.97 | 86.46 | 90.91 | 88.63 |
TP: True Positive; FP:False Positive; FN: False Negative; TN: False Positive.
Fig. 6The calculated confusion matrices of testing data for (A) DenseNet-201, (B) ResNet-18, and (C) SqueezeNet architectures.
The performance metrics of each architecture for testing data.
| Architecture | Test Time(ms) | Accuracy | Precision | Recall | F1-score |
|---|---|---|---|---|---|
| DenseNet_201 | |||||
| ResNet_18 | 0.583 | 91.60 | 84.62 | 89.19 | 86.84 |
| SqueezeNet | 0.771 | 89.92 | 82.05 | 86.49 | 84.21 |
Fig. 7The representative illustration of original chest X-ray images (left-side) and Grad-CAM activation mapping (right-side) of other pneumonia and COVID-19 pneumonia cases.
The comparison of related studies for the automatic analysis of COVID-19 pneumonia from chest x-ray imaging utilizing artificial intelligence.
| Study | Number of Cases | Data Source COVID-19 | Accuracy (%) |
|---|---|---|---|
| 53 COVID-19 (+)5526 COVID-19 (-)8066 Healthy | Wang | 93.30 | |
| 299 COVID-191522 Pneumonia | Cohen | 97.10 | |
| 25 COVID-19 (+)25 COVID-19 (-) | Cohen | 95.38 | |
| 183 COVID-19 (+)8066 Healthy | Wang | 94.30 | |
| 76 COVID-19 (+)5526 Pneumonia8066 Healthy | Cohen | 93.50 | |
| 493 COVID-19 (+)24,596 Pneumonia18,774 Healthy | Cohen, Kaggle | 83.61 | |
| 118 COVID-19 (+)6037 Pneumonia8868 Healthy | Cohen | 91.40 | |
| 135 COVID-19 (+)208 Pneumonia | Cohen, SIRM, Radiopeadia | 93.12 | |
| 184 COVID-19 (+)6290 Healthy | Cohen, Kaggle | 97.00 | |
| 231 COVID-19 (+)4273 Pneumonia1583 Healthy | Cohen | 92.18 | |
| This study |