Explainable Artificial Intelligence for Bias Detection in COVID CT-Scan Classifiers

Sensors (Basel). 2021 Aug 23;21(16):5657. doi: 10.3390/s21165657.

Abstract

Problem: An application of Explainable Artificial Intelligence Methods for COVID CT-Scan classifiers is presented.

Motivation: It is possible that classifiers are using spurious artifacts in dataset images to achieve high performances, and such explainable techniques can help identify this issue.

Aim: For this purpose, several approaches were used in tandem, in order to create a complete overview of the classificatios.

Methodology: The techniques used included GradCAM, LIME, RISE, Squaregrid, and direct Gradient approaches (Vanilla, Smooth, Integrated).

Main results: Among the deep neural networks architectures evaluated for this image classification task, VGG16 was shown to be most affected by biases towards spurious artifacts, while DenseNet was notably more robust against them. Further impacts: Results further show that small differences in validation accuracies can cause drastic changes in explanation heatmaps for DenseNet architectures, indicating that small changes in validation accuracy may have large impacts on the biases learned by the networks. Notably, it is important to notice that the strong performance metrics achieved by all these networks (Accuracy, F1 score, AUC all in the 80 to 90% range) could give users the erroneous impression that there is no bias. However, the analysis of the explanation heatmaps highlights the bias.

Keywords: Computerized Tomography; Covid 19; Explainable AI; computer vision; image classification; medical imaging.

MeSH terms

  • Artificial Intelligence*
  • Bias
  • COVID-19*
  • Humans
  • SARS-CoV-2
  • Tomography, X-Ray Computed