English

InterLUDE: Interactions between Labeled and Unlabeled Data to Enhance Semi-Supervised Learning

Computer Vision and Pattern Recognition 2024-03-19 v1 Machine Learning

Abstract

Semi-supervised learning (SSL) seeks to enhance task performance by training on both labeled and unlabeled data. Mainstream SSL image classification methods mostly optimize a loss that additively combines a supervised classification objective with a regularization term derived solely from unlabeled data. This formulation neglects the potential for interaction between labeled and unlabeled images. In this paper, we introduce InterLUDE, a new approach to enhance SSL made of two parts that each benefit from labeled-unlabeled interaction. The first part, embedding fusion, interpolates between labeled and unlabeled embeddings to improve representation learning. The second part is a new loss, grounded in the principle of consistency regularization, that aims to minimize discrepancies in the model's predictions between labeled versus unlabeled inputs. Experiments on standard closed-set SSL benchmarks and a medical SSL task with an uncurated unlabeled set show clear benefits to our approach. On the STL-10 dataset with only 40 labels, InterLUDE achieves 3.2% error rate, while the best previous method reports 14.9%.

Keywords

Cite

@article{arxiv.2403.10658,
  title  = {InterLUDE: Interactions between Labeled and Unlabeled Data to Enhance Semi-Supervised Learning},
  author = {Zhe Huang and Xiaowei Yu and Dajiang Zhu and Michael C. Hughes},
  journal= {arXiv preprint arXiv:2403.10658},
  year   = {2024}
}

Comments

Semi-supervised Learning; Vision Transformers

R2 v1 2026-06-28T15:22:22.253Z