arXiv:2504.00908cs.CV2025-04

用插值生成伪标签,提升颈动脉3D分割精度

DBF-UNet: A Two-Stage Framework for Carotid Artery Segmentation with Pseudo-Label Generation

  • 通过插值中心线生成未标注切片的伪标签
  • 在公开数据集上达到优于现有方法的分割效果
  • 适合医学图像少样本分割研究者参考

医学图像分析受限于标注数据稀缺,尤其在三维颈动脉分割任务中,现有数据集存在空间不连续的切片标注,仅少量切片由专家完整标注。为此,我们提出两阶段分割框架:第一阶段通过插值已标注切片中心点构建连续血管中心线,并沿中心线传播标签生成未标注切片的插值标注;使用专家标注切片微调SAM-Med2D,而插值标签作为推理时的提示。第二阶段提出新型轻量级网络DBF-UNet,其编码器采用双向特征融合,结合多尺度特征聚合与密集连接以实现有效特征复用,精准分割完整的3D血管结构。在公开数据集上的实验表明,该方法有效缓解了标注稀疏问题,性能优于现有方法。源代码见https://github.com/Haoxuanli-Thu/DBF-UNet。

原文摘要 · Abstract (English)

Medical image analysis faces significant challenges due to limited annotation data, particularly in three-dimensional carotid artery segmentation tasks, where existing datasets exhibit spatially discontinuous slice annotations with only a small portion of expert-labeled slices in complete 3D volumetric data. To address this challenge, we propose a two-stage segmentation framework. First, we construct continuous vessel centerlines by interpolating between annotated slice centroids and propagate labels along these centerlines to generate interpolated annotations for unlabeled slices. The slices with expert annotations are used for fine-tuning SAM-Med2D, while the interpolated labels on unlabeled slices serve as prompts to guide segmentation during inference. In the second stage, we propose a novel Dense Bidirectional Feature Fusion UNet (DBF-UNet). This lightweight architecture achieves precise segmentation of complete 3D vascular structures. The network incorporates bidirectional feature fusion in the encoder and integrates multi-scale feature aggregation with dense connectivity for effective feature reuse. Experimental validation on public datasets demonstrates that our proposed method effectively addresses the sparse annotation challenge in carotid artery segmentation while achieving superior performance compared to existing approaches. The source code is available at https://github.com/Haoxuanli-Thu/DBF-UNet.

医学图像分割伪标签轻量模型

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