arXiv:2505.08212cs.LG2025-05

用图流方法排序未标注样本,提升正例-未标注学习效果

An Effective Flow-based Method for Positive-Unlabeled Learning: 2-HNC

  • 基于相似性构建图流,通过嵌套分割生成负样本排序
  • 在多个数据集上超越现有最优算法,准确率显著提升
  • 适合正例少、负例未知的场景,如异常检测与医疗诊断

在二分类任务中,训练数据仅提供正例,其余为未标注样本,这种设定称为正例-未标注(PU)学习。本文提出一种基于网络流的方法2-HNC,利用样本间的成对相似性。该方法基于Hochbaum的归一化割(HNC),通过求解参数化最小割问题,获得一系列嵌套划分,对应不同内外相似度权衡。这些划分用于对未标注样本按负例可能性进行排序。2-HNC分两阶段:第一阶段仅依赖正例,生成排序;第二阶段将高可能性负样本加入正集并重新分类。最终预测选择使正类比例最接近先验估计的划分。在合成与真实数据集上的大量实验表明,2-HNC表现优异,常优于现有最先进方法。

原文摘要 · Abstract (English)

In many scenarios of binary classification, only positive instances are provided in the training data, leaving the rest of the data unlabeled. This setup, known as positive-unlabeled (PU) learning, is addressed here with a network flow-based method which utilizes pairwise similarities between samples. The method we propose here, 2-HNC, leverages Hochbaum's Normalized Cut (HNC) and the set of solutions it provides by solving a parametric minimum cut problem. The set of solutions, that are nested partitions of the samples into two sets, correspond to varying tradeoff values between the two goals: high intra-similarity inside the sets and low inter-similarity between the two sets. This nested sequence is utilized here to deliver a ranking of unlabeled samples by their likelihood of being negative. Building on this insight, our method, 2-HNC, proceeds in two stages. The first stage generates this ranking without assuming any negative labels, using a problem formulation that is constrained only on positive labeled samples. The second stage augments the positive set with likely-negative samples and recomputes the classification. The final label prediction selects among all generated partitions in both stages, the one that delivers a positive class proportion, closest to a prior estimate of this quantity, which is assumed to be given. Extensive experiments across synthetic and real datasets show that 2-HNC yields strong performance and often surpasses existing state-of-the-art algorithms.

PU学习图流方法无监督排序

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。