AI News HubLIVE
站内改写2 分钟阅读

校准偏好学习:标签排序的案例

本文正式定义了标签排序中的校准概念,开发了涵盖全排序、子排序和top-k排序的层次结构,证明全排序校准蕴含其他但反之不成立,且子排序和top-k校准不可比。实验发现流行标签排序模型校准不佳,且校准与RLHF奖励模型的基准准确性强相关但不完美,揭示了超越top-1准确性的重要质量维度。

来源arXiv Machine Learning作者: Santo M. A. R. Thies, Viktor Bengs, Timo Kaufmann, Sebastian J. Vollmer, Eyke H\"ullermeier

校准(Calibration)是预测概率与真实结果频率之间的对齐程度,对于构建可靠的决策系统至关重要。尽管在分类和回归任务中,校准已经得到了广泛的研究,但在概率标签排序(Probabilistic Label Ranking)领域,校准尚未被正式定义。标签排序的目标是预测一个标签集合上的排序分布,而不仅仅是预测单个标签。传统的处理方法将排序视为类别,这忽略了排序的内在结构,无法捕捉成对比较(pairwise)和top-k预测等重要模态。

为了填补这一空白,本研究首次为概率标签排序形式化了校准概念,并开发了一个层次结构,涵盖了全排序(full rankings)、子排序(sub-rankings)和top-k排序三种校准概念。研究人员证明,全排序校准蕴含子排序校准和top-k校准,但反之并不成立,并且子排序校准和top-k校準之间是不可比较的。这意味着,一个模型如果实现了全排序校准,那么它自然也会在子排序和top-k意义上校准,但反过来不一定,而且子排序校准和top-k校准各自捕捉了不同的质量维度。

在实验部分,研究人员评估了多个流行的标签排序模型,发现这些模型普遍校准不佳,尤其是在子排序和top-k指标上存在显著差异。例如,某些模型可能在top-1预测上表现良好,但在更细粒度的子排序或top-k排名上,其概率估计与实际频率严重偏离。这表明,仅仅关注top-1准确性可能会掩盖模型在排序任务中的真实可靠性。

进一步地,研究团队将这一校准框架应用于强化学习从人类反馈(RLHF)中的奖励模型。RLHF是训练大型语言模型的关键技术,其奖励模型通常需要输出偏好排序。实验结果显示,校准与基准准确性之间存在强相关,但并非完美相关。这说明校准捕捉了超越top-1准确性的有意义的质量维度,例如模型在排名不确定性上的表达能力。这一发现对于改进RLHF训练流程具有重要启示,因为校准良好的奖励模型可能产生更可靠的对齐结果。

总而言之,该研究为标签排序中的校准问题奠定了理论基础,并提供了实证证据,表明当前模型在此方面存在不足。未来的工作可以探索错误校准的下游影响,例如在排序聚合或决策制定中的后果,并开发针对性的校准方法,如温度缩放或保序回归。这些努力有望提升机器学习系统在排序和偏好学习任务中的可信度和实用性。