Explainable COVID-19 Detection Using Chest CT Scans and Deep Learning
arXiv:2011.05317 · doi:10.3390/s21020455
Abstract
This paper explores how well deep learning models trained on chest CT images can diagnose COVID-19 infected people in a fast and automated process. To this end, we adopt advanced deep network architectures and propose a transfer learning strategy using custom-sized input tailored for each deep architecture to achieve the best performance. We conduct extensive sets of experiments on two CT image datasets, namely the SARS-CoV-2 CT-scan and the COVID19-CT. The obtained results show superior performances for our models compared with previous studies, where our best models achieve average accuracy, precision, sensitivity, specificity and F1 score of 99.4%, 99.6%, 99.8%, 99.6% and 99.4% on the SARS-CoV-2 dataset; and 92.9%, 91.3%, 93.7%, 92.2% and 92.5% on the COVID19-CT dataset, respectively. Furthermore, we apply two visualization techniques to provide visual explanations for the models' predictions. The visualizations show well-separated clusters for CT images of COVID-19 from other lung diseases, and accurate localizations of the COVID-19 associated regions.
References in corpus (9)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
- CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning
- COVID-19 identification in chest X-ray images on flat and hierarchical classification scenarios
- What makes ImageNet good for transfer learning?
- Contrastive Cross-site Learning with Redesigned Net for COVID-19 CT Classification
- COVID CT-Net: Predicting Covid-19 From Chest CT Images Using Attentional Convolutional Network
- CVR-Net: A deep convolutional neural network for coronavirus recognition from chest radiography images
- Classification of COVID-19 in CT Scans using Multi-Source Transfer Learning