arXiv · 2311.04055
Feature Space Renormalization for Semi-supervised Learning
Abstract
Semi-supervised learning (SSL) has been proven to be a powerful method for leveraging unlabeled data to alleviate models'dependence on large labeled datasets. The common framework among recent approaches is to train the model on a large amount of unlabeled data with consistency regularization to constrain the model predictions to be invariant to input perturbation. This paper proposes a feature space renormalizati-on (FSR) mechanism for SSL, which imposes consistency on feature representations rather than on labels to enable the model to learn better discriminative features. In order to apply this mechanism to SSL, we design a dual-branch FSR module consisting of a dual-branch header and an FSR block. This module can be seamlessly plugged and played into existing SSL frameworks to enhance the performance of the base SSL. The experimental results show that our proposed FSR module helps the base SSL framework (e.g. CRMatch and FreeMatch), achieve better performance on a variety of standard SSL benchmark datasets, without incurring additional overhead in terms of computation time and GPU memory.
Explore related subjects
Keep this discovery
Jun Sun, Wancheng Zhang, Chao Zhou, Zhongjie Mao, Chao Li, Xiao-Jun Wu. 2023-11-07. Feature Space Renormalization for Semi-supervised Learning. https://arxiv.org/abs/2311.04055
Cite the original work for its findings. Save a collection to share your selection of sources.