Efficient and Visualizable Convolutional Neural Networks for COVID-19 Classification Using Chest CT
With COVID-19 cases rising rapidly, deep learning has emerged as a promising diagnosis technique. However, identifying the most accurate models to characterize COVID-19 patients is challenging because comparing results obtained with different types of data and acquisition processes is non-trivial. In this paper we designed, evaluated, and compared the performance of 20 convolutional neutral networks in classifying patients as COVID-19 positive, healthy, or suffering from other pulmonary lung infections based on Chest CT scans, serving as the first to consider the EfficientNet family for COVID-19 diagnosis and employ intermediate activation maps for visualizing model performance. All models are trained and evaluated in Python using 4173 Chest CT images from the dataset entitled "A COVID multiclass dataset of CT scans," with 2168, 758, and 1247 images of patients that are COVID-19 positive, healthy, or suffering from other pulmonary infections, respectively. EfficientNet-B5 was identified as the best model with an F1 score of 0.9769+/-0.0046, accuracy of 0.9759+/-0.0048, sensitivity of 0.9788+/-0.0055, specificity of 0.9730+/-0.0057, and precision of 0.9751 +/- 0.0051. On an alternate 2-class dataset, EfficientNetB5 obtained an accuracy of 0.9845+/-0.0109, F1 score of 0.9599+/-0.0251, sensitivity of 0.9682+/-0.0099, specificity of 0.9883+/-0.0150, and precision of 0.9526 +/- 0.0523. Intermediate activation maps and Gradient-weighted Class Activation Mappings offered human-interpretable evidence of the model's perception of ground-class opacities and consolidations, hinting towards a promising use-case of artificial intelligence-assisted radiology tools. With a prediction speed of under 0.1 seconds on GPUs and 0.5 seconds on CPUs, our proposed model offers a rapid, scalable, and accurate diagnostic for COVID-19.
Code (1)
Tasks
COVID-19 DiagnosisDiagnosticGeneral ClassificationSensitivitySpecificityMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
ECOVNet: An Ensemble of Deep Convolutional Neural Networks Based on EfficientNet to Detect COVID-19 From Chest X-rays
This paper proposed an ensemble of deep convolutional neural networks (CNN) based on EfficientNet, named ECOVNet, to detect COVID-19 using a large chest X-ray data set. At first, the open-access large chest X-ray collect…
Classification of COVID-19 in chest X-ray images using DeTraC deep convolutional neural network
Chest X-ray is the first imaging technique that plays an important role in the diagnosis of COVID-19 disease. Due to the high availability of large-scale annotated image datasets, great success has been achieved using co…
General ClassificationMedical DiagnosisObject RecognitionSpecificity+1MIC: Medical Image Classification Using Chest X-ray (COVID-19 and Pneumonia) Dataset with the Help of CNN and Customized CNN
The COVID19 pandemic has had a detrimental impact on the health and welfare of the worlds population. An important strategy in the fight against COVID19 is the effective screening of infected patients, with one of the pr…
image-classificationImage ClassificationMedical Image ClassificationStacked Convolutional Neural Network for Diagnosis of COVID-19 Disease from X-ray Images
Automatic and rapid screening of COVID-19 from the chest X-ray images has become an urgent need in this pandemic situation of SARS-CoV-2 worldwide in 2020. However, accurate and reliable screening of patients is a massiv…
Evaluating Convolutional Neural Networks for COVID-19 classification in chest X-ray images
Coronavirus Disease 2019 (COVID-19) pandemic rapidly spread globally, impacting the lives of billions of people. The effective screening of infected patients is a critical step to struggle with COVID-19, and treating the…
Data Augmentation