arXiv:2602.08042stat.MLcs.LG2026-02

通过最大化AUC提升图模型分类边界,更适应复杂标签分布。

Graph-based Semi-Supervised Learning via Maximum Discrimination

  • 用AUC优化代替传统距离度量,增强类别分离能力
  • 所需标注样本数为模型参数的多项式级别
  • 兼顾分类性能与计算效率,适合小样本场景

半监督学习(SSL)旨在标签数据稀缺但无标签数据丰富的情况下训练高精度模型。基于图的半监督学习(GSSL)通过图结构表示捕捉数据内在结构。经典方法如标签传播和标签扩散试图在低维空间中使相同标签的点相互靠近。然而,这些方法在复杂标签分布的数据上表现不佳。本文提出AUC-spec,一种基于图的方法,通过优化由标注点估计的受试者工作特征曲线下面积(AUC)来计算具有最大类间分离性的低维表示。我们在产品流形模型下对方法进行了详细分析,证明所需标注点数量为模型参数的多项式级别。实验表明,AUC-spec在保持与经典及先进方法相当的计算效率的同时,在合成与真实数据集上实现了有竞争力的性能,有效平衡了类分离性与图平滑性。

原文摘要 · Abstract (English)

Semi-supervised learning (SSL) addresses the critical challenge of training accurate models when labeled data is scarce but unlabeled data is abundant. Graph-based SSL (GSSL) has emerged as a popular framework that captures data structure through graph representations. Classic graph SSL methods, such as Label Propagation and Label Spreading, aim to compute low-dimensional representations where points with the same labels are close in representation space. Although often effective, these methods can be suboptimal on data with complex label distributions. In our work, we develop AUC-spec, a graph approach that computes a low-dimensional representation that maximizes class separation. We compute this representation by optimizing the Area Under the ROC Curve (AUC) as estimated via the labeled points. We provide a detailed analysis of our approach under a product-of-manifold model, and show that the required number of labeled points for AUC-spec is polynomial in the model parameters. Empirically, we show that AUC-spec balances class separation with graph smoothness. It demonstrates competitive results on synthetic and real-world datasets while maintaining computational efficiency comparable to the field's classic and state-of-the-art methods.

图神经网络半监督学习AUC优化

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