Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a spatially grounded Concept Bottleneck Model (SG-CBM) for breast ultrasound diagnosis that enhances interpretability by using coarse lesion delineations as weak supervision. By defining anatomically relevant regions of interest, the SG-CBM improves both diagnostic performance and spatial alignment of concept evidence, achieving higher AUROC scores compared to traditional methods. The findings highlight the critical role of data quality in developing trustworthy AI systems for medical imaging, emphasizing the need for robust supervision strategies.
Spatially grounded supervision can significantly boost diagnostic accuracy and interpretability in breast ultrasound AI systems.
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.