Semi-supervised Semantic Segmentation via Boosting Uncertainty on Unlabeled Data

  • 2023-11-30 18:01:03
  • Daoan Zhang, Yunhao Luo, Jianguo Zhang
  • 0

Abstract

We bring a new perspective to semi-supervised semantic segmentation byproviding an analysis on the labeled and unlabeled distributions in trainingdatasets. We first figure out that the distribution gap between labeled andunlabeled datasets cannot be ignored, even though the two datasets are sampledfrom the same distribution. To address this issue, we theoretically analyze andexperimentally prove that appropriately boosting uncertainty on unlabeled datacan help minimize the distribution gap, which benefits the generalization ofthe model. We propose two strategies and design an uncertainty boosteralgorithm, specially for semi-supervised semantic segmentation. Extensiveexperiments are carried out based on these theories, and the results confirmthe efficacy of the algorithm and strategies. Our plug-and-play uncertaintybooster is tiny, efficient, and robust to hyperparameters but can significantlypromote performance. Our approach achieves state-of-the-art performance in ourexperiments compared to the current semi-supervised semantic segmentationmethods on the popular benchmarks: Cityscapes and PASCAL VOC 2012 withdifferent train settings.

 

Quick Read (beta)

loading the full paper ...