English

Enhancing Semi-Supervised Learning via Representative and Diverse Sample Selection

Machine Learning 2024-10-29 v2 Computer Vision and Pattern Recognition

Abstract

Semi-Supervised Learning (SSL) has become a preferred paradigm in many deep learning tasks, which reduces the need for human labor. Previous studies primarily focus on effectively utilising the labelled and unlabeled data to improve performance. However, we observe that how to select samples for labelling also significantly impacts performance, particularly under extremely low-budget settings. The sample selection task in SSL has been under-explored for a long time. To fill in this gap, we propose a Representative and Diverse Sample Selection approach (RDSS). By adopting a modified Frank-Wolfe algorithm to minimise a novel criterion α\alpha-Maximum Mean Discrepancy (α\alpha-MMD), RDSS samples a representative and diverse subset for annotation from the unlabeled data. We demonstrate that minimizing α\alpha-MMD enhances the generalization ability of low-budget learning. Experimental results show that RDSS consistently improves the performance of several popular SSL frameworks and outperforms the state-of-the-art sample selection approaches used in Active Learning (AL) and Semi-Supervised Active Learning (SSAL), even with constrained annotation budgets.

Keywords

Cite

@article{arxiv.2409.11653,
  title  = {Enhancing Semi-Supervised Learning via Representative and Diverse Sample Selection},
  author = {Qian Shao and Jiangrui Kang and Qiyuan Chen and Zepeng Li and Hongxia Xu and Yiwen Cao and Jiajuan Liang and Jian Wu},
  journal= {arXiv preprint arXiv:2409.11653},
  year   = {2024}
}

Comments

NeurIPS 2024

R2 v1 2026-06-28T18:48:32.584Z