摘要
arXiv:2502.15152v2 通告类型: replace-cross
摘要: 半监督语义分割(SSSS)旨在通过利用大量的未标记数据来改进分割性能,同时只有限的数量的标记样本。现有方法常常遭受耦合问题,即过度依赖初始标记数据导致学习效果不佳;确认偏见问题,即错误预测会反复强化自身;以及由有限的边界意识和模糊的边缘提示引起的边界模糊。为了解决这些问题,我们提出了CW-BASS,一种新颖的SSSS框架。为了减轻错误预测的影响,我们为伪标签分配了置信度权重。此外,我们利用边界划定技术,尽管这些技术在弱监督语义分割(WSSS)中已被广泛探索,但在SSSS中仍然未被充分利用。具体而言,我们的方法:(1)通过置信度加权损失减少耦合,该损失根据伪标签的预测置信度分数来调整伪标签的影响;(2)通过动态阈值机制减轻确认偏见,该机制根据模型性能学习过滤伪标签;(3)通过边界感知模块解决边界模糊,该模块在对象边缘附近细化分割;(4)通过置信度衰减策略减少标签噪音,在训练过程中逐步细化伪标签。在Pascal VOC 2012和Cityscapes上的广泛实验表明,CW-BASS达到了最佳性能。值得注意的是,CW-BASS在Cityscapes下的一个具有挑战性和未充分探索的数据集拆分(1/30,3.3%),即100张图像中,实现了65.9%的mIoU,这突显了其在有限标签设置中的有效性。我们的代码可在https://github.com/psychofict/CW-BASS获取。