提出可解释的通道调制方法,提升心冠血管分割在不同数据集间的泛化能力。
AngioDG: Interpretable Channel-informed Feature-modulated Single-source Domain Generalization for Coronary Vessel Segmentation in X-ray Angiography
- 通过通道重要性分析,动态调整特征权重以增强跨域不变性
- 在6个数据集上实现最优的域外性能,且保持域内稳定表现
- 适合医疗影像领域需要高泛化性的模型开发人员
心血管疾病是全球首要死因,而X射线冠状动脉造影(XCA)是实时心脏介入中的金标准。从XCA中分割冠状血管有助于量化评估狭窄程度,提升临床决策。然而,由于成像协议和患者群体差异导致的数据域偏移,以及标注数据集稀缺,使建立具备泛化能力的分割模型极具挑战。单源域泛化(SDG)成为必要解决方案。现有方法多依赖数据增强,可能加剧对合成域的过拟合。本文提出新型方法AngioDG,通过通道正则化策略提升泛化能力:首先识别早期特征通道对任务指标的贡献,实现可解释性;随后重新加权通道,强化域不变特征,抑制域特定特征。在6个XCA数据集上评估,所提方法在域外性能上优于对比方法,同时保持稳定的域内测试表现。
原文摘要 · Abstract (English)
Cardiovascular diseases are the leading cause of death globally, with X-ray Coronary Angiography (XCA) as the gold standard during real-time cardiac interventions. Segmentation of coronary vessels from XCA can facilitate downstream quantitative assessments, such as measurement of the stenosis severity and enhancing clinical decision-making. However, developing generalizable vessel segmentation models for XCA is challenging due to variations in imaging protocols and patient demographics that cause domain shifts. These limitations are exacerbated by the lack of annotated datasets, making Single-source Domain Generalization (SDG) a necessary solution for achieving generalization. Existing SDG methods are largely augmentation-based, which may not guarantee the mitigation of overfitting to augmented or synthetic domains. We propose a novel approach, ``AngioDG", to bridge this gap by channel regularization strategy to promote generalization. Our method identifies the contributions of early feature channels to task-specific metrics for DG, facilitating interpretability, and then reweights channels to calibrate and amplify domain-invariant features while attenuating domain-specific ones. We evaluate AngioDG on 6 x-ray angiography datasets for coronary vessels segmentation, achieving the best out-of-distribution performance among the compared methods, while maintaining consistent in-domain test performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。