HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Switching Temporary Teachers for Semi-Supervised Semantic Segmentation

Switching Temporary Teachers for Semi-Supervised Semantic Segmentation

Abstract

The teacher-student framework, prevalent in semi-supervised semantic segmentation, mainly employs the exponential moving average (EMA) to update a single teacher's weights based on the student's. However, EMA updates raise a problem in that the weights of the teacher and student are getting coupled, causing a potential performance bottleneck. Furthermore, this problem may become more severe when training with more complicated labels such as segmentation masks but with few annotated data. This paper introduces Dual Teacher, a simple yet effective approach that employs dual temporary teachers aiming to alleviate the coupling problem for the student. The temporary teachers work in shifts and are progressively improved, so consistently prevent the teacher and student from becoming excessively close. Specifically, the temporary teachers periodically take turns generating pseudo-labels to train a student model and maintain the distinct characteristics of the student model for each epoch. Consequently, Dual Teacher achieves competitive performance on the PASCAL VOC, Cityscapes, and ADE20K benchmarks with remarkably shorter training times than state-of-the-art methods. Moreover, we demonstrate that our approach is model-agnostic and compatible with both CNN- and Transformer-based models. Code is available at https://github.com/naver-ai/dual-teacher.

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-semantic-segmentation-on-1Dual Teacher (DeepLab v3+ with ResNet-101 pretraind on ImageNet-1K)
Validation mIoU: 79.46
semi-supervised-semantic-segmentation-on-2Dual Teacher
Validation mIoU: 78.4
semi-supervised-semantic-segmentation-on-21Dual Teacher
Validation mIoU: 78.82
semi-supervised-semantic-segmentation-on-22Dual Teacher (DeepLab v3+ with ResNet-101 pretraind on ImageNet-1K)
Validation mIoU: 76.81
semi-supervised-semantic-segmentation-on-4Dual Teacher
Validation mIoU: 81.19
semi-supervised-semantic-segmentation-on-8Dual Teacher
Validation mIoU: 80.52
semi-supervised-semantic-segmentation-on-9Dual Teacher
Validation mIoU: 81.03

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