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

用于可解释医学图像诊断的概念互补瓶颈模型

Concept Complement Bottleneck Model for Interpretable Medical Image Diagnosis

  • Hong Kong University of Science and Technology(香港科技大学)
  • Department of Computer Science and Engineering(计算机科学与工程系)
  • Department of Chemical and Biological Engineering(化学与生物工程系)
  • Division of Life Science(生命科学部)

机构由 AI 辅助整理,请以论文原文为准。

Hongmei Wang, Junlin Hou, Hao Chen

更新

AI总结:

本文提出一种用于可解释医学图像诊断的概念互补瓶颈模型,通过概念适配器和互补策略挖掘并学习新概念以补充现有概念集,实验表明其在概念检测与疾病诊断任务上优于现有方法且具备有效可解释性。

AI中文摘要:

基于人类可理解概念的模型在提升医学图像分析领域中可信赖人工智能的模型可解释性方面受到了广泛关注。这些方法能为模型决策提供令人信服的解释,但严重依赖对预定义概念的详细标注。因此,在概念或标注不完整或质量低下的情况下,它们可能无效。尽管一些方法自动发现有效的新视觉概念而非使用预定义概念,或能通过大语言模型找到一些人类可理解的概念,但它们容易偏离医学诊断证据且难以理解。本文提出一种用于可解释医学图像诊断的概念互补瓶颈模型,旨在补充现有概念集并寻找弥合可解释模型之间差距的新概念。具体而言,我们提出针对特定概念使用概念适配器,在其各自的注意力通道中挖掘概念差异并对概念评分,以支持近乎公平的概念学习。然后,我们设计一种概念互补策略,在学习新概念的同时联合使用已知概念以提升模型性能。在医学数据集上的综合实验表明,我们的模型在概念检测和疾病诊断任务中优于最先进的竞争者,同时提供多样化的解释以有效确保模型可解释性。

英文摘要:

Models based on human-understandable concepts have received extensive attention to improve model interpretability for trustworthy artificial intelligence in the field of medical image analysis. These methods can provide convincing explanations for model decisions but heavily rely on the detailed annotation of pre-defined concepts. Consequently, they may not be effective in cases where concepts or annotations are incomplete or low-quality. Although some methods automatically discover effective and new visual concepts rather than using pre-defined concepts or could find some human-understandable concepts via large Language models, they are prone to veering away from medical diagnostic evidence and are challenging to understand. In this paper, we propose a concept complement bottleneck model for interpretable medical image diagnosis with the aim of complementing the existing concept set and finding new concepts bridging the gap between explainable models. Specifically, we propose to use concept adapters for specific concepts to mine the concept differences and score concepts in their own attention channels to support almost fairly concept learning. Then, we devise a concept complement strategy to learn new concepts while jointly using known concepts to improve model performance. Comprehensive experiments on medical datasets demonstrate that our model outperforms the state-of-the-art competitors in concept detection and disease diagnosis tasks while providing diverse explanations to ensure model interpretability effectively.

补充信息

↑