| Literature DB >> 34305440 |
Sarra Guefrechi1, Marwa Ben Jabra2,3, Adel Ammar4, Anis Koubaa4,5,6, Habib Hamam1.
Abstract
The whole world is facing a health crisis, that is unique in its kind, due to the COVID-19 pandemic. As the coronavirus continues spreading, researchers are concerned by providing or help provide solutions to save lives and to stop the pandemic outbreak. Among others, artificial intelligence (AI) has been adapted to address the challenges caused by pandemic. In this article, we design a deep learning system to extract features and detect COVID-19 from chest X-ray images. Three powerful networks, namely ResNet50, InceptionV3, and VGG16, have been fine-tuned on an enhanced dataset, which was constructed by collecting COVID-19 and normal chest X-ray images from different public databases. We applied data augmentation techniques to artificially generate a large number of chest X-ray images: Random Rotation with an angle between - 10 and 10 degrees, random noise, and horizontal flips. Experimental results are encouraging: the proposed models reached an accuracy of 97.20 % for Resnet50, 98.10 % for InceptionV3, and 98.30 % for VGG16 in classifying chest X-ray images as Normal or COVID-19. The results show that transfer learning is proven to be effective, showing strong performance and easy-to-deploy COVID-19 detection methods. This enables automatizing the process of analyzing X-ray images with high accuracy and it can also be used in cases where the materials and RT-PCR tests are limited.Entities:
Keywords: CNN; COVID-19; Chest X-ray; Convolution Neural Network; Deep learning
Year: 2021 PMID: 34305440 PMCID: PMC8286881 DOI: 10.1007/s11042-021-11192-5
Source DB: PubMed Journal: Multimed Tools Appl ISSN: 1380-7501 Impact factor: 2.757
Fig. 1Overview for COVID-19 and non-COVID-19 Chest X-ray images classification
Content of our prepared dataset
| Without data augmentation | With data augmentation | |
|---|---|---|
| COVID-19 | 623 | 2000 |
| Normal | 3000 | 3000 |
| Total of images | 3623 | 5000 |
Fig. 2(A): Chest X-ray image of a healthy person. (B): COVID-19 chest X-ray image
Fig. 3Proposed VGG16 architecture
Fig. 4Proposed Resnet50 architecture
Fig. 5Proposed InceptionV3 architecture
Fig. 6Plots of (a) Training and validation accuracy and (b) Training and validation loss by using training epochs-InceptionV3
Fig. 7Plots of (a) Training and validation accuracy and (b) Training and validation loss by using training epochs-VGG16
Fig. 8Plots of (a) Training and validation accuracy and (b) Training and validation loss by using training epochs-Resnet50
Classification report for Resnet50, InceptionV3 and VGG16
| Modified version of | Accuracy | sensitivity | Specificity | Precision | Recall | F1 Score |
|---|---|---|---|---|---|---|
| Resnet50 | 97.20 % | 98.25 % | 97.00 % | 97.00 % | 96.00 % | 97.00 % |
| InceptionV3 | 98.10 % | 99.25 % | 98.00 % | 98.00 % | 98.00 % | 98.00 % |
| VGG16 | 98.30 % | 98.25 % | 98.33 % | 98.00 % | 98.00 % | 98.00 % |
Fig. 9The confusion matrix of the proposed VGG16 model
Fig. 10The confusion matrix of the proposed Resnet50 model
Fig. 11The confusion matrix of the proposed InceptionV3 model
Comparison of the most commonly used automatic diagnosis of COVID-19 that are based on chest X-ray images to our fine-tuned models
| Study | Architecture | Accuracy | Number of parameters in Million |
|---|---|---|---|
| Sethy and Behra [ | Resnet50 | 95.38 % | 36 |
| Narin et al. [ | InceptionV3 | 97 % | 26 |
| Ioannis et al. [ | Xception | 85.57 % | 33 |
| Ozturk et al. [ | DarkNet | 98.08 % | 1.1 |
| Ioannis et al. [ | VGG19 | 98.75 % | 143 |
| Fine-tuned Resnet50 | Resnet50 | 97.20 % | 23 |
| Fine-tuned VGG16 | VGG16 | 98.30 % | 15 |
| Fine-tuned InceptionV3 | InceptionV3 | 98.10 % | 21 |