arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.13192stat.MLcs.LG

面向正样本-无标签学习的不确定性感知伪标签选择

Uncertainty-aware Pseudo-label Selection for Positive-Unlabeled Learning

Ludwig-Maximilians-Universität München(慕尼黑大学);Helmholtz Zentrum München—German Research Center for Environmental Health(亥姆霍兹慕尼黑中心——德国环境健康研究中心);Munich Center for Machine Learning(慕尼黑机器学习中心);Fraunhofer Institute for Integrated Circuits IIS(弗劳恩霍夫集成电路研究所);Technical University of Munich(慕尼黑工业大学)

Emilio Dorigatti, Jann Goschenhofer, Benjamin Schubert, Mina Rezaei, Bernd Bischl

AI总结:

针对正样本-无标签学习在不平衡数据集下性能受限的问题,提出不确定性感知伪标签方法PUUPL,通过增强少数类信号扩展标记集并量化不确定性抑制确认偏差,在多场景下优于基线,经真实应用验证有效。

摘要:

正样本-无标签学习(Positive-Unlabeled Learning,PUL)旨在仅利用正样本和无标签训练数据学习二分类器。尽管现实应用中常存在多数样本属于某一类别的不平衡数据集,但当前大多数PUL方法并未研究该场景下的性能,严重限制了其实际适用性。为此,本文提出通过不确定性感知伪标签流程(PUUPL)解决PUL场景下的数据集不平衡与模型校准问题:通过增强少数类的信号,伪标签利用无标签集中的新样本扩展已标记数据集,同时显式不确定性量化可防止有害确认偏差的产生,从而提升预测性能。一系列实验表明,在高度不平衡场景下,PUUPL相较近期基线方法取得了显著性能提升,在平衡PU场景中也表现优异。此外,本文通过消融实验与敏感性分析阐明了PUUPL各组成部分的作用。最后,在一个不平衡数据集的真实应用中,验证了所提方法的优势。

英文摘要:

Positive-unlabeled learning (PUL) aims at learning a binary classifier from only positive and unlabeled training data. Even though real-world applications often involve imbalanced datasets where the majority of examples belong to one class, most contemporary approaches to PUL do not investigate performance in this setting, thus severely limiting their applicability in practice. In this work, we thus propose to tackle the issues of imbalanced datasets and model calibration in a PUL setting through an uncertainty-aware pseudo-labeling procedure (PUUPL): by boosting the signal from the minority class, pseudo-labeling expands the labeled dataset with new samples from the unlabeled set, while explicit uncertainty quantification prevents the emergence of harmful confirmation bias leading to increased predictive performance. Within a series of experiments, PUUPL yields substantial performance gains in highly imbalanced settings while also showing strong performance in balanced PU scenarios across recent baselines. We furthermore provide ablations and sensitivity analyses to shed light on PUUPL's several ingredients. Finally, a real-world application with an imbalanced dataset confirms the advantage of our approach.

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

↑