论文

RLEC:让单个LLM按指定置信度目标回答或弃答

Reinforcement Learning Done Right for Explicit Confidence Targets

精选理由

训练后的模型能按你指定的准确率目标自动决定答还是不答,不用再调阈值,做可靠性应用的很值得看。

论文提出RLEC方法,让一个大语言模型在推理时接受用户指定的置信度目标p,并自行决定回答还是弃答,目标是回答部分准确率不低于p。作者指出现有方法失败的两个原因:区分不同p值的奖励信号在优化中丢失,且固定错误罚分在p趋近1时无界增长。RLEC通过保留区分信号、引入稳定奖励族和奖励调度,首次成功训练出支持显式置信度目标的单一模型,在多个基准和模型规模上保持通用能力并取得有竞争力的校准表现。

原文 · arXiv cs.LG

Reinforcement Learning Done Right for Explicit Confidence Targets

Large language models (LLMs) can produce plausible but incorrect answers when they are uncertain, which can mislead users and undermine model reliability. Training models to abstain with reinforcement learning is a natural approach, but most existing methods use a fixed error penalty and therefore cannot adapt their behavior to different confidence requirements at inference time. We study \emph{explicit confidence targets}, where a user specifies a target \(p\) in the prompt and the model directly chooses whether to answer or abstain, with the goal of achieving an accuracy of at least \(p\) among answered questions. However, learning to respond to different confidence targets with a single model has proven difficult, and prior attempts have failed to distinguish different targets. We show that this failure comes from two problems: the reward signal that distinguishes different values of $p$ is often lost during optimization, while the original error penalty grows without bound as $p$ approaches one. We propose RLEC (\textbf{R}einforcement \textbf{L}earning with \textbf{E}xplicit \textbf{C}onfidence Targets), which preserves the learning signal needed to distinguish confidence targets, introduces a stable reward family that keeps the desired answer condition unchanged, and uses reward scheduling to balance effective learning and training stability. Together, these designs enable the first successful training of a single LLM for explicit confidence target control. Extensive experiments across diverse benchmarks and model scales show that RLEC largely preserves general capabilities and achieves competitive calibration performance. RLEC also supports fine-grained target control without an external decision system or threshold calibration after training. By making explicit confidence targets part of the trained policy, RLEC opens a new and more direct path toward behavioral calibration.