LLM2D
带有边界意识的自信心度加权半监督语义分割学习
Confidence-Weighted Boundary-Aware Learning for Semi-Supervised Semantic Segmentation
作者: Ebenezer Tarubinga, Jenifer Kalafatovich Espinoza
发布日期: 2/24/2025
arXiv ID: oai:arXiv.org:2502.15152v1

摘要

arXiv:2502.15152v1 类型: cross 摘要:半监督语义分割(SSSS)旨在通过利用未标记数据来提高分割性能,同时仅使用有限的标记样本。现有的SSSS方法往往面临耦合问题,即过度依赖初始标记数据会导致学习效果不佳;确信偏差问题,即错误预测会反复得到强化;以及由于边界感知不足和边缘信息模糊导致的边界模糊。为了解决这些问题,我们提出了一种新的SSSS框架CW-BASS。为了减轻错误预测的影响,我们为伪标签分配了置信度权重。此外,我们利用边界划界技术,尽管这些技术在弱监督语义分割(WSSS)中已被广泛研究,但在SSSS中仍被低估。具体而言,我们的方法:(1)通过一个根据伪标签预测的置信度分数调整伪标签影响的置信度加权损失函数来减少耦合;(2)通过动态阈值机制来减少确信偏差,该机制可以学习根据模型性能筛选伪标签;(3)通过一个边界感知模块来解决边界模糊问题,该模块可增强对象边界附近的分割准确性;(4)通过一个置信度衰减策略在训练过程中逐步精炼伪标签来减少标签噪声。在Pascal VOC 2012和Cityscapes上的广泛实验表明,我们的方法实现了最先进的性能。此外,仅使用1/8或12.5%的标记数据,我们的方法在Pascal VOC 2012上的mIoU达到了75.81,突显了其在有限标记设置中的有效性。