Exploration of Interpretability Techniques for Deep COVID-19 Classification using Chest X-ray Images

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chatterjee, Soumick, Saad, Fatima, Sarasaen, Chompunuch, Ghosh, Suhita, Krug, Valerie, Khatun, Rupali, Mishra, Rahul, Desai, Nirja, Radeva, Petia, Rose, Georg, Stober, Sebastian, Speck, Oliver, Nürnberger, Andreas
Format: Preprint
Veröffentlicht: 2020
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914672261726208
author Chatterjee, Soumick
Saad, Fatima
Sarasaen, Chompunuch
Ghosh, Suhita
Krug, Valerie
Khatun, Rupali
Mishra, Rahul
Desai, Nirja
Radeva, Petia
Rose, Georg
Stober, Sebastian
Speck, Oliver
Nürnberger, Andreas
author_facet Chatterjee, Soumick
Saad, Fatima
Sarasaen, Chompunuch
Ghosh, Suhita
Krug, Valerie
Khatun, Rupali
Mishra, Rahul
Desai, Nirja
Radeva, Petia
Rose, Georg
Stober, Sebastian
Speck, Oliver
Nürnberger, Andreas
contents The outbreak of COVID-19 has shocked the entire world with its fairly rapid spread and has challenged different sectors. One of the most effective ways to limit its spread is the early and accurate diagnosing infected patients. Medical imaging, such as X-ray and Computed Tomography (CT), combined with the potential of Artificial Intelligence (AI), plays an essential role in supporting medical personnel in the diagnosis process. Thus, in this article five different deep learning models (ResNet18, ResNet34, InceptionV3, InceptionResNetV2 and DenseNet161) and their ensemble, using majority voting have been used to classify COVID-19, pneumoniæ and healthy subjects using chest X-ray images. Multilabel classification was performed to predict multiple pathologies for each patient, if present. Firstly, the interpretability of each of the networks was thoroughly studied using local interpretability methods - occlusion, saliency, input X gradient, guided backpropagation, integrated gradients, and DeepLIFT, and using a global technique - neuron activation profiles. The mean Micro-F1 score of the models for COVID-19 classifications ranges from 0.66 to 0.875, and is 0.89 for the ensemble of the network models. The qualitative results showed that the ResNets were the most interpretable models. This research demonstrates the importance of using interpretability methods to compare different models before making a decision regarding the best performing model.
format Preprint
id arxiv_https___arxiv_org_abs_2006_02570
institution arXiv
publishDate 2020
record_format arxiv
spellingShingle Exploration of Interpretability Techniques for Deep COVID-19 Classification using Chest X-ray Images
Chatterjee, Soumick
Saad, Fatima
Sarasaen, Chompunuch
Ghosh, Suhita
Krug, Valerie
Khatun, Rupali
Mishra, Rahul
Desai, Nirja
Radeva, Petia
Rose, Georg
Stober, Sebastian
Speck, Oliver
Nürnberger, Andreas
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
The outbreak of COVID-19 has shocked the entire world with its fairly rapid spread and has challenged different sectors. One of the most effective ways to limit its spread is the early and accurate diagnosing infected patients. Medical imaging, such as X-ray and Computed Tomography (CT), combined with the potential of Artificial Intelligence (AI), plays an essential role in supporting medical personnel in the diagnosis process. Thus, in this article five different deep learning models (ResNet18, ResNet34, InceptionV3, InceptionResNetV2 and DenseNet161) and their ensemble, using majority voting have been used to classify COVID-19, pneumoniæ and healthy subjects using chest X-ray images. Multilabel classification was performed to predict multiple pathologies for each patient, if present. Firstly, the interpretability of each of the networks was thoroughly studied using local interpretability methods - occlusion, saliency, input X gradient, guided backpropagation, integrated gradients, and DeepLIFT, and using a global technique - neuron activation profiles. The mean Micro-F1 score of the models for COVID-19 classifications ranges from 0.66 to 0.875, and is 0.89 for the ensemble of the network models. The qualitative results showed that the ResNets were the most interpretable models. This research demonstrates the importance of using interpretability methods to compare different models before making a decision regarding the best performing model.
title Exploration of Interpretability Techniques for Deep COVID-19 Classification using Chest X-ray Images
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2006.02570