arXiv:2504.05184eess.IVcs.CV2025-04被引 2

提出新损失函数与网络架构,提升冠脉造影图像血管分割精度。

MSA-UNet3+: Multi-Scale Attention UNet3+ with New Supervised Prototypical Contrastive Loss for Coronary DSA Image Segmentation

  • 设计监督原型对比损失,增强特征区分能力。
  • 在私有数据集上达最高Dice和F1值,平均对齐差降低32.7%。
  • 适合医学影像分割研究者与临床辅助诊断系统开发人员。

准确分割冠状动脉数字减影血管造影(DSA)图像对诊断和治疗冠心病至关重要。尽管深度学习取得进展,但类内差异大和类别不平衡仍限制血管精确分割。现有方法难以有效应对这些问题。此外,现有分割网络编码器不直接生成语义嵌入,影响解码器重建分割掩膜的效果。本文提出一种监督原型对比损失(SPCL),结合监督对比学习与原型对比学习,提升冠脉DSA图像分割性能。监督对比损失强化编码器中的语义嵌入,改善特征区分性;原型对比损失使模型聚焦前景类,通过仅关注难分类的背景样本缓解类内差异和类别不平衡。将SPCL应用于MSA-UNet3+——一种多尺度注意力增强的UNet3+架构。该架构包含多尺度注意力编码器(M-encoder)、多尺度空洞瓶颈(MSD-Bottleneck)用于多尺度特征提取,以及上下文注意力融合模块(CAFM),在保留细粒度细节的同时增强上下文理解。在私有冠脉DSA数据集上的实验表明,MSA-UNet3+优于现有最优方法,达到最高的Dice系数和F1分数,同时显著降低平均对齐差(ASD)和平均曲率差(ACD)。该框架为精准识别冠状动脉狭窄提供支持,助力诊断与治疗决策。代码将开源于https://github.com/rayanmerghani/MSA-UNet3plus。

原文摘要 · Abstract (English)

Accurate segmentation of coronary Digital Subtraction Angiography (DSA) images is essential for diagnosing and treating coronary artery disease (CAD). Despite advances in deep learning, challenges such as high intra-class variance and class imbalance limit precise vessel delineation. Existing approaches for coronary DSA segmentation cannot effectively address these issues. Furthermore, existing segmentation network encoders do not directly generate semantic embeddings, which could enable the decoder to reconstruct segmentation masks more effectively. We propose a Supervised Prototypical Contrastive Loss (SPCL) that combines supervised and prototypical contrastive learning to enhance coronary DSA image segmentation. The supervised contrastive loss enforces semantic embeddings in the encoder, improving feature differentiation. The prototypical contrastive loss enables the model to focus on the foreground class while alleviating high intra-class variance and class imbalance by concentrating only on hard-to-classify background samples. We implement the proposed SPCL within MSA-UNet3+, a Multi-Scale Attention-Enhanced UNet3+ architecture. The architecture integrates a Multi-Scale Attention Encoder (M-encoder), a Multi-Scale Dilated Bottleneck (MSD-Bottleneck) for multi-scale feature extraction, and a Contextual Attention Fusion Module (CAFM) to preserve fine-grained details while improving contextual understanding. Experiments on a private coronary DSA dataset demonstrate that MSA-UNet3+ outperforms state-of-the-art methods, achieving the highest Dice coefficient and F1-score while significantly reducing ASD and ACD. The framework provides precise vessel segmentation for accurate identification of coronary stenosis and supports informed diagnostic and therapeutic decisions. The code will be released at https://github.com/rayanmerghani/MSA-UNet3plus.

医学图像分割对比学习冠脉

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