arXiv:2411.04276cs.LG2024-11被引 1

评估极端多标签分类器的概率校准能力,提出新指标ECE@k

Labels in Extremes: How Well Calibrated are Extreme Multi-label Classifiers?

  • 提出校准@k(ECE@k)衡量顶部k个标签的概率可靠性
  • 不同模型校准性能差异大,但事后等距回归可显著提升校准度
  • 适合需要准确概率估计的广告推荐、文档标注等场景

极端多标签分类(XMLC)问题常见于相关产品推荐、大规模文档打标或广告预测,标签空间可达数百万。分类器隐含两个任务:评估每个标签的预期价值,并选择最优候选。传统评估仅关注分数相对顺序,但实际应用中准确估计标签相关概率至关重要,例如判断是否支付广告展示费用。现有研究未系统评估校准性。本文通过九种来自四个模型族的模型,在七个基准数据集上进行系统评估,发现直接使用期望校准误差(ECE)在长尾分布下无效。为此,提出校准@k(如ECE@k),聚焦前k个标签的概率质量,更适配XMLC场景。实验显示各模型校准表现差异显著,但通过计算高效的等距回归进行事后校准,可显著改善校准性且不牺牲精度。因此,从业者可优先考虑准确性选模型,再用等距回归实现校准。

原文摘要 · Abstract (English)

Extreme multilabel classification (XMLC) problems occur in settings such as related product recommendation, large-scale document tagging, or ad prediction, and are characterized by a label space that can span millions of possible labels. There are two implicit tasks that the classifier performs: \emph{Evaluating} each potential label for its expected worth, and then \emph{selecting} the best candidates. For the latter task, only the relative order of scores matters, and this is what is captured by the standard evaluation procedure in the XMLC literature. However, in many practical applications, it is important to have a good estimate of the actual probability of a label being relevant, e.g., to decide whether to pay the fee to be allowed to display the corresponding ad. To judge whether an extreme classifier is indeed suited to this task, one can look, for example, to whether it returns \emph{calibrated} probabilities, which has hitherto not been done in this field. Therefore, this paper aims to establish the current status quo of calibration in XMLC by providing a systematic evaluation, comprising nine models from four different model families across seven benchmark datasets. As naive application of Expected Calibration Error (ECE) leads to meaningless results in long-tailed XMC datasets, we instead introduce the notion of \emph{calibration@k} (e.g., ECE@k), which focusses on the top-$k$ probability mass, offering a more appropriate measure for evaluating probability calibration in XMLC scenarios. While we find that different models can exhibit widely varying reliability plots, we also show that post-training calibration via a computationally efficient isotonic regression method enhances model calibration without sacrificing prediction accuracy. Thus, the practitioner can choose the model family based on accuracy considerations, and leave calibration to isotonic regression.

极端多标签概率校准等距回归

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