无需标注数据,跨中心检测肺栓塞准确率提升三倍
Using Unsupervised Domain Adaptation Semantic Segmentation for Pulmonary Embolism Detection in Computed Tomography Pulmonary Angiogram (CTPA) Images
- 用Transformer+均值教师框架实现无监督跨中心分割
- 在两个数据集间交并比从0.11提升至0.42,显著增强泛化能力
- 特别擅长捕捉小病灶,适合医疗影像少样本部署场景
尽管深度学习在肺栓塞(PE)的辅助诊断中展现出巨大潜力,但在计算机断层扫描肺动脉造影(CTPA)中的实际应用常受“领域偏移”和专家标注成本高昂的限制。为此,本文提出一种基于Transformer主干网络与均值教师架构的无监督域适应(UDA)框架,用于跨中心语义分割。核心在于通过学习特征空间中的深层结构信息来提升伪标签可靠性。具体设计三个模块:(1)原型对齐(PA)机制减少类别级分布差异;(2)全局与局部对比学习(GLCL)捕捉像素级拓扑关系与全局语义;(3)基于注意力的辅助局部预测(AALP)模块,自动从Transformer注意力图中提取高信息切片以增强对小病灶的敏感性。在跨中心数据集(FUMPE 和 CAD-PE)上的实验验证表明性能显著提升:在 FUMPE → CAD-PE 任务中,交并比从 0.1152 提升至 0.4153;在 CAD-PE → FUMPE 任务中,从 0.1705 提升至 0.4302。此外,该方法在 MMWHS 数据集的 CT→MRI 跨模态任务中实现了 69.9% 的 Dice 得分,且未使用任何目标域标签进行模型选择,验证了其在多样化临床环境中的鲁棒性与泛化能力。
原文摘要 · Abstract (English)
While deep learning has demonstrated considerable promise in computer-aided diagnosis for pulmonary embolism (PE), practical deployment in Computed Tomography Pulmonary Angiography (CTPA) is often hindered by "domain shift" and the prohibitive cost of expert annotations. To address these challenges, an unsupervised domain adaptation (UDA) framework is proposed, utilizing a Transformer backbone and a Mean-Teacher architecture for cross-center semantic segmentation. The primary focus is placed on enhancing pseudo-label reliability by learning deep structural information within the feature space. Specifically, three modules are integrated and designed for this task: (1) a Prototype Alignment (PA) mechanism to reduce category-level distribution discrepancies; (2) Global and Local Contrastive Learning (GLCL) to capture both pixel-level topological relationships and global semantic representations; and (3) an Attention-based Auxiliary Local Prediction (AALP) module designed to reinforce sensitivity to small PE lesions by automatically extracting high-information slices from Transformer attention maps. Experimental validation conducted on cross-center datasets (FUMPE and CAD-PE) demonstrates significant performance gains. In the FUMPE -> CAD-PE task, the IoU increased from 0.1152 to 0.4153, while the CAD-PE -> FUMPE task saw an improvement from 0.1705 to 0.4302. Furthermore, the proposed method achieved a 69.9% Dice score in the CT -> MRI cross-modality task on the MMWHS dataset without utilizing any target-domain labels for model selection, confirming its robustness and generalizability for diverse clinical environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。