对比九种CAM方法在脑出血分类中的可解释性,发现无分割监督下仍能精准定位病灶。
Benchmarking Class Activation Map Methods for Explainable Brain Hemorrhage Classification on Hemorica Dataset
- 用九种CAM算法从分类模型提取像素级定位信息,分阶段评估表现。
- AblationCAM在像素级分割上达到Dice=0.57、IoU=0.40,HiResCAM最准框定病灶位置。
- 首个定量比较脑出血场景下CAM方法的基准研究,适合医疗AI可解释性方向参考。
可解释人工智能(XAI)已成为医学影像研究的关键,旨在提升深度学习模型的透明度与临床信任。本研究聚焦脑出血诊断,通过类激活映射(CAM)技术探索模型可解释性。构建了一套流程,利用九种先进CAM算法从多个网络阶段提取像素级分割与检测标注,并在包含切片标签和高质量分割掩码的Hemorica数据集上进行定量评估。采用Dice、IoU及像素重叠率等指标对比不同CAM方法。结果表明,EfficientNetV2S第5阶段性能最优,其中HiResCAM在边界框对齐上表现最佳,AblationCAM在像素级任务中取得最高Dice值0.57和IoU值0.40,充分说明模型虽仅以分类训练,未受分割监督,仍具备较强定位能力。据当前所知,这是首个针对脑出血检测量化比较CAM方法的工作,建立了可复现的基准,凸显了基于XAI的流程在临床辅助诊断中的潜力。
原文摘要 · Abstract (English)
Explainable Artificial Intelligence (XAI) has become an essential component of medical imaging research, aiming to increase transparency and clinical trust in deep learning models. This study investigates brain hemorrhage diagnosis with a focus on explainability through Class Activation Mapping (CAM) techniques. A pipeline was developed to extract pixellevel segmentation and detection annotations from classification models using nine state-of-the-art CAM algorithms, applied across multiple network stages, and quantitatively evaluated on the Hemorica dataset, which uniquely provides both slice-level labels and high-quality segmentation masks. Metrics including Dice, IoU, and pixel-wise overlap were employed to benchmark CAM variants. Results show that the strongest localization performance occurred at stage 5 of EfficientNetV2S, with HiResCAM yielding the highest bounding-box alignment and AblationCAM achieving the best pixel-level Dice (0.57) and IoU (0.40), representing strong accuracy given that models were trained solely for classification without segmentation supervision. To the best of current knowledge, this is among the f irst works to quantitatively compare CAM methods for brain hemorrhage detection, establishing a reproducible benchmark and underscoring the potential of XAI-driven pipelines for clinically meaningful AI-assisted diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。