English

Pseudo Labels for Single Positive Multi-Label Learning

Machine Learning 2023-06-05 v1 Computer Vision and Pattern Recognition

Abstract

The cost of data annotation is a substantial impediment for multi-label image classification: in every image, every category must be labeled as present or absent. Single positive multi-label (SPML) learning is a cost-effective solution, where models are trained on a single positive label per image. Thus, SPML is a more challenging domain, since it requires dealing with missing labels. In this work, we propose a method to turn single positive data into fully-labeled data: Pseudo Multi-Labels. Basically, a teacher network is trained on single positive labels. Then, we treat the teacher model's predictions on the training data as ground-truth labels to train a student network on fully-labeled images. With this simple approach, we show that the performance achieved by the student model approaches that of a model trained on the actual fully-labeled images.

Keywords

Cite

@article{arxiv.2306.01034,
  title  = {Pseudo Labels for Single Positive Multi-Label Learning},
  author = {Julio Arroyo},
  journal= {arXiv preprint arXiv:2306.01034},
  year   = {2023}
}

Comments

ICLR 2023, Tiny Papers Track

R2 v1 2026-06-28T10:53:51.316Z