用少量标签+大量无标签数据,更准评估分类器性能。
Evaluating multiple models using labeled and unlabeled data
- 构建半监督混合模型,联合建模标签与预测得分分布。
- 相比仅用标签数据,误差降低5.1倍;优于次优方法2.4倍。
- 适合标签稀缺场景,如医疗、内容审核、分子预测等。
在缺乏大规模标注数据的情况下,机器学习分类器的评估仍具挑战性。尽管标注数据可能成本高昂或难以获取,但无标签数据却极为丰富。本文提出半监督模型评估(SSME),首次利用:(i) 同一任务常存在多个分类器,(ii) 分类器输出连续分数,(iii) 无标签数据远多于标签数据这一特点。核心思想是通过半监督混合模型估计真实标签与分类器预测的联合分布,进而计算准确率、期望校准误差等指标。我们在四个实际场景中验证:(1) 医疗健康,(2) 内容审核,(3) 分子性质预测,(4) 图像标注。结果表明,SSME比现有方法更准确,相较于仅使用标签数据,误差降低5.1倍;相较次优方法,降低2.4倍。此外,该方法在评估测试分布子集(如特定人群)及语言模型性能时也表现更优。
原文摘要 · Abstract (English)
It remains difficult to evaluate machine learning classifiers in the absence of a large, labeled dataset. While labeled data can be prohibitively expensive or impossible to obtain, unlabeled data is plentiful. Here, we introduce Semi-Supervised Model Evaluation (SSME), a method that uses both labeled and unlabeled data to evaluate machine learning classifiers. SSME is the first evaluation method to take advantage of the fact that: (i) there are frequently multiple classifiers for the same task, (ii) continuous classifier scores are often available for all classes, and (iii) unlabeled data is often far more plentiful than labeled data. The key idea is to use a semi-supervised mixture model to estimate the joint distribution of ground truth labels and classifier predictions. We can then use this model to estimate any metric that is a function of classifier scores and ground truth labels (e.g., accuracy or expected calibration error). We present experiments in four domains where obtaining large labeled datasets is often impractical: (1) healthcare, (2) content moderation, (3) molecular property prediction, and (4) image annotation. Our results demonstrate that SSME estimates performance more accurately than do competing methods, reducing error by 5.1x relative to using labeled data alone and 2.4x relative to the next best competing method. SSME also improves accuracy when evaluating performance across subsets of the test distribution (e.g., specific demographic subgroups) and when evaluating the performance of language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。