通过空间约束提升乳腺超声诊断模型的可解释性与可信度
Spatially Grounded Concept Bottleneck Models for Trustworthy Breast Ultrasound Diagnosis

- 用病变区域划分兴趣区,强制概念激活与解剖位置一致
- 在五折交叉验证中,诊断与概念识别的准确率均提升,空间对齐度显著增强
- 适合关注AI医疗可解释性与临床部署可信度的研究者
概念瓶颈模型通过人类可理解的概念进行诊断决策,但其可信度常受限于标注质量与粒度。在医学影像中,概念激活可能受无关区域干扰,导致空间解释不准确。本文提出一种数据驱动的空间约束概念瓶颈模型(SG-CBM),利用粗略病灶分割作为弱监督信号,引导解剖上合理的概念证据生成。针对乳腺超声,从每个病灶掩码中提取两个临床相关区域:(i) 病灶内部用于形态学概念,(ii) 后方声影带用于后方现象。通过分组空间对齐目标训练概念图,并用线性瓶颈分类器保持语义一致性。在五折分层组交叉验证中,该模型在诊断和概念层面的宏平均AUC均提升,且概念证据的空间对齐度显著提高。进一步通过‘训练污染/测试清洁’的标注质量压力测试,量化了监督质量对诊断与空间忠实性的影响。结果强调了在可部署医疗AI系统中,需重视数据质量感知的监督设计与系统性可信度验证。
原文摘要 · Abstract (English)
Concept Bottleneck Models provide interpretable-by-design predictions by mediating diagnosis through human-understandable concepts, but in medical imaging, their trustworthiness is often limited by the quality and granularity of available supervision. In particular, predicted concept activations can be driven by irrelevant regions, leading to spatially unfaithful explanations. We study a data-centric spatially grounded Concept Bottleneck Model (SG-CBM) that leverages coarse lesion delineations as weak supervision to encourage anatomically plausible concept evidence. For breast ultrasound, we derive two clinically motivated zones from each lesion mask: (i) an in-lesion region of interest for morphology-related concepts and (ii) a posterior acoustic band for posterior phenomena. We train concept maps using a grouped spatial grounding objective and preserve semantic faithfulness with a linear bottleneck classifier. Across five-fold stratified group cross-validation, the proposed SG-CBM improves diagnostic AUROC and concept macro-AUROC while markedly increasing spatial alignment of concept evidence. We also perform a Train-corrupt/Test-clean annotation-quality stress test to quantify the impact of supervision quality on diagnosis and spatial faithfulness. Overall, the results underscore the need for data-quality-aware supervision design and systematic trustworthiness validation for deployable healthcare AI systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。