HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Dual Student: Breaking the Limits of the Teacher in Semi-supervised Learning

Zhanghan Ke; Daoye Wang; Qiong Yan; Jimmy Ren; Rynson W.H. Lau

Dual Student: Breaking the Limits of the Teacher in Semi-supervised Learning

Abstract

Recently, consistency-based methods have achieved state-of-the-art results in semi-supervised learning (SSL). These methods always involve two roles, an explicit or implicit teacher model and a student model, and penalize predictions under different perturbations by a consistency constraint. However, the weights of these two roles are tightly coupled since the teacher is essentially an exponential moving average (EMA) of the student. In this work, we show that the coupled EMA teacher causes a performance bottleneck. To address this problem, we introduce Dual Student, which replaces the teacher with another student. We also define a novel concept, stable sample, following which a stabilization constraint is designed for our structure to be trainable. Further, we discuss two variants of our method, which produce even higher performance. Extensive experiments show that our method improves the classification performance significantly on several main SSL benchmarks. Specifically, it reduces the error rate of the 13-layer CNN from 16.84% to 12.39% on CIFAR-10 with 1k labels and from 34.10% to 31.56% on CIFAR-100 with 10k labels. In addition, our method also achieves a clear improvement in domain adaptation.

Code Repositories

60972823l/SSL-DNLL
pytorch
Mentioned in GitHub
ZHKKKe/DualStudent
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-image-classification-on-2Dual Student
Top 1 Accuracy: 63.52%
Top 5 Accuracy: 83.58%
semi-supervised-image-classification-on-cifarDual Student (600)
Percentage error: 8.89
semi-supervised-image-classification-on-cifar-11Dual Student (600)
Accuracy: 85.83
semi-supervised-image-classification-on-cifar-12Dual Student (600)
Accuracy: 89.28
semi-supervised-image-classification-on-cifar-2Dual Student (480)
Percentage error: 32.77
semi-supervised-image-classification-on-svhn-1Dual Student
Accuracy: 95.76
semi-supervised-image-classification-on-svhn-3Dual Student
Accuracy: 96.04

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
Dual Student: Breaking the Limits of the Teacher in Semi-supervised Learning | Papers | HyperAI