CVR-Net: A deep convolutional neural network for coronavirus recognition from chest radiography images
arXiv:2007.11993
Abstract
The novel Coronavirus Disease 2019 (COVID-19) is a global pandemic disease spreading rapidly around the world. A robust and automatic early recognition of COVID-19, via auxiliary computer-aided diagnostic tools, is essential for disease cure and control. The chest radiography images, such as Computed Tomography (CT) and X-ray, and deep Convolutional Neural Networks (CNNs), can be a significant and useful material for designing such tools. However, designing such an automated tool is challenging as a massive number of manually annotated datasets are not publicly available yet, which is the core requirement of supervised learning systems. In this article, we propose a robust CNN-based network, called CVR-Net (Coronavirus Recognition Network), for the automatic recognition of the coronavirus from CT or X-ray images. The proposed end-to-end CVR-Net is a multi-scale-multi-encoder ensemble model, where we have aggregated the outputs from two different encoders and their different scales to obtain the final prediction probability. We train and test the proposed CVR-Net on three different datasets, where the images have collected from different open-source repositories. We compare our proposed CVR-Net with state-of-the-art methods, which are trained and tested on the same datasets. We split three datasets into five different tasks, where each task has a different number of classes, to evaluate the multi-tasking CVR-Net. Our model achieves an overall F1-score & accuracy of 0.997 & 0.998; 0.963 & 0.964; 0.816 & 0.820; 0.961 & 0.961; and 0.780 & 0.780, respectively, for task-1 to task-5. As the CVR-Net provides promising results on the small datasets, it can be an auspicious computer-aided diagnostic tool for the diagnosis of coronavirus to assist the clinical practitioners and radiologists. Our source codes and model are publicly available at https://github.com/kamruleee51/CVR-Net.
31 Pages
References in corpus (14)
- Adam: A Method for Stochastic Optimization
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- A Survey on Deep Learning in Medical Image Analysis
- MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
- Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?
- Covid-19: Automatic detection from X-Ray images utilizing Transfer Learning with Convolutional Neural Networks
- Can AI help in screening Viral and COVID-19 pneumonia?
- CoroNet: A deep neural network for detection and diagnosis of COVID-19 from chest x-ray images
- COVIDX-Net: A Framework of Deep Learning Classifiers to Diagnose COVID-19 in X-Ray Images
- COVID-CT-Dataset: A CT Scan Dataset about COVID-19
- Efficient Neural Architecture Search via Parameter Sharing
- COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs
- Extracting possibly representative COVID-19 Biomarkers from X-Ray images with Deep Learning approach and image data related to Pulmonary Diseases
- Estimating Uncertainty and Interpretability in Deep Learning for Coronavirus (COVID-19) Detection
Cited by in corpus (3)
- Explainable COVID-19 Detection Using Chest CT Scans and Deep Learning
- Detecting COVID-19 from Chest Computed Tomography Scans using AI-Driven Android Application
- CXR-Net: An Encoder-Decoder-Encoder Multitask Deep Neural Network for Explainable and Accurate Diagnosis of COVID-19 pneumonia with Chest X-ray Images