提出动态伪标签学习方法,提升隐蔽目标检测精度。
UCOD-DPL: Unsupervised Camouflaged Object Detection via Dynamic Pseudo-label Learning
- 用自适应伪标签模块融合多源标签,减少噪声干扰。
- 设计双分支对抗解码器,更好区分背景与隐蔽目标。
- 模拟人类观察习惯,对小目标二次精修,适合遥感等场景。
无监督隐蔽目标检测(UCOD)因无需像素级标注而受到关注。现有方法通常采用固定策略生成伪标签,并使用1x1卷积层作为简单解码器,导致性能远低于全监督方法。主要问题在于:1)伪标签含大量噪声,模型易学错知识;2)简单解码器无法捕捉隐蔽目标的语义特征,尤其对小目标,因伪标签分辨率低且前景背景混淆严重。为此,我们提出基于教师-学生框架的动态伪标签学习方法UCOD-DPL,包含自适应伪标签模块(APM)、双分支对抗解码器(DBA)和看两次机制。APM自适应融合固定策略与教师模型生成的伪标签,防止过拟合同时保留自我修正能力;DBA通过不同分割目标的对抗学习,帮助模型克服前景-背景混淆;看两次机制模仿人类聚焦隐蔽目标的行为,对小目标进行二次精修。大量实验表明,该方法性能优异,甚至超越部分全监督方法。代码已开源。
原文摘要 · Abstract (English)
Unsupervised Camoflaged Object Detection (UCOD) has gained attention since it doesn't need to rely on extensive pixel-level labels. Existing UCOD methods typically generate pseudo-labels using fixed strategies and train 1 x1 convolutional layers as a simple decoder, leading to low performance compared to fully-supervised methods. We emphasize two drawbacks in these approaches: 1). The model is prone to fitting incorrect knowledge due to the pseudo-label containing substantial noise. 2). The simple decoder fails to capture and learn the semantic features of camouflaged objects, especially for small-sized objects, due to the low-resolution pseudo-labels and severe confusion between foreground and background pixels. To this end, we propose a UCOD method with a teacher-student framework via Dynamic Pseudo-label Learning called UCOD-DPL, which contains an Adaptive Pseudo-label Module (APM), a Dual-Branch Adversarial (DBA) decoder, and a Look-Twice mechanism. The APM module adaptively combines pseudo-labels generated by fixed strategies and the teacher model to prevent the model from overfitting incorrect knowledge while preserving the ability for self-correction; the DBA decoder takes adversarial learning of different segmentation objectives, guides the model to overcome the foreground-background confusion of camouflaged objects, and the Look-Twice mechanism mimics the human tendency to zoom in on camouflaged objects and performs secondary refinement on small-sized objects. Extensive experiments show that our method demonstrates outstanding performance, even surpassing some existing fully supervised methods. The code is available now.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。