| Literature DB >> 35020175 |
Abstract
Clinical reports show that COVID-19 disease has impacts on the cardiovascular system in addition to the respiratory system. Available COVID-19 diagnostic methods have been shown to have limitations. In addition to current diagnostic methods such as low-sensitivity standard RT-PCR tests and expensive medical imaging devices, the development of alternative methods for the diagnosis of COVID-19 disease would be beneficial for control of the COVID-19 pandemic. Further, it is important to quickly and accurately detect abnormalities caused by COVID-19 on the cardiovascular system via ECG. In this study, the diagnosis of COVID-19 disease is proposed using a novel deep Convolutional Neural Network model by using only ECG trace images created from ECG signals of COVID-19 infected patients based on the abnormalities caused by the COVID-19 virus on the cardiovascular system. An overall classification accuracy of 98.57%, 93.20%, 96.74% and AUC value of 0.9966, 0.9771, 0.9905 is achieved for COVID-19 vs. Normal, COVID-19 vs. Abnormal Heartbeats, COVID-19 vs. Myocardial Infarction binary classification tasks, respectively. In addition, an overall classification accuracy of 86.55% and 83.05% is achieved for COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction and Normal vs. COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction multi-classification tasks. This study is believed to have great potential to speed up the diagnosis and treatment of COVID-19 patients, saving clinicians time and facilitating the control of the pandemic.Entities:
Keywords: COVID-19 diagnosis; Cardiovascular diseases diagnosis; Convolutional neural networks; Deep learning; Electrocardiography; Machine learning
Mesh:
Year: 2022 PMID: 35020175 PMCID: PMC8753334 DOI: 10.1007/s13246-022-01102-w
Source DB: PubMed Journal: Phys Eng Sci Med ISSN: 2662-4729
Dataset description
| Number of images in the original dataset | Number of images used in this study | |
|---|---|---|
| Normal | 859 | 713 |
| Myocardial infarction | 77 | 77 |
| Recovered myocardial infarction | 203 | 187 |
| Abnormal heartbeat | 548 | 500 |
| COVID-19 | 250 | 218 |
Number of images used in this study for each classification task before and after data augmentation
| Classification task | Number of images before data augmentation | Number of images after data augmentation | |
|---|---|---|---|
| Task-1 | COVID-19 | 218 | 650 |
| Normal | 713 | 720 | |
| Task-2 | COVID-19 | 218 | 650 |
| Abnormal heartbeats | 500 | 600 | |
| Task-3 | COVID-19 | 218 | 230 |
| Myocardial infarction | 77 | 230 | |
| Task-4 | COVID-19 | 218 | 650 |
| Abnormal heartbeats | 500 | 500 | |
| Myocardial infarction | 77 | 300 | |
| Task-5 | Normal | 713 | 720 |
| COVID-19 | 218 | 650 | |
| Abnormal heartbeats | 500 | 500 | |
| Myocardial infarction | 77 | 300 |
Fig. 1Sample ECG trace images from dataset, a COVID-19 ECG, b Normal ECG, c Abnormal Heartbeats ECG and d Myocardial ECG
Fig. 2The proposed CNN architecture
Fig. 3Activations of convolutional layer
The structure of the proposed network
| CNN layer | Output shape | Parameter # | |
|---|---|---|---|
| 1 | Input | 224 × 224 × 3 | 0 |
| 2 | Convolutional | 222 × 222 × 96 | 2688 |
| 3 | ReLU | 222 × 222 × 96 | 0 |
| 4 | Normalization | 222 × 222 × 96 | 0 |
| 5 | Max pooling | 109 × 109 × 96 | 0 |
| 6 | Convolutional | 111 × 111 × 192 | 83,136 |
| 7 | ReLU | 111 × 111 × 192 | 0 |
| 8 | Max pooling | 54 × 54 × 192 | 0 |
| 9 | Convolutional | 56 × 56 × 256 | 221,440 |
| 10 | ReLU | 56 × 56 × 256 | 0 |
| 11 | Max pooling | 25 × 25 × 96 | 0 |
| 12 | Convolutional | 27 × 27 × 256 | 295,168 |
| 13 | ReLU | 27 × 27 × 256 | 0 |
| 14 | Max pooling | 11 × 11 × 256 | 0 |
| 15 | Flatten | 1 × 1 × 30976 | 0 |
| 16 | FC | 1 × 1 × 4096 | 126,881,792 |
| 17 | ReLU | 1 × 1 × 4096 | 0 |
| 18 | Dropout | 1 × 1 × 4096 | 0 |
| 19 | FC | 1 × 1 × 2 | 8194 |
| 20 | Softmax | 1 × 1 × 2 | 0 |
| 21 | Classification | – | 0 |
Fig. 4Accuracy and Loss plot for classification of a COVID-19 vs. Normal, b COVID-19 vs. Abnormal Heartbeats, c COVID-19 vs. Myocardial Infarction, d COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction and e Normal vs. COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction
Fig. 5Confusion Matrix for classification of a COVID-19 vs. Normal, b COVID-19 vs. Abnormal Heartbeats, c COVID-19 vs. Myocardial Infarction, d COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction and e Normal vs. COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction
Fig. 6ROC Curve for classification of a COVID-19 vs. Normal, b COVID-19 vs. Abnormal Heartbeats, c COVID-19 vs. Myocardial Infarction, d COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction and e Normal vs. COVID-19 vs. Abnormal Heartbeats vs. Myocardial Infarction
Performance evaluation metrics in terms of TP, TN, FP, FN, accuracy, AUC, specificity, sensitivity and precision
| COVID-19 vs normal | COVID-19 vs. abnormal heartbeats | COVID-19 vs. myocardial infarction | COVID-19 vs. abnormal heartbeats vs. myocardial infarction | Normal vs. COVID-19 vs. abnormal heartbeats vs. myocardial infarction | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| TP | 129 | 147 | 122 | 111 | 46 | 43 | 117 | 90 | 44 | 114 | 124 | 83 | 40 |
| TN | 147 | 129 | 111 | 122 | 43 | 46 | 159 | 165 | 217 | 267 | 305 | 301 | 368 |
| FP | 3 | 1 | 9 | 8 | 3 | 0 | 1 | 25 | 13 | 23 | 5 | 39 | 12 |
| FN | 1 | 3 | 8 | 9 | 0 | 3 | 13 | 10 | 16 | 36 | 6 | 17 | 20 |
| Total | 130 | 150 | 130 | 120 | 46 | 46 | 130 | 100 | 60 | 150 | 130 | 100 | 60 |
| Accuracy% ± CI% | 98.57 ± 1.14 | 93.20 ± 2.73 | 96.74 ± 3.54 | 86.55 ± 3.47 | 83.05 ± 3.96 | ||||||||
| AUC (%) | 99.66 ± 0.26 | 97.71 ± 1.82 | 99.05 ± 0.68 | NA | NA | ||||||||
| Specificity (%) | 98.00 ± 1.29 | 92.50 ± 3.39 | 93.48 ± 5.10 | 99.38 ± 0.76 | 98.39 ± 1.64 | ||||||||
| Sensitivity (%) | 99.23 ± 0.59 | 93.85 ± 3.99 | 99.89 ± 0.60 | 90.00 ± 6.27 | 95.39 ± 3.59 | ||||||||
| Precision (%) | 97.73 ± 1.91 | 93.13 ± 4.24 | 93.88 ± 4.20 | 99.15 ± 0.87 | 96.12 ± 3.59 | ||||||||
CI 95% confidence interval, AUC area under the curve, TP true positive, TN true negative, FP false positive, FN false negative
Overall accuracy (%) results of different classification tasks for different well-known CNN models
| COVID-19 vs normal | COVID-19 vs abnormal heartbeats | COVID-19 vs myocardial infarction | COVID-19 vs abnormal heartbeats vs. myocardial infarction | Normal vs COVID-19 vs abnormal heartbeats vs myocardial infarction | |
|---|---|---|---|---|---|
| ResNet-50 | 96.22 ± 1.41 | 91.52 ± 2.72 | 95.91 ± 3.50 | 83.45 ± 5.04 | 78.08 ± 3.21 |
| ResNet-101 | 97.43 ± 2.10 | 92.60 ± 4.51 | 94.43 ± 4.21 | 82.51 ± 4.66 | 80.76 ± 9.53 |
| DenseNet | 96.27 ± 2.48 | 89.72 ± 6.05 | 93.92 ± 5.85 | 86.38 ± 4.35 | 76.83 ± 4.87 |
| InceptionV3 | 95.90 ± 3.51 | 91.67 ± 5.67 | 92.86 ± 8.93 | 87.61 ± 4.12 | 79.35 ± 4.23 |
| VGG-16 | 97.79 ± 2.00 | 92.07 ± 3.97 | 95.29 ± 4.75 | 88.75 ± 3.45 | 83.74 ± 3.38 |
| VGG-19 | 96.71 ± 4.01 | 91.80 ± 4.86 | 94.14 ± 4.21 | 86.95 ± 11.17 | 83.32 ± 11.15 |
| Proposed model | 98.57 ± 1.41 | 93.20 ± 2.73 | 96.74 ± 2.54 | 86.55 ± 2.47 | 83.05 ± 2.96 |