用超图建模医学影像概念关系,少标注也能解释诊断结果。
Learning Label-Efficient Interpretable Medical Image Diagnosis via Semi-supervised Hypergraph Concept Bottleneck Model

- 通过双层超图捕捉概念间复杂关系,提升可解释性。
- 在帕拉多斯胎盘谱系数据上达到92.3%准确率,仅需少量标注。
- 适合需要透明决策的临床场景,如超声诊断与皮肤癌筛查。
深度学习虽在医学影像分析中取得卓越诊断精度,但其决策过程缺乏可解释性,阻碍了临床应用,尤其在高风险场景中透明度至关重要。例如,在胎盘植入谱系(PAS)的超声影像中,细微征象难以可靠识别,黑箱模型无法提供可信评分。为此,概念瓶颈模型(CBM)通过引入临床有意义的中间概念,使诊断过程可被医生审查与优化。然而传统CBM难以刻画复杂概念依赖,且需昂贵专家标注,限制其扩展性。本文提出一种新型半监督CBM框架,采用双层超图学习:概念级超图增强推理能力,图像级超图生成领域自适应伪标签。在新标注的PAS超声数据集及公开乳腺超声数据集上的实验验证了该方法的有效性。其通用性在皮肤病变图像数据集SkinCon上进一步得到证实。代码已开源。
原文摘要 · Abstract (English)
Deep learning has revolutionized medical image analysis, delivering exceptional diagnostic accuracy across diverse applications. Yet, the lack of interpretability in its decision-making hinders clinical adoption, particularly in high-stakes medical contexts where transparency is paramount for trustworthiness. For example, in Placenta Accreta Spectrum (PAS), subtle cues in ultrasound imaging challenge reliable diagnosis, rendering black-box models untrustworthy for accurate scoring. To address this, Concept Bottleneck Models (CBMs) offer a promising avenue by embedding clinically meaningful intermediate concepts into the diagnosis pipeline, enabling clinicians to scrutinize and refine model outputs. However, conventional CBMs falter in capturing complex inter-concept dependencies and demand costly, expert-driven concept annotations, limiting their scalability. This study introduces a novel semi-supervised CBM framework designed for medical imaging, which leverages dual-level hypergraph learning to model high-order concept dependencies and generate domain-adaptive pseudo-labels. Our approach achieves superior interpretability and performance by integrating a concept-level hypergraph for enhanced reasoning and an image-level hypergraph for robust pseudo-label generation. Experiments on a newly annotated PAS ultrasound dataset and a breast ultrasound public dataset demonstrate the effectiveness of the proposed concept label-efficient interpretable framework. Its universality is further validated on the dermoscopic image dataset SkinCon. The code is available at https://github.com/scott-yjyang/HyperCBM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。