Estimating Uncertainty and Interpretability in Deep Learning for Coronavirus (COVID-19) Detection

Abstract

Deep Learning has achieved state of the art performance in medical imaging. However, these methods for disease detection focus exclusively on improving the accuracy of classification or predictions without quantifying uncertainty in a decision. Knowing how much confidence there is in a computer-based medical diagnosis is essential for gaining clinicians trust in the technology and therefore improve treatment. Today, the 2019 Coronavirus (SARS-CoV-2) infections are a major healthcare challenge around the world. Detecting COVID-19 in X-ray images is crucial for diagnosis, assessment and treatment. However, diagnostic uncertainty in the report is a challenging and yet inevitable task for radiologist. In this paper, we investigate how drop-weights based Bayesian Convolutional Neural Networks (BCNN) can estimate uncertainty in Deep Learning solution to improve the diagnostic performance of the human-machine team using publicly available COVID-19 chest X-ray dataset and show that the uncertainty in prediction is highly correlates with accuracy of prediction. We believe that the availability of uncertainty-aware deep learning solution will enable a wider adoption of Artificial Intelligence (AI) in a clinical setting.

References in corpus (5)

• Weight Uncertainty in Neural Networks
• Rapid AI Development Cycle for the Coronavirus (COVID-19) Pandemic: Initial Results for Automated Detection & Patient Monitoring using Deep Learning CT Image Analysis
• Sanity Checks for Saliency Maps
• Understanding Measures of Uncertainty for Adversarial Example Detection
• Loss-Calibrated Approximate Inference in Bayesian Neural Networks

Cited by in corpus (32)

• Artificial Intelligence (AI) and Big Data for Coronavirus (COVID-19) Pandemic: A Survey on the State-of-the-Arts
• Blockchain and AI-based Solutions to Combat Coronavirus (COVID-19)-like Epidemics: A Survey
• An Uncertainty-aware Transfer Learning-based Framework for Covid-19 Diagnosis
• UncertaintyFuseNet: Robust Uncertainty-aware Hierarchical Feature Fusion Model with Ensemble Monte Carlo Dropout for COVID-19 Detection
• Study of Different Deep Learning Approach with Explainable AI for Screening Patients with COVID-19 Symptoms: Using CT Scan and Chest X-ray Image Dataset
• Modelling the second wave of COVID-19 infections in France and Italy via a Stochastic SEIR model
• Diagnosis of COVID-19 Using Machine Learning and Deep Learning: A review
• Automatic Detection of Coronavirus Disease (COVID-19) in X-ray and CT Images: A Machine Learning-Based Approach
• Checklist for responsible deep learning modeling of medical images based on COVID-19 detection studies
• Robust Screening of COVID-19 from Chest X-ray via Discriminative Cost-Sensitive Learning
• COVID-19 detection from scarce chest x-ray image data using few-shot deep learning approach
• Medical Imaging and Computational Image Analysis in COVID-19 Diagnosis: A Review
• DeepCOVIDExplainer: Explainable COVID-19 Diagnosis Based on Chest X-ray Images
• Medical Imaging with Deep Learning for COVID- 19 Diagnosis: A Comprehensive Review
• CVR-Net: A deep convolutional neural network for coronavirus recognition from chest radiography images
• COVIDGR dataset and COVID-SDNet methodology for predicting COVID-19 based on Chest X-Ray images
• COVID-MobileXpert: On-Device COVID-19 Patient Triage and Follow-up using Chest X-rays
• Understanding the COVID19 Outbreak: A Comparative Data Analytics and Study
• Learning to Synthesize Programs as Interpretable and Generalizable Policies
• Confidence Aware Neural Networks for Skin Cancer Detection
• Automated Model Design and Benchmarking of 3D Deep Learning Models for COVID-19 Detection with Chest CT Scans
• A Survey on the Use of AI and ML for Fighting the COVID-19 Pandemic
• ViDi: Descriptive Visual Data Clustering as Radiologist Assistant in COVID-19 Streamline Diagnostic
• Shapley variable importance clouds for interpretable machine learning
• REDAffectiveLM: Leveraging Affect Enriched Embedding and Transformer-based Neural Language Model for Readers' Emotion Detection
• Uncertainty-Aware COVID-19 Detection from Imbalanced Sound Data
• Multiscale Attention Guided Network for COVID-19 Diagnosis Using Chest X-ray Images
• Looking at the posterior: accuracy and uncertainty of neural-network predictions
• Semi-supervised Learning for COVID-19 Image Classification via ResNet
• Reducing Risk and Uncertainty of Deep Neural Networks on Diagnosing COVID-19 Infection
• Twin Augmented Architectures for Robust Classification of COVID-19 Chest X-Ray Images
• Classification of COVID-19 from CXR Images in a 15-class Scenario: an Attempt to Avoid Bias in the System