UncertaintyFuseNet: Robust Uncertainty-aware Hierarchical Feature Fusion Model with Ensemble Monte Carlo Dropout for COVID-19 Detection

Abstract

The COVID-19 (Coronavirus disease 2019) pandemic has become a major global threat to human health and well-being. Thus, the development of computer-aided detection (CAD) systems that are capable to accurately distinguish COVID-19 from other diseases using chest computed tomography (CT) and X-ray data is of immediate priority. Such automatic systems are usually based on traditional machine learning or deep learning methods. Differently from most of existing studies, which used either CT scan or X-ray images in COVID-19-case classification, we present a simple but efficient deep learning feature fusion model, called UncertaintyFuseNet, which is able to classify accurately large datasets of both of these types of images. We argue that the uncertainty of the model's predictions should be taken into account in the learning process, even though most of existing studies have overlooked it. We quantify the prediction uncertainty in our feature fusion model using effective Ensemble MC Dropout (EMCD) technique. A comprehensive simulation study has been conducted to compare the results of our new model to the existing approaches, evaluating the performance of competing models in terms of Precision, Recall, F-Measure, Accuracy and ROC curves. The obtained results prove the efficiency of our model which provided the prediction accuracy of 99.08\% and 96.35\% for the considered CT scan and X-ray datasets, respectively. Moreover, our UncertaintyFuseNet model was generally robust to noise and performed well with previously unseen data. The source code of our implementation is freely available at: https://github.com/moloud1987/UncertaintyFuseNet-for-COVID-19-Classification.

arXiv comment16 pages, 18 figures

References in corpus (16)

• A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challenges
• Neural Architecture Search: A Survey
• CoroNet: A deep neural network for detection and diagnosis of COVID-19 from chest x-ray images
• Medical Image Fusion: A survey of the state of the art
• Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networks
• A modified deep convolutional neural network for detecting COVID-19 and pneumonia from chest X-ray images based on the concatenation of Xception and ResNet50V2
• COVID-19 identification in chest X-ray images on flat and hierarchical classification scenarios
• COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs
• Improving performance of CNN to predict likelihood of COVID-19 using chest X-ray images with preprocessing algorithms
• A Review of Generalized Zero-Shot Learning Methods
• Estimating Uncertainty and Interpretability in Deep Learning for Coronavirus (COVID-19) Detection
• Residual Attention U-Net for Automated Multi-Class Segmentation of COVID-19 Chest CT Images
• An Uncertainty-aware Transfer Learning-based Framework for Covid-19 Diagnosis
• Partial Connection Based on Channel Attention for Differentiable Neural Architecture Search
• Analyzing the Role of Model Uncertainty for Electronic Health Records
• Classification of COVID-19 in CT Scans using Multi-Source Transfer Learning

Cited by in corpus (1)

• Automated Detection and Forecasting of COVID-19 using Deep Learning Techniques: A Review