References & Citations
Computer Science > Machine Learning
Title: Bridging the Gap: Learning Pace Synchronization for Open-World Semi-Supervised Learning
(Submitted on 21 Sep 2023 (v1), last revised 17 Apr 2024 (this version, v2))
Abstract: In open-world semi-supervised learning, a machine learning model is tasked with uncovering novel categories from unlabeled data while maintaining performance on seen categories from labeled data. The central challenge is the substantial learning gap between seen and novel categories, as the model learns the former faster due to accurate supervisory information. Moreover, capturing the semantics of unlabeled novel category samples is also challenging due to the missing label information. To address the above issues, we introduce 1) the adaptive synchronizing marginal loss which imposes class-specific negative margins to alleviate the model bias towards seen classes, and 2) the pseudo-label contrastive clustering which exploits pseudo-labels predicted by the model to group unlabeled data from the same category together in the output space. Extensive experiments on benchmark datasets demonstrate that previous approaches may significantly hinder novel class learning, whereas our method strikingly balances the learning pace between seen and novel classes, achieving a remarkable 3% average accuracy increase on the ImageNet dataset. Importantly, we find that fine-tuning the self-supervised pre-trained model significantly boosts the performance, which is overlooked in prior literature. Our code is available at this https URL
Submission history
From: Bo Ye [view email][v1] Thu, 21 Sep 2023 09:44:39 GMT (7918kb,D)
[v2] Wed, 17 Apr 2024 12:27:25 GMT (7919kb,D)
Link back to: arXiv, form interface, contact.