arXiv:2505.11075cs.CV2025-05被引 2

解决半监督实例分割中伪标签质量混淆问题,提升小样本下的分割精度。

Pseudo-Label Quality Decoupling and Correction for Semi-Supervised Instance Segmentation

  • 分离类别与掩码质量评估,分别设置阈值过滤伪标签。
  • 动态修正类别伪标签,缓解类别混淆问题。
  • 像素级加权损失,降低噪声掩码的影响,适合小样本场景。

半监督实例分割(SSIS)利用少量标注数据对图像像素进行分类和实例分组,但常因伪标签的类别和掩码噪声导致性能不稳定。现有方法通常用单一阈值同时评估类别与掩码质量,造成两者权衡损失。本文提出伪标签质量解耦与校正框架(PL-DC):在实例层面,设计双阈值解耦过滤机制,独立控制分类与分组质量;在类别层面,引入动态类别校正模块,缓解类别混淆;在像素层面,采用掩码不确定性感知机制,对不同像素重加权掩码损失,降低噪声影响。在COCO和Cityscapes数据集上的实验表明,PL-DC显著提升性能,达到新最优结果。尤其在极小标注数据下表现突出:仅用1% COCO标注数据时,mAP提升+11.6;5% Cityscapes标注数据时,提升+15.5。代码将开源。

原文摘要 · Abstract (English)

Semi-Supervised Instance Segmentation (SSIS) involves classifying and grouping image pixels into distinct object instances using limited labeled data. This learning paradigm usually faces a significant challenge of unstable performance caused by noisy pseudo-labels of instance categories and pixel masks. We find that the prevalent practice of filtering instance pseudo-labels assessing both class and mask quality with a single score threshold, frequently leads to compromises in the trade-off between the qualities of class and mask labels. In this paper, we introduce a novel Pseudo-Label Quality Decoupling and Correction (PL-DC) framework for SSIS to tackle the above challenges. Firstly, at the instance level, a decoupled dual-threshold filtering mechanism is designed to decouple class and mask quality estimations for instance-level pseudo-labels, thereby independently controlling pixel classifying and grouping qualities. Secondly, at the category level, we introduce a dynamic instance category correction module to dynamically correct the pseudo-labels of instance categories, effectively alleviating category confusion. Lastly, we introduce a pixel-level mask uncertainty-aware mechanism at the pixel level to re-weight the mask loss for different pixels, thereby reducing the impact of noise introduced by pixel-level mask pseudo-labels. Extensive experiments on the COCO and Cityscapes datasets demonstrate that the proposed PL-DC achieves significant performance improvements, setting new state-of-the-art results for SSIS. Notably, our PL-DC shows substantial gains even with minimal labeled data, achieving an improvement of +11.6 mAP with just 1% COCO labeled data and +15.5 mAP with 5% Cityscapes labeled data. The code will be public.

半监督实例分割伪标签小样本

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