HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Revisiting and Maximizing Temporal Knowledge in Semi-supervised Semantic Segmentation

Wooseok Shin; Hyun Joon Park; Jin Sob Kim; Sung Won Han

Revisiting and Maximizing Temporal Knowledge in Semi-supervised Semantic Segmentation

Abstract

In semi-supervised semantic segmentation, the Mean Teacher- and co-training-based approaches are employed to mitigate confirmation bias and coupling problems. However, despite their high performance, these approaches frequently involve complex training pipelines and a substantial computational burden, limiting the scalability and compatibility of these methods. In this paper, we propose a PrevMatch framework that effectively mitigates the aforementioned limitations by maximizing the utilization of the temporal knowledge obtained during the training process. The PrevMatch framework relies on two core strategies: (1) we reconsider the use of temporal knowledge and thus directly utilize previous models obtained during training to generate additional pseudo-label guidance, referred to as previous guidance. (2) we design a highly randomized ensemble strategy to maximize the effectiveness of the previous guidance. Experimental results on four benchmark semantic segmentation datasets confirm that the proposed method consistently outperforms existing methods across various evaluation protocols. In particular, with DeepLabV3+ and ResNet-101 network settings, PrevMatch outperforms the existing state-of-the-art method, Diverse Co-training, by +1.6 mIoU on Pascal VOC with only 92 annotated images, while achieving 2.4 times faster training. Furthermore, the results indicate that PrevMatch induces stable optimization, particularly in benefiting classes that exhibit poor performance. Code is available at https://github.com/wooseok-shin/PrevMatch

Code Repositories

wooseok-shin/PrevMatch
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-semantic-segmentation-on-1PrevMatch (ResNet-50)
Validation mIoU: 78.8%
semi-supervised-semantic-segmentation-on-1PrevMatch (ResNet-101)
Validation mIoU: 80.1%
semi-supervised-semantic-segmentation-on-10PrevMatch (ResNet-101)
Validation mIoU: 81.6
semi-supervised-semantic-segmentation-on-10PrevMatch (ResNet-50)
Validation mIoU: 79.3
semi-supervised-semantic-segmentation-on-2PrevMatch (ResNet-101)
Validation mIoU: 78.9%
semi-supervised-semantic-segmentation-on-2PrevMatch (ResNet-50)
Validation mIoU: 77.8%
semi-supervised-semantic-segmentation-on-21PrevMatch (ResNet-101)
Validation mIoU: 81.4
semi-supervised-semantic-segmentation-on-22PrevMatch (ResNet-101)
Validation mIoU: 77.7%
semi-supervised-semantic-segmentation-on-22PrevMatch (ResNet-50)
Validation mIoU: 75.8%
semi-supervised-semantic-segmentation-on-27PrevMatch (ResNet-101)
Validation mIoU: 77.0
semi-supervised-semantic-segmentation-on-27PrevMatch (ResNet-50)
Validation mIoU: 73.4
semi-supervised-semantic-segmentation-on-28PrevMatch (ResNet-101)
Validation mIoU: 78.5
semi-supervised-semantic-segmentation-on-28PrevMatch (ResNet-50)
Validation mIoU: 75.4
semi-supervised-semantic-segmentation-on-29PrevMatch (ResNet-50)
Validation mIoU: 77.5
semi-supervised-semantic-segmentation-on-29PrevMatch (ResNet-101)
Validation mIoU: 79.6
semi-supervised-semantic-segmentation-on-30PrevMatch (ResNet-50)
Validation mIoU: 78.6
semi-supervised-semantic-segmentation-on-30PrevMatch (ResNet-101)
Validation mIoU: 80.4
semi-supervised-semantic-segmentation-on-4PrevMatch (ResNet-101)
Validation mIoU: 81.9
semi-supervised-semantic-segmentation-on-8PrevMatch (ResNet-50)
Validation mIoU: 79.2%
semi-supervised-semantic-segmentation-on-8PrevMatch (ResNet-101)
Validation mIoU: 80.1%
semi-supervised-semantic-segmentation-on-9PrevMatch (ResNet-101)
Validation mIoU: 80.8
semi-supervised-semantic-segmentation-on-coco-1PrevMatch
Validation mIoU: 40.2
semi-supervised-semantic-segmentation-on-coco-2PrevMatch
Validation mIoU: 45.7
semi-supervised-semantic-segmentation-on-coco-3PrevMatch
Validation mIoU: 48.4

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp