arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2101.07563eess.IVcs.CV
收藏

使用StyleGAN在医学图像上实现深度学习模型的视觉可解释性

Using StyleGAN for Visual Interpretability of Deep Learning Models on Medical Images

Owkin, Inc.(Owkin公司)

Kathryn Schutte, Olivier Moindrot, Paul Hérent, Jean-Baptiste Schiratti, Simon Jégou

摘要:

随着基于人工智能的医疗设备在放射学和组织学等成像领域变得越来越普遍,基础预测模型的可解释性对于扩大其在临床实践中的应用至关重要。现有的基于热图的可解释性方法(如GradCAM)仅突出了预测特征的位置,但没有解释它们如何有助于预测。在本文中,我们提出了一种新的可解释性方法,可用于理解任何黑盒模型对图像的预测,通过显示如何修改输入图像以产生不同的预测。StyleGAN在医学图像上进行训练,以提供潜在向量和图像之间的映射。我们的方法识别潜在空间中的最佳方向,以在模型预测中创建变化。通过沿着这个方向移动输入图像的潜在表示,我们可以产生一系列具有变化预测的新合成图像。我们在组织学和放射学图像上验证了我们的方法,并证明了它能够提供比GradCAM热图更有意义的解释。我们的方法揭示了模型学习的模式,这使得临床医生能够建立对模型预测的信任,发现新的生物标志物,并最终揭示潜在的偏见。

英文摘要:

As AI-based medical devices are becoming more common in imaging fields like radiology and histology, interpretability of the underlying predictive models is crucial to expand their use in clinical practice. Existing heatmap-based interpretability methods such as GradCAM only highlight the location of predictive features but do not explain how they contribute to the prediction. In this paper, we propose a new interpretability method that can be used to understand the predictions of any black-box model on images, by showing how the input image would be modified in order to produce different predictions. A StyleGAN is trained on medical images to provide a mapping between latent vectors and images. Our method identifies the optimal direction in the latent space to create a change in the model prediction. By shifting the latent representation of an input image along this direction, we can produce a series of new synthetic images with changed predictions. We validate our approach on histology and radiology images, and demonstrate its ability to provide meaningful explanations that are more informative than GradCAM heatmaps. Our method reveals the patterns learned by the model, which allows clinicians to build trust in the model's predictions, discover new biomarkers and eventually reveal potential biases.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑