Semi-supervised image semantic segmentation method with semantic regions patching and uncertainty-guided loss

Dinghao Guo,Dali Chen,Xin Lin,Zheng Xue,Wei Zheng,Xianling Li
DOI: https://doi.org/10.1007/s00371-024-03623-9
IF: 2.835
2024-10-08
The Visual Computer
Abstract:Semi-supervised semantic segmentation is a challenging task in the computer vision field, facing two major difficulties: (1) the lack of high-quality training data and (2) the confirmation bias caused by incorrect pseudo-labels during training. To address these issues, we propose a semi-supervised semantic segmentation method called SRPSeg. First, a novel mixed sample data augmentation approach, SRPmix, is proposed. It generates high-quality training images containing multiple semantic targets by extracting semantic regions from multiple unlabeled images, thereby not only enhancing the diversity and complexity of the generated images but also effectively addressing the issue of semantic sparsity. Second, a new loss function called uncertainty-guided loss is introduced. This loss function leverages uncertainty estimation to compute reliability weights for pixel predictions and incorporates these weights into the weighted cross-entropy loss. This effectively assists the model in mitigating the interference from incorrect pseudo-labels. Experiments performed on the PASCAL VOC 2012 dataset across a range of semi-supervised settings have demonstrated that SRPSeg exhibits competitive performance when compared with state-of-the-art methods. Our code is publicly available at https://github.com/tobenan/SRPSeg.
computer science, software engineering
What problem does this paper attempt to address?