用伪标签提升弱监督下的隐藏物体分割效果
Segment Concealed Objects with Incomplete Supervision
- 基于SAM生成伪标签,统一处理弱/半监督数据
- 通过特征分组提升复杂背景下的分割连贯性
- 可直接嵌入现有模型,适合隐藏物体检测场景
不完全监督的隐藏物体分割(ISCOS)旨在利用弱标注或半标注数据,对与环境高度融合的物体进行分割。该任务面临两大挑战:一是训练数据提供的监督信息有限;二是隐蔽场景中物体与背景在外观上高度相似,难以区分。本文提出首个统一方法SEE,解决上述问题。为应对不完全监督,设计基于视觉基础模型SAM的统一均值教师框架,以教师模型生成的粗略掩码作为提示,生成伪标签。同时引入一系列策略优化伪标签生成、存储与监督,确保高质量伪标签被保留并用于指导学生模型训练。为缓解内在相似性带来的困难,设计混合粒度特征分组模块,对不同粒度特征进行聚类并聚合,增强分割一致性,提升单/多物体图像的完整分割性能。在多个ISCOS任务上验证了方法的有效性,实验结果表明其达到当前最优性能,且可作为即插即用模块提升已有模型表现。
原文摘要 · Abstract (English)
Incompletely-Supervised Concealed Object Segmentation (ISCOS) involves segmenting objects that seamlessly blend into their surrounding environments, utilizing incompletely annotated data, such as weak and semi-annotations, for model training. This task remains highly challenging due to (1) the limited supervision provided by the incompletely annotated training data, and (2) the difficulty of distinguishing concealed objects from the background, which arises from the intrinsic similarities in concealed scenarios. In this paper, we introduce the first unified method for ISCOS to address these challenges. To tackle the issue of incomplete supervision, we propose a unified mean-teacher framework, SEE, that leverages the vision foundation model, ``\emph{Segment Anything Model (SAM)}'', to generate pseudo-labels using coarse masks produced by the teacher model as prompts. To mitigate the effect of low-quality segmentation masks, we introduce a series of strategies for pseudo-label generation, storage, and supervision. These strategies aim to produce informative pseudo-labels, store the best pseudo-labels generated, and select the most reliable components to guide the student model, thereby ensuring robust network training. Additionally, to tackle the issue of intrinsic similarity, we design a hybrid-granularity feature grouping module that groups features at different granularities and aggregates these results. By clustering similar features, this module promotes segmentation coherence, facilitating more complete segmentation for both single-object and multiple-object images. We validate the effectiveness of our approach across multiple ISCOS tasks, and experimental results demonstrate that our method achieves state-of-the-art performance. Furthermore, SEE can serve as a plug-and-play solution, enhancing the performance of existing models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。