Mask of Truth: Model Sensitivity to Unexpected Regions of Medical Images.

Journal: Journal of imaging informatics in medicine
Published Date:

Abstract

The development of larger models for medical image analysis has led to increased performance. However, it also affected our ability to explain and validate model decisions. Models can use non-relevant parts of images, also called spurious correlations or shortcuts, to obtain high performance on benchmark datasets but fail in real-world scenarios. In this work, we challenge the capacity of convolutional neural networks (CNN) to classify chest X-rays and eye fundus images while masking out clinically relevant parts of the image. We show that all models trained on the PadChest dataset, irrespective of the masking strategy, are able to obtain an area under the curve (AUC) above random. Moreover, the models trained on full images obtain good performance on images without the region of interest (ROI), even superior to the one obtained on images only containing the ROI. We also reveal a possible spurious correlation in the Chákṣu dataset while the performances are more aligned with the expectation of an unbiased model. We go beyond the performance analysis with the usage of the explainability method SHAP and the analysis of embeddings. We asked a radiology resident to interpret chest X-rays under different masking to complement our findings with clinical knowledge.

Authors

  • Théo Sourget
    IT University of Copenhagen, Copenhagen, Denmark. tsou@itu.dk.
  • Michelle Hestbek-Møller
    IT University of Copenhagen, Copenhagen, Denmark.
  • Amelia Jiménez-Sánchez
    BCN MedTech, Department of Information and Communication Technologies, Universitat Pompeu Fabra, Barcelona, Spain. Electronic address: amelia.jimenez@upf.edu.
  • Jack Junchi Xu
    Copenhagen University Hospital, Herlev and Gentofte, Copenhagen, Denmark.
  • Veronika Cheplygina
    Medical Image Analysis, Department Biomedical Engineering, Eindhoven University of Technology, Eindhoven, the Netherlands. Electronic address: v.cheplygina@tue.nl.

Keywords

No keywords available for this article.