arXiv:2507.02399cs.CVcs.LG2025-07

用稀疏标注提升医学图像分割精度,自动生成精准边界伪标签

TABNet: A Triplet Augmentation Self-Recovery Framework with Boundary-Aware Pseudo-Labels for Medical Image Segmentation

  • 通过三重增强与自恢复机制,强化稀疏标注下的特征学习
  • 在ACDC和MSCMR seg数据集上优于现有弱监督方法,接近全监督效果
  • 适合缺乏大量标注的医疗影像研究者使用

医学图像分割是临床应用的核心任务,但大规模全标注数据集获取成本高昂。涂鸦标注作为稀疏标注方式,虽高效低成本,却因标注稀疏限制目标区域特征学习,且缺乏边界监督,给分割网络训练带来挑战。本文提出TAB Net,一种新型弱监督医学图像分割框架,包含两个核心模块:三重增强自恢复(TAS)模块与边界感知伪标签(BAP)模块。TAS模块通过强度变换、裁剪和拼图三种互补增强策略,分别提升模型对纹理对比度变化的敏感性、局部解剖结构捕捉能力及全局解剖布局建模能力;通过引导网络从多样化增强输入中恢复完整掩码,促进在稀疏监督下的深层语义理解。BAP模块通过融合双分支预测生成加权伪标签,并引入边界感知损失,实现更精确的伪监督与细粒度轮廓优化。在公开数据集ACDC和MSCMR seg上的实验表明,TAB Net显著优于当前最先进的基于涂鸦的弱监督分割方法,性能接近全监督水平。

原文摘要 · Abstract (English)

Background and objective: Medical image segmentation is a core task in various clinical applications. However, acquiring large-scale, fully annotated medical image datasets is both time-consuming and costly. Scribble annotations, as a form of sparse labeling, provide an efficient and cost-effective alternative for medical image segmentation. However, the sparsity of scribble annotations limits the feature learning of the target region and lacks sufficient boundary supervision, which poses significant challenges for training segmentation networks. Methods: We propose TAB Net, a novel weakly-supervised medical image segmentation framework, consisting of two key components: the triplet augmentation self-recovery (TAS) module and the boundary-aware pseudo-label supervision (BAP) module. The TAS module enhances feature learning through three complementary augmentation strategies: intensity transformation improves the model's sensitivity to texture and contrast variations, cutout forces the network to capture local anatomical structures by masking key regions, and jigsaw augmentation strengthens the modeling of global anatomical layout by disrupting spatial continuity. By guiding the network to recover complete masks from diverse augmented inputs, TAS promotes a deeper semantic understanding of medical images under sparse supervision. The BAP module enhances pseudo-supervision accuracy and boundary modeling by fusing dual-branch predictions into a loss-weighted pseudo-label and introducing a boundary-aware loss for fine-grained contour refinement. Results: Experimental evaluations on two public datasets, ACDC and MSCMR seg, demonstrate that TAB Net significantly outperforms state-of-the-art methods for scribble-based weakly supervised segmentation. Moreover, it achieves performance comparable to that of fully supervised methods.

医学图像弱监督分割伪标签

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