用答案的对数概率做奖励,让大模型推理更稳定高效。
Likelihood-Based Reward Designs for General LLM Reasoning
- 用参考答案的对数概率作为奖励信号,无需额外验证器。
- 在可验证和不可验证场景下均表现良好,尤其在长文本推理中胜过传统方法。
- 与预训练损失一致,适合链式思维微调,适合研究者和工程师使用。
通过强化学习微调大语言模型(LLMs)进行推理时,通常需要为每个基准设计特定的奖励函数,常为二值奖励。这带来两个问题:需人工设计奖励,且二值奖励可能稀疏。本文系统研究基于生成参考答案或数据中其他续写内容的概率或对数概率的奖励机制,其优势在于无需特定验证器,且可大规模获取。我们对比多种似然奖励变体与标准基线,在标准数学推理基准和无外部验证器的长文本回答上评估性能。结果表明,将参考答案的对数概率作为链式思维(CoT)学习的奖励是唯一在所有设置中表现良好的方案。该奖励与预训练阶段使用的下一个词对数似然损失一致。在可验证场景中,对数概率奖励取得与二值奖励相当或更优的成功率,并显著降低困惑度;在不可验证场景中,表现接近监督微调(SFT)。而基于概率的方法(如VeriFree)在不可验证场景中因正确答案概率趋零而失效。总体而言,对数概率奖励成为链式思维微调的可行方法,贯通了短、可验证与长、不可验证回答场景。
原文摘要 · Abstract (English)
Fine-tuning large language models (LLMs) on reasoning benchmarks via reinforcement learning requires a specific reward function, often binary, for each benchmark. This comes with two potential limitations: the need to design the reward, and the potentially sparse nature of binary rewards. Here, we systematically investigate rewards derived from the probability or log-probability of emitting the reference answer (or any other prompt continuation present in the data), which have the advantage of not relying on specific verifiers and being available at scale. Several recent works have advocated for the use of similar rewards (e.g., VeriFree, JEPO, RLPR, NOVER). We systematically compare variants of likelihood-based rewards with standard baselines, testing performance both on standard mathematical reasoning benchmarks, and on long-form answers where no external verifier is available. We find that using the log-probability of the reference answer as the reward for chain-of-thought (CoT) learning is the only option that performs well in all setups. This reward is also consistent with the next-token log-likelihood loss used during pretraining. In verifiable settings, log-probability rewards bring comparable or better success rates than reinforcing with standard binary rewards, and yield much better perplexity. In non-verifiable settings, they perform on par with SFT. On the other hand, methods based on probability, such as VeriFree, flatline on non-verifiable settings due to vanishing probabilities of getting the correct answer. Overall, this establishes log-probability rewards as a viable method for CoT fine-tuning, bridging the short, verifiable and long, non-verifiable answer settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。