Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Revisiting Weak-to-Strong Consistency in Semi-Supervised Semantic Segmentation

About

In this work, we revisit the weak-to-strong consistency framework, popularized by FixMatch from semi-supervised classification, where the prediction of a weakly perturbed image serves as supervision for its strongly perturbed version. Intriguingly, we observe that such a simple pipeline already achieves competitive results against recent advanced works, when transferred to our segmentation scenario. Its success heavily relies on the manual design of strong data augmentations, however, which may be limited and inadequate to explore a broader perturbation space. Motivated by this, we propose an auxiliary feature perturbation stream as a supplement, leading to an expanded perturbation space. On the other, to sufficiently probe original image-level augmentations, we present a dual-stream perturbation technique, enabling two strong views to be simultaneously guided by a common weak view. Consequently, our overall Unified Dual-Stream Perturbations approach (UniMatch) surpasses all existing methods significantly across all evaluation protocols on the Pascal, Cityscapes, and COCO benchmarks. Its superiority is also demonstrated in remote sensing interpretation and medical image analysis. We hope our reproduced FixMatch and our results can inspire more future works. Code and logs are available at https://github.com/LiheYoung/UniMatch.

Lihe Yang, Lei Qi, Litong Feng, Wayne Zhang, Yinghuan Shi• 2022

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU30.9
2731
Semantic segmentationPASCAL VOC 2012 (val)
Mean IoU80.43
2040
Semantic segmentationCityscapes (val)
mIoU79.5
572
Semantic segmentationGTA5 → Cityscapes (val)
mIoU72.1
533
Change DetectionLEVIR-CD (test)--
357
Semantic segmentationPASCAL Context (val)
mIoU43.7
323
Semantic segmentationCityscapes (val)
mIoU79.5
287
Change DetectionWHU-CD (test)
IoU85.1
286
Semantic segmentationCOCO 2014 (val)
mIoU49.8
251
Semantic segmentationPascal VOC augmented 2012 (val)
mIoU77.9
162
Showing 10 of 142 rows
...

Other info

Code

Follow for update