arXiv:2501.19227cs.CVcs.AI2025-01被引 4

结合主动学习与半监督学习,降低标注成本并提升分割精度

Integrating Semi-Supervised and Active Learning for Semantic Segmentation

  • 主动学习筛选难样本,用特征相似性自动修正伪标签
  • 在两个数据集上均超越现有方法,显著减少人工标注需求
  • 适合标注成本高、数据量大的图像分割场景

本文提出一种融合改进的半监督学习框架的新型主动学习方法,以降低人工标注成本并提升模型性能。该方法有效利用主动学习选出的标注数据及未选中的未标注数据。主动学习模块识别伪标签可能出错的区域,进而引入自动高效的伪标签自修正(PLAR)模块,通过比较像素特征与已标注区域的特征来修正错误伪标签。该方法不增加标注预算,基于聚类假设——同一类别的像素在特征空间中应具有相似表示。同时,仅对未标注数据中信息不足、难以判断的困难区域进行人工标注。我们在自然图像和遥感影像两个基准数据集上评估了该混合半监督主动学习框架,在语义分割任务中均优于当前最优方法。

原文摘要 · Abstract (English)

In this paper, we propose a novel active learning approach integrated with an improved semi-supervised learning framework to reduce the cost of manual annotation and enhance model performance. Our proposed approach effectively leverages both the labelled data selected through active learning and the unlabelled data excluded from the selection process. The proposed active learning approach pinpoints areas where the pseudo-labels are likely to be inaccurate. Then, an automatic and efficient pseudo-label auto-refinement (PLAR) module is proposed to correct pixels with potentially erroneous pseudo-labels by comparing their feature representations with those of labelled regions. This approach operates without increasing the labelling budget and is based on the cluster assumption, which states that pixels belonging to the same class should exhibit similar representations in feature space. Furthermore, manual labelling is only applied to the most difficult and uncertain areas in unlabelled data, where insufficient information prevents the PLAR module from making a decision. We evaluated the proposed hybrid semi-supervised active learning framework on two benchmark datasets, one from natural and the other from remote sensing imagery domains. In both cases, it outperformed state-of-the-art methods in the semantic segmentation task.

主动学习半监督语义分割

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