arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.01034cs.LGcs.CV

用于单正例多标签学习的伪标签

Pseudo Labels for Single Positive Multi-Label Learning

California Institute of Technology(加州理工学院)

Julio Arroyo

AI总结:

针对单正例多标签学习中的缺失标签问题,本文提出Pseudo Multi-Labels,通过教师网络生成伪全标签并训练学生网络,使其性能接近真实全标签监督模型。

摘要:

数据标注成本是多标签图像分类的一大显著障碍:在每张图像中,每个类别都必须被标注为存在或不存在。单正例多标签(SPML)学习是一种具有成本效益的解决方案,其中模型基于每张图像的单个正标签进行训练。因此,SPML是一个更具挑战性的领域,因为它需要处理缺失标签。在本工作中,我们提出一种将单正例数据转化为全标签数据的方法:Pseudo Multi-Labels(伪多标签)。具体而言,首先在单正例标签上训练一个教师网络。随后,我们将教师模型在训练数据上的预测视为真实标签,以在全标签图像上训练一个学生网络。通过这一简单方法,我们表明学生模型取得的性能接近在真实全标签图像上训练的模型。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑