GRAPE: Graph-Augmented Prototype Explanations for Interactive Medical Image Diagnosis
Not on the map. Why?
Abstract (arXiv)
Prototype-based medical image classifiers present three clinical limitations: they treat findings as independent, silently amplify unsafe physician feedback, and require full retraining whenever a new finding is needed. We present GRAPE (Graph-Augmented Prototype Explanations), a unified architecture that addresses all three challenges. First, a Graph Attention Task Head models anatomical concept co-occurrence, boosting macro-F1 by +13.8,pp over the prototype baseline on TBX11K. Second, a Concept-Mismatch Safety Check - the first such mechanism in prototype-based medical classifiers - warns when the model's dominant finding inside a doctor-drawn region conflicts with the claimed label, catching 85% of erroneous annotations versus 51% for MC-Dropout with no extra inference cost. Third, Open-Vocabulary Prototype Anchoring aligns visual prototypes to clinical text, allowing a new finding to be added from a single labeled image without modifying any other component. On NIH ChestX-ray14, one Effusion example recovers full-supervision localization accuracy; on TBX11K, prototype maps achieve 2.6x better lesion localization than end-to-end baselines. All three capabilities add only +1~ms latency at interactive batch size. The project page is https://github.com/KurbanIntelligenceLab/GRAPE.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 31 references that are in the map. First posted on arXiv 29 Jun 2026.
Builds on
26 of 31 references.
References · 26
- 2017Graph Attention Networkscited by 8,923
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2020Concept Bottleneck Modelscited by 1,032
- 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 972
- 2018RISE: Randomized Input Sampling for Explanation of Black-box Modelscited by 838
- 2018This Looks Like That: Deep Learning for Interpretable Image Recognitioncited by 737
- 2018Evidential Deep Learning to Quantify Classification Uncertaintycited by 732
- 2022MedCLIP: Contrastive Learning from Unpaired Medical Images and Textcited by 570
- 2023BiomedCLIP: a multimodal biomedical foundation model pretrained from fifteen million scientific image-text pairscited by 520
- 2017CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learningcited by 461
- 2019Multi-Label Image Recognition with Graph Convolutional Networkscited by 322
14 more
- 2016Top-down Neural Attention by Excitation Backpropcited by 321
- 2022Post-hoc Concept Bottleneck Modelscited by 293
- 2020Neural Prototype Trees for Interpretable Fine-grained Image Recognitioncited by 214
- 2023Learning to Exploit Temporal Structure for Biomedical Vision-Language Processingcited by 211
- 2018Acquisition of Localization Confidence for Accurate Object Detectioncited by 176
- 2021Deformable ProtoPNet: An Interpretable Image Classifier Using Deformable Prototypescited by 120
- 2021Do Concept Bottleneck Models Learn as Intended?cited by 116
- 2017Interactive Medical Image Segmentation using Deep Learning with Image-specific Fine-tuningcited by 111
- 2020Attention-Driven Dynamic Graph Convolutional Network for Multi-Label Image Recognitioncited by 80
- 2023This Looks Like Those: Illuminating Prototypical Concepts Using Multiple Visualizationscited by 36
- 2023B-cos Alignment for Inherently Interpretable CNNs and Vision Transformerscited by 31
- 2023MProtoNet: A Case-Based Interpretable Model for Brain Tumor Classification with 3D Multi-parametric Magnetic Resonance Imagingcited by 11
- 2025Interactive Medical Image Analysis with Concept-based Similarity Reasoningcited by 11
- 2023Revisiting Computer-Aided Tuberculosis Diagnosiscited by 6
Built on by
Not cited yet.