Tareque Rahman Ornob, Gourab Roy, Enamul Hassan
Patients with the COVID-19 infection may have pneumonia-like symptoms as well as respiratory problems which may harm the lungs. From medical images, coronavirus illness may be accurately identified and predicted using a variety of machine learning methods. Most of the published machine learning methods may need extensive hyperparameter adjustment and are unsuitable for small datasets. By leveraging the data in a comparatively small dataset, few-shot learning algorithms aim to reduce the requirement of large datasets. This inspired us to develop a few-shot learning model for early detection of COVID-19 to reduce the post-effect of this dangerous disease. The proposed architecture combines few-shot learning with an ensemble of pre-trained convolutional neural networks to extract feature vectors from CT scan images for similarity learning. The proposed Triplet Siamese Network as the few-shot learning model classified CT scan images into Normal, COVID-19, and Community-Acquired Pneumonia. The suggested model achieved an overall accuracy of 98.719%, a specificity of 99.36%, a sensitivity of 98.72%, and a ROC score of 99.9% with only 200 CT scans per category for training data.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | AUC-ROC | 0.9992 | CovidExpert |
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | Accuracy | 0.98719 | CovidExpert |
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | Macro F1 | 0.9872 | CovidExpert |
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | Macro Precision | 0.9873 | CovidExpert |
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | Macro Recall | 0.9872 | CovidExpert |
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | Micro Precision | 0.9872 | CovidExpert |
| Few-Shot Learning | Large COVID-19 CT scan slice dataset | Specificity | 0.9936 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | AUC-ROC | 0.9992 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | Accuracy | 0.98719 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | Macro F1 | 0.9872 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | Macro Precision | 0.9873 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | Macro Recall | 0.9872 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | Micro Precision | 0.9872 | CovidExpert |
| Meta-Learning | Large COVID-19 CT scan slice dataset | Specificity | 0.9936 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | AUC-ROC | 0.9992 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | Accuracy | 0.98719 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | Macro F1 | 0.9872 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | Macro Precision | 0.9873 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | Macro Recall | 0.9872 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | Micro Precision | 0.9872 | CovidExpert |
| COVID-19 Diagnosis | Large COVID-19 CT scan slice dataset | Specificity | 0.9936 | CovidExpert |