arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.20511cs.CV

利用因果推理方法解释医学图像分割模型

Leveraging Causal Reasoning Method for Explaining Medical Image Segmentation Models

Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences(中国科学院深圳先进技术研究所);University of Chinese Academy of Sciences(中国科学院大学);Shenzhen University of Advanced Technology(深圳先进技术大学);Shenzhen Polytechnic University(深圳职业技术大学);University of Macau(澳门大学);Southern University of Science and Technology(南方科技大学)

Limai Jiang, Ruitao Xie, Bokai Yang, Huazhen Huang, Juan He, Yufu Huo, Zikai Wang, Yang Wei, Yunpeng Cai

AI总结:

本文提出利用因果推理框架解释医学图像分割模型,通过量化输入区域和网络组件对分割区域的影响,提升模型可解释性与优化潜力。

摘要:

医学图像分割在临床决策中起着至关重要的作用,能够精确定位病变并指导干预。尽管在分割准确性上取得了显著进展,但大多数深度模型的黑盒性质引发了对在高风险医疗场景中可靠性的日益关注。当前的解释技术主要集中在分类任务上,使分割领域相对未被充分探索。我们引入了一种用于分割任务的解释模型,该模型采用因果推理框架,并将平均处理效应(ATE)反向传播到量化指标中,以确定输入区域和网络组件对目标分割区域的影响。通过与最近的分割可解释性技术在两个代表性医学成像数据集上的比较,我们证明了我们的方法比现有方法提供了更忠实的解释。此外,我们使用我们的方法对多个基础分割模型进行了系统的因果分析,揭示了不同模型之间在感知策略上的显著异质性,甚至在相同模型的不同输入之间也是如此。这表明我们的方法有潜力为优化分割模型提供重要的见解。我们的代码可在https://github.com/lcmmai/PdCR上找到。

英文摘要:

Medical image segmentation plays a vital role in clinical decision-making, enabling precise localization of lesions and guiding interventions. Despite significant advances in segmentation accuracy, the black-box nature of most deep models has raised growing concerns about their trustworthiness in high-stakes medical scenarios. Current explanation techniques have primarily focused on classification tasks, leaving the segmentation domain relatively underexplored. We introduced an explanation model for segmentation task which employs the causal inference framework and backpropagates the average treatment effect (ATE) into a quantification metric to determine the influence of input regions, as well as network components, on target segmentation areas. Through comparison with recent segmentation explainability techniques on two representative medical imaging datasets, we demonstrated that our approach provides more faithful explanations than existing approaches. Furthermore, we carried out a systematic causal analysis of multiple foundational segmentation models using our method, which reveals significant heterogeneity in perceptual strategies across different models, and even between different inputs for the same model. Suggesting the potential of our method to provide notable insights for optimizing segmentation models. Our code can be found at https://github.com/lcmmai/PdCR.

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

↑