| Literature DB >> 34177037 |
Abdullahi Umar Ibrahim1, Mehmet Ozsoz1, Sertan Serte2, Fadi Al-Turjman3, Salahudeen Habeeb Kolapo4.
Abstract
Reverse-Transcription Polymerase Chain Reaction (RT-PCR) method is currently the gold standard method for detection of viral strains in human samples, but this technique is very expensive, take time and often leads to misdiagnosis. The recent outbreak of COVID-19 has led scientists to explore other options such as the use of artificial intelligence driven tools as an alternative or a confirmatory approach for detection of viral pneumonia. In this paper, we utilized a Convolutional Neural Network (CNN) approach to detect viral pneumonia in x-ray images using a pretrained AlexNet model thereby adopting a transfer learning approach. The dataset used for the study was obtained in the form of optical Coherence Tomography and chest X-ray images made available by Kermany et al. (2018, https://doi.org/10.17632/rscbjbr9sj.3) with a total number of 5853 pneumonia (positive) and normal (negative) images. To evaluate the average efficiency of the model, the dataset was split into on 50:50, 60:40, 70:30, 80:20 and 90:10 for training and testing respectively. To evaluate the performance of the model, 10 K Cross-validation was carried out. The performance of the model using overall dataset was compared with the means of cross-validation and the currents state of arts. The classification model has shown high performance in terms of accuracy, sensitivity and specificity. 70:30 split performed better compare to other splits with accuracy of 98.73%, sensitivity of 98.59% and specificity of 99.84%.Entities:
Keywords: CNN; COVID‐19; artificial intelligence; pretrained AlexNet; viral pneumonia
Year: 2021 PMID: 34177037 PMCID: PMC8209916 DOI: 10.1111/exsy.12705
Source DB: PubMed Journal: Expert Syst ISSN: 0266-4720 Impact factor: 2.812
Classification of pneumonia based on pathogens
| Pathogen | Species |
|---|---|
|
|
|
| Bacteria |
|
| Fungi |
|
Detection of different types of pneumonia using AI‐driven tools
| Reference | Type of pneumonia | Dataset | Result |
|---|---|---|---|
| Stephen et al., | Viral pneumonia (strain not specified) | 5856 X‐ray images | Average Ac of 94.81% training and 93.01% for validation |
| Rajpurkar et al., | Not specified | 108,948 X‐ray images | 0.6333 AUC |
| X. Wang et al., | Not specified | 100, 000 X‐ray images | 0.8887 AUC |
| Wang, Kang et al. 2020 | Viral pneumonia (COVID‐19) | 453 CT scan images | The model achieved validation AC of 82.9%, SV of 84% and SP of 80.5%, testing AC of 73.1%, SV of 74% and SF of 67%. |
| Saraiva et al., | Viral pneumonia (strain not specified) | 5863 Chest X‐Ray Images | AC of 95.30% |
| Chouhan et al., | Viral and Bacterial pneumonia (strains not specified) | 5863 Chest X‐Ray Images | Different models were used |
| Xu et al., | viral pneumonia (COVID‐19, Influenza‐A) | 618 CT scan Images | Ac of 86.7%. |
| Rajaraman et al., | Viral and Bacterial pneumonia (strains not specified) | 5856 chest X‐Ray | Ac of 96.2% accuracy for bacterial pneumonia and 93.6% for viral pneumonia |
| Zech et al., | Viral and Bacterial pneumonia (strains not specified) | 158,323 chest radiographs | Different models were used |
Abbreviations: Ac, Accuracy; AUC, Area under the curve; Sf, Specificity; Sv, Sensitivity.
FIGURE 1The workflow is represented schematically. CXR images are used to train the network using Pretrained AlexNet model for classification of pneumonia and normal (healthy)
FIGURE 2Pediatric CXR scans. Left: Pneumonia. Right: Normal CXR scan
Dataset description
| Label | Number |
|---|---|
| Positive | 4273 |
| Negative | 1583 |
| Total | 5856 |
FIGURE 3Training of models using AlexNet model. AlexNet model contain 5 convolution (CONV) blocks or layers. The first 2 CONV layers are made up of 3 operations which include convolution, max pooling and normalization. Third and fourth layer are made up of only convolution while fifth layer is made up of convolution and max pooling. The last 3 layers are 2 fully connected layers (FCL) and output layer with SoftMax activation function for classification
Data split
| Split | Training | Split | Testing | |||
|---|---|---|---|---|---|---|
| S/No | % | Positive | Negative | % | Positive | Negative |
| 1 | 50 | 2137 | 792 | 50 | 2136 | 791 |
| 2 | 60 | 2564 | 950 | 40 | 1709 | 633 |
| 3 | 70 | 2991 | 1108 | 30 | 1282 | 475 |
| 4 | 80 | 3418 | 1266 | 20 | 855 | 317 |
| 5 | 90 | 3846 | 1425 | 10 | 427 | 158 |
Note: Total number of dataset = 5856, Positive = 4273, Negative = 1583.
Confusion matrix
|
│ Predicted │ | — Actual — | ||
| True Positive (+) | False Negative (−) | ||
| True Positive | True + | False + | |
| False Negative | False − | True − | |
General dataset result
| Split | Training accuracy | Testing accuracy | Sensitivity | Specificity |
|---|---|---|---|---|
| 50–50 | 97.96 | 97.94 | 96.71 | 99.00 |
| 60–40 | 98.94 | 98.95 | 99.09 | 98.81 |
| 70–30 | 99.19 | 98.73 | 98.59 | 98.84 |
| 80–20 | 99.36 | 100.00 | 99.11 | 99.66 |
| 90–10 | 99.86 | 100.00 | 99.70 | 100.00 |
Cross validation result for pneumonia
| K fold | Tr(A) | V | Ts(A) | Sv | Sf |
|---|---|---|---|---|---|
| 1 | 98.35 | 0.9835 | 96.67 | 0.9800 | 0.9846 |
| 2 | 96.78 | 0.9678 | 94.71 | 0.9767 | 0.9650 |
| 3 | 97.72 | 0.9772 | 96.55 | 0.9867 | 0.9743 |
| 4 | 97.56 | 0.9756 | 94.71 | 0.9567 | 0.9815 |
| 5 | 97.72 | 0.9772 | 98.16 | 0.9567 | 0.9835 |
| 6 | 97.48 | 0.9748 | 94.14 | 0.9867 | 0.9712 |
| 7 | 96.86 | 0.9686 | 93.45 | 0.9800 | 0.9650 |
| 8 | 98.35 | 0.9835 | 96.21 | 0.9633 | 0.9897 |
| 9 | 98.27 | 0.9827 | 95.63 | 0.9867 | 0.9815 |
| 10 | 97.88 | 0.9788 | 97.13 | 0.9633 | 0.9835 |
| Average |
976.97/10 97.70 |
9.76970/10 0.9770 |
960.36/10 96.04 |
9.37368/10 0.9734 |
9.7798/10 0.9779 |
Abbreviations: Sf, Specificity; Sv, Sensitivity; Tr(A), Training accuracy; Ts(A), Testing accuracy; V, Validation.
Comparison between general dataset and cross validation
| Split | Training accuracy | Testing accuracy | Sensitivity | Specificity |
|---|---|---|---|---|
| 50–50 | 97.96 | 97.94 | 96.71 | 99.00 |
| 60–40 | 98.94 | 98.95 | 99.09 | 98.81 |
| 70–30 | 99.19 | 98.73 | 98.59 | 98.84 |
| 80–20 | 99.36 | 100.00 | 99.11 | 99.66 |
| 90–10 | 99.86 | 100.00 | 99.70 | 100.00 |
| CV | 97.70 | 96.04 | 97.35 | 97.98 |
Abbreviation: CV, Cross validation.
FIGURE 4Classification of pneumonia using AlexNet
Comparison with similar studies from literature
| Rf | No of dataset | Model | A/AUC | Sv | Sf |
|---|---|---|---|---|---|
| 70:30 | 5856 | PA | 98.73 | 98.59 | 98.84 |
| CV | 5856 | PA | 97.35 | 97.35 | 97.78 |
| Stephen et al., | 5856 | CNN | 94.81 | ‐ | ‐ |
| Chouhan et al., | 5856 | PA | 92.86 | ‐ | ‐ |
| Saraiva et al., | 5856 | CNN | 95.30 | ‐ | ‐ |
| Rajaraman et al., | 5856 | CNN | 92.2, 93.6 | ‐ | ‐ |
| Kanaparthi et al., | 108,948 | PA | 0.6333 | ‐ | ‐ |
| Rajpurkar et al., | 100,000 | CHeXNet | 0.8887 | ‐ | ‐ |
Abbreviations: A, Accuracy; AUC, Area under the curve; CV, Cross validation; PA, Pretrained AlexNet; Sf, Specificity; Sv, Sensitivity.