arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.02869eess.IVcs.CVcs.LG
收藏

视觉诊断与预后判断的信息瓶颈归因

Information Bottleneck Attribution for Visual Explanations of Diagnosis and Prognosis

Northwestern University(西北大学);Ege University(爱琴海大学);Tokat State Hospital(托卡特国立医院);NVIDIA(英伟达);University of Catania(卡塔尼亚大学);Columbia University Medical Center(哥伦比亚大学医学中心);National Cancer Institute, National Institutes of Health(美国国立卫生研究院国家癌症研究所)

Ugur Demir, Ismail Irmakci, Elif Keles, Ahmet Topcu, Ziyue Xu, Concetto Spampinato, Sachin Jambawalikar, Evrim Turkbey, Baris Turkbey, Ulas Bagci

摘要:

视觉解释方法在注释数据有限或不可用的患者的预后中具有重要作用。已经有几次尝试使用基于梯度的归因方法来定位来自医学扫描的病理,而不使用分割标签。这一研究方向一直受到缺乏鲁棒性和可靠性的阻碍。这些方法对网络参数高度敏感。在这项研究中,我们引入了一个强大的视觉解释方法来解决这个问题的医疗应用。我们提供了一种用于通用目的的创新视觉解释算法,作为示例应用,我们证明了其在不使用密集分割标签的情况下以高准确性和鲁棒性量化由Covid-19引起的肺部病变的有效性。这种方法克服了常用的Grad-CAM及其扩展版本的缺点。我们提出的策略背后的前提是,在确保分类器预测保持相似的同时,信息流被最小化。我们的研究结果表明,瓶颈条件提供了一个更稳定的严重性估计比类似的归因方法。

英文摘要:

Visual explanation methods have an important role in the prognosis of the patients where the annotated data is limited or unavailable. There have been several attempts to use gradient-based attribution methods to localize pathology from medical scans without using segmentation labels. This research direction has been impeded by the lack of robustness and reliability. These methods are highly sensitive to the network parameters. In this study, we introduce a robust visual explanation method to address this problem for medical applications. We provide an innovative visual explanation algorithm for general purpose and as an example application, we demonstrate its effectiveness for quantifying lesions in the lungs caused by the Covid-19 with high accuracy and robustness without using dense segmentation labels. This approach overcomes the drawbacks of commonly used Grad-CAM and its extended versions. The premise behind our proposed strategy is that the information flow is minimized while ensuring the classifier prediction stays similar. Our findings indicate that the bottleneck condition provides a more stable severity estimation than the similar attribution methods.

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

↑