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

CVPR 2023  ยท  Lihe Yang, Lei Qi, Litong Feng, Wayne Zhang, Yinghuan Shi ยท

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.

PDF Abstract CVPR 2023 PDF CVPR 2023 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Semi-supervised Medical Image Segmentation ACDC 10% labeled data UniMatch Dice (Average) 89.92 # 1
Semi-supervised Medical Image Segmentation ACDC 20% labeled data UniMatch Dice (Average) 90.47 # 2
Semi-supervised Medical Image Segmentation ACDC 5% labeled data UniMatch Dice (Average) 87.61 # 1
Semi-Supervised Semantic Segmentation ADE20K 1/16 labeled UniMatch Validation mIoU 31.5 # 3
Semi-Supervised Semantic Segmentation ADE20K 1/32 labeled UniMatch Validation mIoU 28.1 # 3
Semi-Supervised Semantic Segmentation Cityscapes 100 samples labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 73.0 # 2
Semi-Supervised Semantic Segmentation Cityscapes 12.5% labeled UniMatch (DeepLab v3+ with ImageNet-pretrained ResNet-101, single scale inference) Validation mIoU 77.92% # 3
Semi-Supervised Semantic Segmentation Cityscapes 25% labeled UniMatch (DeepLab v3+ with ImageNet-pretrained ResNet-101, single scale inference) Validation mIoU 79.22% # 4
Semi-Supervised Semantic Segmentation Cityscapes 50% labeled UniMatch Validation mIoU 79.5% # 6
Semi-Supervised Semantic Segmentation Cityscapes 6.25% labeled UniMatch (DeepLab v3+ with ResNet-101 pretraind on ImageNet-1K) Validation mIoU 76.59% # 4
Semi-Supervised Semantic Segmentation COCO 1/128 labeled UniMatch Validation mIoU 44.5 # 4
Semi-Supervised Semantic Segmentation COCO 1/256 labeled UniMatch Validation mIoU 38.9 # 4
Semi-Supervised Semantic Segmentation COCO 1/32 labeled UniMatch Validation mIoU 49.8 # 3
Semi-Supervised Semantic Segmentation COCO 1/512 labeled UniMatch Validation mIoU 31.9 # 4
Semi-Supervised Semantic Segmentation COCO 1/64 labeled UniMatch Validation mIoU 48.2 # 4
Semi-supervised Change Detection LEVIR-CD - 10% labeled data UniMatch IoU 82.0 # 2
Semi-supervised Change Detection LEVIR-CD - 20% labeled data UniMatch IoU 81.7 # 2
Semi-supervised Change Detection LEVIR-CD - 40% labeled data UniMatch IoU 82.1 # 2
Semi-supervised Change Detection LEVIR-CD - 5% labeled data UniMatch IoU 80.7 # 2
Semi-Supervised Semantic Segmentation Pascal VOC 2012 12.5% labeled UniMatch Validation mIoU 81.92% # 2
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 1464 labels UniMatch (DeepLab v3 with ResNet-101) Validation mIoU 81.2 # 4
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 183 labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 77.20 # 4
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 25% labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 80.43 # 3
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 366 labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 78.80 # 4
Semi-Supervised Semantic Segmentation Pascal VOC 2012 6.25% labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 80.94 # 2
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 732 labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 79.90 # 4
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 92 labeled UniMatch (DeepLab v3+ with ResNet-101) Validation mIoU 75.20 # 4
Semi-supervised Change Detection WHU - 10% labeled data UniMatch IoU 81.7 # 1
Semi-supervised Change Detection WHU - 20% labeled data UniMatch IoU 81.7 # 2
Semi-supervised Change Detection WHU - 40% labeled data UniMatch IoU 85.1 # 2
Semi-supervised Change Detection WHU - 5% labeled data UniMatch IoU 80.2 # 1

Methods