Explainable Deep Neural Models for COVID-19 Prediction from Chest X-Rays with Region of Interest Visualization

COVID-19 has been designated as a once-in-a-century pandemic, and its impact is still being felt severely in many countries, due to the extensive human and green casualties. While several vaccines are under various stage of development, effective screening procedures that help detect the disease at early stages in a non-invasive and resource-optimized manner are the need of the hour. X-ray imaging is fairly accessible in most healthcare institutions and can prove useful in diagnosing this respiratory disease. Although a chest X-ray scan is a viable method to detect the presence of this disease, the scans must be analyzed by trained experts accurately and quickly if large numbers of tests are to be processed. In this paper, a benchmarking study of different preprocessing techniques and state-of-the-art deep learning models is presented to provide comprehensive insights into both the objective and subjective evaluation of their performance. To analyze and prevent possible sources of bias, we preprocessed the dataset in two ways - first, we segmented the lungs alone, and secondly, we formed a bounding box around the lung and used only this area to train. Among the models chosen to benchmark, which were DenseNet201, EfficientNetB7, and VGG-16, DenseNet201 performed better for all three datasets.

[1]  Raymond Y Huang,et al.  Artificial Intelligence Augmentation of Radiologist Performance in Distinguishing COVID-19 from Pneumonia of Other Origin at Chest CT , 2021, Radiology.

[2]  Tahmina Zebin,et al.  COVID-19 detection and disease progression visualization: Deep learning on chest X-rays for classification and coarse localization , 2020, Appl. Intell..

[3]  Dilbag Singh,et al.  Classification of the COVID-19 infected patients using DenseNet201 based deep transfer learning , 2020, Journal of biomolecular structure & dynamics.

[4]  Mohammad Farukh Hashmi,et al.  Efficient Pneumonia Detection in Chest Xray Images Using Deep Transfer Learning , 2020, Diagnostics.

[5]  Mohammad Rahimzadeh,et al.  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 , 2020, Informatics in Medicine Unlocked.

[6]  U. Rajendra Acharya,et al.  Automated detection of COVID-19 cases using deep neural networks with X-ray images , 2020, Computers in Biology and Medicine.

[7]  Vinay Namboodiri,et al.  CovidAID: COVID-19 Detection Using Chest X-Ray , 2020, ArXiv.

[8]  Joseph Paul Cohen,et al.  COVID-19 Image Data Collection , 2020, ArXiv.

[9]  Prajoy Podder,et al.  Hybrid deep learning for detecting lung diseases from X-ray images , 2020, Informatics in Medicine Unlocked.

[10]  P. Agarwal,et al.  Chest Imaging Appearance of COVID-19 Infection , 2020, Radiology. Cardiothoracic imaging.

[11]  Quoc V. Le,et al.  EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks , 2019, ICML.

[12]  Thuy Thanh Nguyen,et al.  Applying Multi-CNNs model for detecting abnormal problem on chest x-ray images , 2018, 2018 10th International Conference on Knowledge and Systems Engineering (KSE).

[13]  Yong Wu,et al.  Convolution Neural Network based Transfer Learning for Classification of Flowers , 2018, 2018 IEEE 3rd International Conference on Signal and Image Processing (ICSIP).

[14]  Michael H. Goldbaum,et al.  Labeled Optical Coherence Tomography (OCT) and Chest X-Ray Images for Classification , 2018 .

[15]  Marcia Hon,et al.  Towards Alzheimer's disease classification through transfer learning , 2017, 2017 IEEE International Conference on Bioinformatics and Biomedicine (BIBM).

[16]  Ramprasaath R. Selvaraju,et al.  Grad-CAM: Why did you say that? Visual Explanations from Deep Networks via Gradient-based Localization , 2016 .

[17]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[18]  Fei-Fei Li,et al.  ImageNet: A large-scale hierarchical image database , 2009, 2009 IEEE Conference on Computer Vision and Pattern Recognition.