본문 바로가기

세미나

쓰기

 

FlexMatch: Boosting Semi-Supervised Learning with Curriculum Pseudo Labeling

by김누리 2021.11.23 18:01
발표자 김누리 
발표일자 2021-11-23 
저자 Zhang, Bowen, et al. 
학회명 Advances in Neural Information Processing Systems 34 (2021) 
논문지  
The recently proposed FixMatch achieved state-of-the-art results on most semi-supervised learning (SSL) benchmarks. However, like other modern SSL algorithms, FixMatch uses a pre-defined constant threshold for all classes to select unlabeled data that contribute to the training, thus failing to consider different learning status and learning difficulties of different classes. To address this issue, we propose Curriculum Pseudo Labeling (CPL), a curriculum learning approach to leverage unlabeled data according to the model's learning status. The core of CPL is to flexibly adjust thresholds for different classes at each time step to let pass informative unlabeled data and their pseudo labels. CPL does not introduce additional parameters or computations (forward or backward propagation). We apply CPL to FixMatch and call our improved algorithm FlexMatch. FlexMatch achieves state-of-the-art performance on a variety of SSL benchmarks, with especially strong performances when the labeled data are extremely limited or when the task is challenging. For example, FlexMatch outperforms FixMatch by 14.32% and 24.55% on CIFAR-100 and STL-10 datasets respectively, when there are only 4 labels per class. CPL also significantly boosts the convergence speed, e.g., FlexMatch can use only 1/5 training time of FixMatch to achieve even better performance. Furthermore, we show that CPL can be easily adapted to other SSL algorithms and remarkably improve their performances. We open-source our code at https://github.com/TorchSSL/TorchSSL.

첨부 [1]
목록

    2022

      GPipe: Easy Scaling with Micro-Batch Pipeline Parallelism
      2022.02.22
      발표자: 조영성     발표일자: 2022-02-22     저자: Yanping Huang, Youlong Cheng, Ankur Bapna, Orhan Firat, Mia Xu Chen, Dehao Chen, HyoukJoong Lee, Jiquan Ngiam, Quoc V. Le, Yonghui Wu, Zhifeng Chen     논문지: https://arxiv.org/abs/1811.06965    
      Contrastive Code Representation Learning
      2022.02.15
      발표자: 최윤석     발표일자: 2022-02-15     저자: Paras Jain, Ajay Jain, Tianjun Zhang, Pieter Abbeel, Joseph E. Gonzalez, Ion Stoica     학회명: EMNLP 2021    
이전 Prev 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 다음 Next