arXiv:2512.13977cs.CV2025-12

用可解释AI诊断脑血管分割模型在跨数据集时的失败原因。

XAI-Driven Diagnosis of Generalization Failure in State-Space Cerebrovascular Segmentation Models: A Case Study on Domain Shift Between RSNA and TopCoW Datasets

  • 通过对比RSNA与TopCoW数据集差异,定位域偏移根源。
  • 发现模型注意力从真实血管结构转向错误预测,导致骰子分数从0.86降至0.29。
  • 揭示模型学习了虚假关联,适合关注模型可信度的研究者阅读。

深度学习模型在医学影像中的临床部署受域偏移严重制约。本文以状态空间模型UMamaba在脑血管分割任务中的表现为例,提出一种两阶段诊断方法。首先量化源数据集(RSNA CTA Aneurysm)与目标数据集(TopCoW Circle of Willis CT)之间的域差距,发现两者在Z分辨率和背景噪声上存在显著差异。模型在目标数据集上的Dice分数从0.8604骤降至0.2902。第二阶段采用Seg-XRes-CAM分析注意力机制失效原因,通过计算注意力图与真实标注及自身预测掩码的重叠度,发现模型在目标域中放弃真实解剖特征,转而聚焦于自身错误预测。定量结果显示,注意力与真实血管的交并比(IoU)仅约0.101(阈值0.3),但与错误预测的交并比高达0.282,证明模型学习了虚假相关性。该研究验证了可解释AI在识别新兴架构中数据偏差方面的关键诊断价值。

原文摘要 · Abstract (English)

The clinical deployment of deep learning models in medical imaging is severely hindered by domain shift. This challenge, where a high-performing model fails catastrophically on external datasets, is a critical barrier to trustworthy AI. Addressing this requires moving beyond simple performance metrics toward deeper understanding, making Explainable AI (XAI) an essential diagnostic tool in medical image analysis. We present a rigorous, two-phase approach to diagnose the generalization failure of state-of-the-art State-Space Models (SSMs), specifically UMamaba, applied to cerebrovascular segmentation. We first established a quantifiable domain gap between our Source (RSNA CTA Aneurysm) and Target (TopCoW Circle of Willis CT) datasets, noting significant differences in Z-resolution and background noise. The model's Dice score subsequently plummeted from 0.8604 (Source) to 0.2902 (Target). In the second phase, which is our core contribution, we utilized Seg-XRes-CAM to diagnose the cause of this failure. We quantified the model's focus by measuring the overlap between its attention maps and the Ground Truth segmentations, and between its attention maps and its own Prediction Mask. Our analysis proves the model failed to generalize because its attention mechanism abandoned true anatomical features in the Target domain. Quantitative metrics confirm the model's focus shifted away from the Ground Truth vessels (IoU~0.101 at 0.3 threshold) while still aligning with its own wrong predictions (IoU~0.282 at 0.3 threshold). This demonstrates the model learned spurious correlations, confirming XAI is a powerful diagnostic tool for identifying dataset bias in emerging architectures.

可解释AI医学影像域泛化注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。