Towards Robust Pseudo-Label Learning in Semantic Segmentation: An Encoding Perspective
About
Pseudo-label learning is widely used in semantic segmentation, particularly in label-scarce scenarios such as unsupervised domain adaptation (UDA) and semisupervised learning (SSL). Despite its success, this paradigm can generate erroneous pseudo-labels, which are further amplified during training due to utilization of one-hot encoding. To address this issue, we propose ECOCSeg, a novel perspective for segmentation models that utilizes error-correcting output codes (ECOC) to create a fine-grained encoding for each class. ECOCSeg offers several advantages. First, an ECOC-based classifier is introduced, enabling model to disentangle classes into attributes and handle partial inaccurate bits, improving stability and generalization in pseudo-label learning. Second, a bit-level label denoising mechanism is developed to generate higher-quality pseudo-labels, providing adequate and robust supervision for unlabeled images. ECOCSeg can be easily integrated with existing methods and consistently demonstrates significant improvements on multiple UDA and SSL benchmarks across different segmentation architectures. Code is available at https://github.com/Woof6/ECOCSeg.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | SYNTHIA to Cityscapes (val) | Rider IoU54.5 | 435 | |
| Semantic segmentation | COCO | mIoU51.6 | 96 | |
| Change Detection | WHU-CD | mIoU84.6 | 55 | |
| Semantic segmentation | Pascal VOC 1/16 labeled 2012 (train) | mIoU79.2 | 53 | |
| Semantic segmentation | GTA to Cityscapes (val) | Road Accuracy97.9 | 44 | |
| Semantic segmentation | Pascal 1/8 labeled VOC 2012 | mIoU79.8 | 30 | |
| Semantic segmentation | Pascal 1/4 labels VOC 2012 | mIoU80.3 | 30 | |
| Semantic segmentation | Pascal VOC high-quality (test) | mIoU88.9 | 25 | |
| Multiclass Segmentation | ACDC (test) | mIoU0.905 | 15 |