用两两对比判断法提升大模型推理选优效果
PairJudge RM: Perform Best-of-N Sampling with Knockout Tournament
- 采用双候选方案并行对比判断,避免主观打分偏差
- 在数学难题上实现40%~60%的性能提升
- 适合需要高精度推理的评测与生成场景
最佳- N(BoN)采样是提升大语言模型推理能力的常用策略,依赖奖励模型从多个生成结果中选出最优解。然而传统奖励模型常给出任意且不一致的评分,限制其效果。为此,我们提出成对判别奖励模型(PairJudge RM)结合淘汰赛机制进行BoN采样。针对一道数学题,PairJudge RM同时对两个候选解进行带思维链的正确性判断,无需赋分,通过并行比对实现解法交叉验证。在淘汰赛中,该模型逐轮进行成对判断,剔除错误解。我们构建了包含432,000组成对判断的PairJudge-432K数据集,基于NumiaMath数据集并用gemini-1.5-flash标注,通过监督微调训练模型。在MATH-500和奥数基准测试中,显著优于基线模型,在最困难的前50%问题上实现40%至60%的相对提升。
原文摘要 · Abstract (English)
Best-of-N (BoN) sampling, a common strategy for test-time scaling of Large Language Models (LLMs), relies on reward models to select the best candidate solution from multiple generations. However, traditional reward models often assign arbitrary and inconsistent scores, limiting their effectiveness. To address this, we propose a Pairwise Judge Reward Model (PariJudge RM) combined with a knockout tournament for BoN sampling. Instead of assigning absolute scores, given one math problem, PariJudge RM judges two candidate solutions' correctness with chain-of-thought reasoning simultaneously. This approach eliminates the need for scoring and enables cross-validation of solutions through parallel judgment. In the knockout tournament, PariJudge RM conducts pairwise Judgment between candidate solutions and eliminates the incorrect ones iteratively. We construct PairJudge-432K, a large-scale dataset of 432K pairwise judgments derived from NumiaMath and annotated using \texttt{gemini-1.5-flash}, and train the PariJudge RM via supervised fine-tuning. Experiments on MATH-500 and the Olympiad Bench demonstrate significant improvements over baseline reward models. And a 40\% to 60\% relative improvement is achieved on the top 50\% challenging problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。