Share your thoughts, 1 month free Claude Pro on usSee more
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)
mIoU38
2888
Semantic segmentationPASCAL VOC 2012 (val)
Mean IoU80.43
2142
Semantic segmentationGTA5 → Cityscapes (val)
mIoU72.1
586
Semantic segmentationCityscapes (val)
mIoU79.5
572
Change DetectionLEVIR-CD (test)--
485
Change DetectionWHU-CD (test)
IoU85.1
372
Semantic segmentationPASCAL Context (val)
mIoU43.7
360
Semantic segmentationCOCO 2014 (val)
mIoU49.8
304
Semantic segmentationCityscapes (val)
mIoU79.5
297
Medical Image SegmentationBUSI (test)
Dice57.03
216
Showing 10 of 160 rows
...

Other info

Code

Follow for update