arXiv:2607.03528cs.LGcs.AI2026-07中稿 · ICML

让大模型学会挑着回答,只在有把握时输出,降低错误率。

Aligning Language Models with Selective Prediction

论文配图:Aligning Language Models with Selective Prediction
图 1 · 摘自论文原文
  • 用强化学习优化模型选择性预测能力,直接提升风险与覆盖的平衡。
  • 在域内和域外任务上,错误率显著下降,同时保持较高响应覆盖率。
  • 适合对可靠性要求高的实际应用,如医疗、金融等高风险场景。

大型语言模型(LLMs)正越来越多地被部署为高风险现实世界AI系统中的关键决策组件,模型可靠性成为首要关注问题。本文聚焦于通过选择性预测(SP)提升LLM可靠性——即仅对模型可能正确的情况进行预测(覆盖),其余输入则标记为需人工介入。这本质上是在风险与覆盖之间取得平衡,实现人机协作。为将SP融入LLM,我们关注后训练对齐阶段,提出以选择性预测性能指标为目标进行对齐,不同于传统方法侧重准确率或校准度。我们提出一种新框架:用于选择奖励的强化学习(RLSR),以风险-覆盖曲线下面积(AURC)为优化目标。在多个域内与域外任务中,该方法相比多种基线显著改善了风险-覆盖权衡。

原文摘要 · Abstract (English)

Large language models (LLMs) are increasingly deployed as critical decision-making components in high-stakes real-world AI systems, rendering LLM reliability a foremost practical concern. In this paper, we focus on enhancing LLM reliability through selective prediction (SP), a strategy that allows an LLM to only predict for inputs where it is likely to be correct (i.e., coverage) and hence reduce the error rate (i.e., risk) on that portion of inputs -- flagging the remaining inputs for future human discretion. In other words, SP improves LLM reliability by balancing the risk-coverage trade-off and enabling seamless human-AI collaboration. To integrate SP into LLMs, we focus on the LLM post-training alignment stage and propose to align LLMs with SP performance metrics, in contrast with existing LLM alignment methods that focus primarily on correctness or calibration metrics. Specifically, we propose a novel alignment framework, Reinforcement Learning for Selection Reward (RLSR), which targets the area under the risk-coverage curve (AURC) -- a popular SP performance metric -- as its alignment objective. RLSR achieves substantially better risk-coverage trade-off compared to multiple alignment baselines on both in-domain and out-of-domain tasks.

大模型选择性预测可靠性强化学习

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