arXiv:2505.18958cs.CV2025-05被引 1

融合文本提示与混合视觉编码器,提升医学图像分割精度与泛化能力

CDPDNet: Integrating Text Guidance with Hybrid Vision Encoders for Medical Image Segmentation

  • 结合CNN、DINOv2与CLIP,融合局部与全局视觉特征
  • 在多个数据集上分割平均Dice达0.89以上,优于现有方法
  • 适合处理标注不全的医学数据,适用于跨数据集场景

现有公开医学分割数据集多为部分标注,仅对部分解剖结构提供注释。多数据集联合训练时,不完整标注限制了模型对共享解剖表征的学习。此外,纯视觉框架难以捕捉复杂解剖关系和任务特异性差异,导致分割精度下降且泛化能力弱。本文提出一种新型CLIP-DINO提示驱动分割网络(CDPDNet),将自监督视觉变换器与基于CLIP的文本嵌入结合,并引入任务特定文本提示以应对挑战。框架基于卷积神经网络(CNN)构建,融合DINOv2提取细粒度与全局视觉特征,通过多头交叉注意力模块实现特征融合,弥补CNN长程建模不足。同时,将CLIP生成的文本嵌入投影至视觉空间,帮助模型学习器官与肿瘤间的复杂关系。为应对部分标注问题并增强任务区分能力,设计文本任务提示生成(TTPG)模块,动态生成任务相关提示。在多个医学影像数据集上的实验表明,CDPDNet持续优于现有最先进方法。代码与预训练模型已开源:https://github.com/wujiong-hub/CDPDNet.git。

原文摘要 · Abstract (English)

Most publicly available medical segmentation datasets are only partially labeled, with annotations provided for a subset of anatomical structures. When multiple datasets are combined for training, this incomplete annotation poses challenges, as it limits the model's ability to learn shared anatomical representations among datasets. Furthermore, vision-only frameworks often fail to capture complex anatomical relationships and task-specific distinctions, leading to reduced segmentation accuracy and poor generalizability to unseen datasets. In this study, we proposed a novel CLIP-DINO Prompt-Driven Segmentation Network (CDPDNet), which combined a self-supervised vision transformer with CLIP-based text embedding and introduced task-specific text prompts to tackle these challenges. Specifically, the framework was constructed upon a convolutional neural network (CNN) and incorporated DINOv2 to extract both fine-grained and global visual features, which were then fused using a multi-head cross-attention module to overcome the limited long-range modeling capability of CNNs. In addition, CLIP-derived text embeddings were projected into the visual space to help model complex relationships among organs and tumors. To further address the partial label challenge and enhance inter-task discriminative capability, a Text-based Task Prompt Generation (TTPG) module that generated task-specific prompts was designed to guide the segmentation. Extensive experiments on multiple medical imaging datasets demonstrated that CDPDNet consistently outperformed existing state-of-the-art segmentation methods. Code and pretrained model are available at: https://github.com/wujiong-hub/CDPDNet.git.

医学图像分割文本提示多模态DINOv2

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