A CNN based coronavirus disease prediction system for chest X-rays

J Ambient Intell Humaniz Comput. 2022 Feb 27:1-15. doi: 10.1007/s12652-022-03775-3. Online ahead of print.

Abstract

Coronavirus disease (COVID-19) proliferated globally in early 2020, causing existential dread in the whole world. Radiography is crucial in the clinical staging and diagnosis of COVID-19 and offers high potential to improve healthcare plans for tackling the pandemic. However high variations in infection characteristics and low contrast between normal and infected regions pose great challenges in preparing radiological reports. To address these challenges, this study presents CODISC-CNN (CNN based Coronavirus DIsease Prediction System for Chest X-rays) that can automatically extract the features from chest X-ray images for the disease prediction. However, to get the infected region of X-ray, edges of the images are detected by applying image preprocessing. Furthermore, to attenuate the shortage of labeled datasets data augmentation has been adapted. Extensive experiments have been performed to classify X-ray images into two classes (Normal and COVID), three classes (Normal, COVID, and Virus Bacteria), and four classes (Normal, COVID, and Virus Bacteria, and Virus Pneumonia) with the accuracy of 97%, 89%, and 84% respectively. The proposed CNN-based model outperforms many cutting-edge classification models and boosts state-of-the-art performance.

Keywords: COVID-19; Convolutional neural network; Deep learning; X-ray Images.