| Literature DB >> 35800691 |
Bhavana Kaushik1, Deepika Koundal1, Neelam Goel2, Atef Zaguia3, Assaye Belay4, Hamza Turabieh3.
Abstract
Chest X-ray (CXR) scans are emerging as an important diagnostic tool for the early spotting of COVID and other significant lung diseases. The recognition of visual symptoms is difficult and can take longer time by radiologists as CXR provides various signs of viral infection. Therefore, artificial intelligence-based method for automated identification of COVID by utilizing X-ray images has been found to be very promising. In the era of deep learning, effective utilization of existing pretrained generalized models is playing a decisive role in terms of time and accuracy. In this paper, the benefits of weights of existing pretrained model VGG16 and InceptionV3 have been taken. Base model has been created using pretrained models (VGG16 and InceptionV3). The last fully connected (FC) layer has been added as per the number of classes for classification of CXR in binary and multi-class classification by appropriately using transfer learning. Finally, combination of layers is made by integrating the FC layer weights of both the models (VGG16 and InceptionV3). The image dataset used for experimentation consists of healthy, COVID, pneumonia viral, and pneumonia bacterial. The proposed weight fusion method has outperformed the existing models in terms of accuracy, achieved 99.5% accuracy in binary classification over 20 epochs, and 98.2% accuracy in three-class classification over 100 epochs.Entities:
Mesh:
Year: 2022 PMID: 35800691 PMCID: PMC9253872 DOI: 10.1155/2022/7124199
Source DB: PubMed Journal: Comput Intell Neurosci
Figure 1Flow of the proposed methodology and architecture.
Binary classification of unhealthy vs healthy patient's results.
| Category | F1-score | Recall | Precision | Support |
|---|---|---|---|---|
| COVID-19/pneumonia | 0.93 | 0.88 | 1.00 | 24 |
| Healthy | 0.99 | 1.00 | 0.97 | 101 |
Figure 2Plotting of (a) accuracy and (b) loss of training and testing of proposed model over 20 epochs for binary classification of unhealthy (COVID-19/pneumonia) vs healthy patients.
Figure 3Classification of (a–c) healthy (no findings) and (d–f) unhealthy (COVID-19/pneumonia) class testing results.
Classification of healthy, COVID-19, and pneumonia results.
| Category | Precision | Recall | F1-score | Support |
|---|---|---|---|---|
| COVID-19 | 0.94 | 0.86 | 0.89 | 25 |
| Healthy | 0.84 | 0.96 | 0.93 | 90 |
| Pneumonia | 0.88 | 0.85 | 0.86 | 90 |
Figure 4(a and b) Plotting of accuracy and loss of training and testing of proposed model over 100 epochs for classification of healthy, COVID-19, and pneumonia.
Figure 5Classification of (a–c) healthy, (d–f) COVID-19, and (g–i) pneumonia class testing results.
Figure 6(a and b) Plotting of accuracy and loss of training and testing of proposed model over 20 epochs to classify healthy, COVID-19, viral pneumonia, and bacterial pneumonia.
Comparison of the proposed model with other state-of-the-art models.
| Author | Dataset | Approach/method | Accuracy |
|---|---|---|---|
| Nishio et al. [ | 1248 CXR images—998 (training), 125 (validation), and 125 (test) images | Pretrained model is VGG16 integration of conventional method with augmentation methods | 83.6% (non-COVID-19 pneumonia, COVID-19 pneumonia, and healthy) |
|
| |||
| Sharma et al. [ | 352 X-ray images (317—training and 35—testing) | Transfer learning used and external dataset validation performed | 100% and 75% accuracy achieved for training and testing (24 epochs-based model) |
|
| |||
| Wang et al. [ | Training (normal—7966, pneumonia—8620, and COVID-19—128) | Fusion of pretrained model ResNet101 and ResNet152 | 96.1% |
| Testing (normal—885, pneumonia—956, and COVID-19—12) | |||
|
| |||
| Mahmud et al. [ | Normal X-rays (1583), viral pneumonia (1493), and bacterial pneumonia (2780) | Utilizes depth-wise convolution with different dilation levels for the extraction of features from CXR | 96.9% (COVID/viral pneumonia), 97.4% (COVID/normal), 94.7% (COVID/bacterial pneumonia) |
|
| |||
| Fawaz et al. [ | The dataset consists of 21,165 X-ray images COVID-19-positive (3616), viral pneumonia (1345), images of normal patients (10,192), and lung opacity (non-COVID-19) (6012) | ResNet50 and AlexNet are used and features mined by convolutional neural network (CNN) using traditional methods | 99% accuracy—COVID-19/viral pneumonia/lung opacity/normal |
|
| |||
| Proposed | Normal—1583 | Weight fusion of InceptionV3 and VGG16 on FC layers | 99.5% accuracy for (healthy/unhealthy) 20 epochs and 98.2% (normal/COVID/pneumonia) 100 epochs |
| Pneumonia—4273 | |||
| Bacterial pneumonia—1860 | |||
| Viral pneumonia—2413 | |||
| COVID-19—714 | |||