通过补全概念集提升医学影像诊断可解释性
Concept Complement Bottleneck Model for Interpretable Medical Image Diagnosis
- 用概念适配器捕捉特定概念差异并独立评分
- 联合已知与新发现概念,提升诊断准确率
- 适合需要可解释性的医疗AI研发与临床应用
基于人类可理解概念的模型在医学图像分析中备受关注,有助于提升可信AI的可解释性。这类方法虽能提供合理决策解释,但严重依赖预定义概念的精细标注,难以应对概念缺失或标注质量差的情况。尽管部分方法可通过大语言模型自动发现新概念,但常偏离医学诊断证据且难以理解。本文提出一种概念补全瓶颈模型,旨在补全现有概念集并发现连接解释模型的新概念。具体而言,利用概念适配器对特定概念进行差异挖掘,并在各自注意力通道中评分,支持公平的概念学习;设计概念补全策略,在使用已知概念的同时学习新概念,以提升模型性能。在多个医学数据集上的实验表明,该模型在概念检测和疾病诊断任务上均优于当前最优方法,同时提供多样化的解释,有效保障模型可解释性。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。