Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Debiasing, calibrating, and improving Semi-supervised Learning performance via simple Ensemble Projector

About

Recent studies on semi-supervised learning (SSL) have achieved great success. Despite their promising performance, current state-of-the-art methods tend toward increasingly complex designs at the cost of introducing more network components and additional training procedures. In this paper, we propose a simple method named Ensemble Projectors Aided for Semi-supervised Learning (EPASS), which focuses mainly on improving the learned embeddings to boost the performance of the existing contrastive joint-training semi-supervised learning frameworks. Unlike standard methods, where the learned embeddings from one projector are stored in memory banks to be used with contrastive learning, EPASS stores the ensemble embeddings from multiple projectors in memory banks. As a result, EPASS improves generalization, strengthens feature representation, and boosts performance. For instance, EPASS improves strong baselines for semi-supervised learning by 39.47\%/31.39\%/24.70\% top-1 error rate, while using only 100k/1\%/10\% of labeled data for SimMatch, and achieves 40.24\%/32.64\%/25.90\% top-1 error rate for CoMatch on the ImageNet dataset. These improvements are consistent across methods, network architectures, and datasets, proving the general effectiveness of the proposed methods. Code is available at https://github.com/beandkay/EPASS.

Khanh-Binh Nguyen• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)--
3518
Image ClassificationCIFAR-10 (test)--
3381
Image ClassificationSTL-10 Clean (test)
Error Rate5.4
33
Image ClassificationCIFAR-100 USB (test)
Error Rate16.78
26
Image ClassificationSTL-10 USB (test)
Error Rate21.14
25
Image ClassificationTissueMNIST USB (test)
Error Rate50.4
23
Imbalanced Semi-Supervised LearningCIFAR-10 lambda=50 Long-Tailed (test)
Error Rate13.3
15
Imbalanced Semi-Supervised LearningCIFAR-10 Long-Tailed lambda=150 (test)
Error Rate0.202
15
Imbalanced Semi-Supervised LearningCIFAR-100 Long-Tailed (lambda=20) (test)
Error Rate42.7
15
Image ClassificationImageNet 1K 100K labels
Top-1 Error39.47
15
Showing 10 of 19 rows

Other info

Code

Follow for update