| Literature DB >> 34960313 |
Amin Ul Haq1, Jian Ping Li1, Sultan Ahmad2, Shakir Khan3, Mohammed Ali Alshara3, Reemiah Muneer Alotaibi3.
Abstract
COVID-19 is a transferable disease that is also a leading cause of death for a large number of people worldwide. This disease, caused by SARS-CoV-2, spreads very rapidly and quickly affects the respiratory system of the human being. Therefore, it is necessary to diagnosis this disease at the early stage for proper treatment, recovery, and controlling the spread. The automatic diagnosis system is significantly necessary for COVID-19 detection. To diagnose COVID-19 from chest X-ray images, employing artificial intelligence techniques based methods are more effective and could correctly diagnosis it. The existing diagnosis methods of COVID-19 have the problem of lack of accuracy to diagnosis. To handle this problem we have proposed an efficient and accurate diagnosis model for COVID-19. In the proposed method, a two-dimensional Convolutional Neural Network (2DCNN) is designed for COVID-19 recognition employing chest X-ray images. Transfer learning (TL) pre-trained ResNet-50 model weight is transferred to the 2DCNN model to enhanced the training process of the 2DCNN model and fine-tuning with chest X-ray images data for final multi-classification to diagnose COVID-19. In addition, the data augmentation technique transformation (rotation) is used to increase the data set size for effective training of the R2DCNNMC model. The experimental results demonstrated that the proposed (R2DCNNMC) model obtained high accuracy and obtained 98.12% classification accuracy on CRD data set, and 99.45% classification accuracy on CXI data set as compared to baseline methods. This approach has a high performance and could be used for COVID-19 diagnosis in E-Healthcare systems.Entities:
Keywords: COVID-19 diagnosis; accuracy; clinical images data; convolution neural network; multi classification; transfer learning
Mesh:
Year: 2021 PMID: 34960313 PMCID: PMC8707954 DOI: 10.3390/s21248219
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
2DCNN model structure for multi-classification.
| Number | Layer (Name) |
|---|---|
| 1 | Conv2D (64, (7, 2)) |
| 2 | Activation (‘ReLU’) |
| 3 | MaxPool2D (pool-size = (2, 2)) |
| 4 | Conv2D (64, (3, 3) |
| 5 | Activation (‘ReLU’) |
| 6 | MaxPool2D (pool-size = (2, 2)) |
| 7 | Conv2D (64, (3, 3)) |
| 8 | Activation (‘ReLU’) |
| 9 | MaxPool2D (pool-size = (2, 2)) |
| 10 | Flatten () |
| 11 | Dense (64) |
| 12 | Activation (‘ReLU’) |
| 13 | Dropout (0.5) |
| 14 | Dense (3) |
| 15 | Activation (‘Softmax’) |
Figure 1ResNet-50 architecture.
Figure 2Proposed R2DCNNMC model for COVID-19 diagnosis.
R2DCNNMC model parameters.
| Parameters | Value |
|---|---|
| Optimizer | SGD |
| Learning rate | 0.0001 |
| Number of epoch | 100 |
| Bach size | 100 |
| Mini-batch size | 9 |
| Training data set | 70% |
| Validation data set | 30% |
Figure 3Types of chest X-ray images in CRD data set.
Figure 4Types of chest X-ray images in CXI data set.
2DCNN model performance evaluation on original and augmented CRD and CXI data sets.
| Data Set | Parameters | Assessment Measures | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
|
|
|
|
|
|
|
|
|
| ||
| CDR | original | SGD | 0.0001 | 95.20 | 97.00 | 80.25 | 92.40 | 93.00 | 95.09 | 96.00 |
| augmented | - | - | 96.00 | 96.45 | 97.00 | 97.43 | 96.33 | 96.52 | 97.23 | |
| CXI | original | - | - | 97.02 | 98.00 | 99.25 | 100.00 | 99.26 | 97.00 | 99.00 |
| augmented | - | - | 97.65 | 99.10 | 97.86 | 99.80 | 99.87 | 97.73 | 99.23 | |
ResNet-50 model performance evaluation on original and augmented CRD and CXI data sets.
| Data Set | Parameters | Assessment Measures | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
|
|
|
|
|
|
|
|
|
| ||
| CDR | original | SGD | 0.0001 | 94.03 | 96.32 | 83.25 | 97.10 | 93.50 | 95.00 | 94.20 |
| augmented | - | - | 95.20 | 97.00 | 99.00 | 88.21 | 96.12 | 97.34 | 95.00 | |
| CXI | original | - | - | 92.34 | 94.46 | 97.67 | 98.00 | 94.98 | 93.00 | 92.10 |
| augmented | - | - | 94.87 | 95.98 | 95.51 | 93.00 | 95.24 | 95.00 | 93.19 | |
R2DCNNMC model performance evaluation on original and augmented CRD and CXI data sets.
| Data Set | Parameters | Assessment Measures | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
|
|
|
|
|
|
|
|
|
| ||
| CRD | original | SGD | 0.0001 | 97.66 | 99.00 | 89.18 | 99.10 | 99.30 | 98.00 | 97.03 |
| augmented | - | - | 98.12 | 99.28 | 93.00 | 99.56 | 99.70 | 98.23 | 98.60 | |
| CXI | original | - | - | 98.17 | 100.00 | 96.25 | 99.24 | 99.70 | 99.46 | 99.23 |
| augmented | - | - | 99.45 | 99.63 | 96.99 | 100.00 | 99.83 | 99.78 | 99.90 | |
R2DCNNMC model accuracy comparison with baseline methods.
| Method | Accuracy (%) | Reference |
|---|---|---|
| ResNet + SVM | 95.38 | [ |
| GAN + Resnet18 | 99 | [ |
| VGG-16+ CNN | 91.24 | [ |
| TLRV1 | 94.4 | [ |
| DTL | 95.72 | [ |
| ResNet-50 | 96.23 | [ |
| COVID-Net-TM | 92.4 | [ |
| DRE-Net | 86 | [ |
| COVIDx-Net | 90 | [ |
| TM | 93.3 | [ |
| TL | 93 | [ |
| COVID-Net CT-2 | 98.1 | [ |
| DarkCovidNet | 90.8 | [ |
| ResNet50 | 90 | [ |
| VGG19 + CNN | 98.05 | [ |
| VGG-19 | 98.87 | [ |
| Proposed method (R2DCNNMC) | 98.12 | 2021 |
| Proposed method (R2DCNNMC) | 99.45 | 2021 |