通过特征密度感知增强医学图像分割的聚类紧凑性
Neighbor Does Matter: Density-Aware Contrastive Learning for Medical Semi-supervised Segmentation
- 利用特征空间密度定位稀疏区域,构建密度感知邻接图
- 在多器官分割数据集上达到最优性能,显著提升小样本下分割精度
- 适合标注稀缺的医学图像分割任务,尤其对低对比组织有效
在医学图像分析中,多器官半监督分割面临标注不足和软组织对比度低的挑战。现有方法多依赖单个样本的伪标签与一致性正则化,忽略特征空间中的邻域信息。本文提出特征密度感知对比学习(DACL),通过密度估计定位特征簇中的稀疏区域,并将稀疏区锚定特征向高密度正样本近似簇心拉近,增强类内紧凑性。方法结合标签引导的协同训练与密度引导的几何正则化,为无标签数据提供互补监督。在多器官分割挑战数据集上的实验表明,该方法优于现有最先进方法,验证了其在医学图像分割任务中的有效性。
原文摘要 · Abstract (English)
In medical image analysis, multi-organ semi-supervised segmentation faces challenges such as insufficient labels and low contrast in soft tissues. To address these issues, existing studies typically employ semi-supervised segmentation techniques using pseudo-labeling and consistency regularization. However, these methods mainly rely on individual data samples for training, ignoring the rich neighborhood information present in the feature space. In this work, we argue that supervisory information can be directly extracted from the geometry of the feature space. Inspired by the density-based clustering hypothesis, we propose using feature density to locate sparse regions within feature clusters. Our goal is to increase intra-class compactness by addressing sparsity issues. To achieve this, we propose a Density-Aware Contrastive Learning (DACL) strategy, pushing anchored features in sparse regions towards cluster centers approximated by high-density positive samples, resulting in more compact clusters. Specifically, our method constructs density-aware neighbor graphs using labeled and unlabeled data samples to estimate feature density and locate sparse regions. We also combine label-guided co-training with density-guided geometric regularization to form complementary supervision for unlabeled data. Experiments on the Multi-Organ Segmentation Challenge dataset demonstrate that our proposed method outperforms state-of-the-art methods, highlighting its efficacy in medical image segmentation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。