arXiv:2602.11536cs.CV2026-02

用解剖结构知识提升血管造影自监督预训练效果

Vascular anatomy-aware self-supervised pre-training for X-ray angiogram analysis

  • 基于血管解剖结构设计掩码策略和一致性损失
  • 在6个数据集上实现领先性能,迁移能力显著
  • 适合医学影像领域研究者与临床辅助系统开发者

X射线血管造影是心血管疾病诊断的金标准,但深度学习方法受限于标注数据稀缺。现有自监督学习(SSL)在该领域应用有限,主要因缺乏有效框架与大规模数据集。为此,我们提出血管解剖感知的掩码图像建模框架VasoMIM,包含两个关键设计:解剖引导的掩码策略与解剖一致性损失。前者通过战略性遮蔽含血管区域,促使模型学习鲁棒的血管语义;后者保持原图与重建图间血管结构一致性,增强表征判别力。同时构建目前最大的血管造影预训练数据集XA-170K。在四个下游任务、六个数据集上的验证表明,VasoMIM具有优异的可迁移性,性能优于现有方法。这些结果凸显VasoMIM作为基础模型在推动各类血管造影分析任务中的巨大潜力。代码与数据已开源。

原文摘要 · Abstract (English)

X-ray angiography is the gold standard imaging modality for cardiovascular diseases. However, current deep learning approaches for X-ray angiogram analysis are severely constrained by the scarcity of annotated data. While large-scale self-supervised learning (SSL) has emerged as a promising solution, its potential in this domain remains largely unexplored, primarily due to the lack of effective SSL frameworks and large-scale datasets. To bridge this gap, we introduce a vascular anatomy-aware masked image modeling (VasoMIM) framework that explicitly integrates domain-specific anatomical knowledge. Specifically, VasoMIM comprises two key designs: an anatomy-guided masking strategy and an anatomical consistency loss. The former strategically masks vessel-containing patches to compel the model to learn robust vascular semantics, while the latter preserves structural consistency of vessels between original and reconstructed images, enhancing the discriminability of the learned representations. In conjunction with VasoMIM, we curate XA-170K, the largest X-ray angiogram pre-training dataset to date. We validate VasoMIM on four downstream tasks across six datasets, where it demonstrates superior transferability and achieves state-of-the-art performance compared to existing methods. These findings highlight the significant potential of VasoMIM as a foundation model for advancing a wide range of X-ray angiogram analysis tasks. VasoMIM and XA-170K will be available at https://github.com/Dxhuang-CASIA/XA-SSL.

自监督学习医学影像血管分析预训练

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