SPARE让大模型一步生成带参考评估的解题过程,高效训练推理奖励模型。
SPARE: Single-Pass Annotation with Reference-Guided Evaluation for Automatic Process Supervision and Reward Modelling
- 单次生成中对齐参考解法并评估每步正确性,实现高效过程标注。
- 仅用16%数据在ProcessBench上达到领先性能,比MCTS快2.3倍。
- 适合训练推理奖励模型与强化学习微调,可替代人工标注。
过程监督在提升大语言模型多步推理能力方面至关重要,但高效高质量的自动过程标注仍是难题。为此,我们提出单次生成参考引导评估(SPARE),一种新型结构化框架,通过一次生成同时对齐解题步骤与参考解法,并显式判断其正确性。我们在四个不同数据集上验证了SPARE的有效性:数学推理(GSM8K、MATH)、多跳问答(MuSiQue-Ans)和空间推理(SpaRP)。结果表明,该方法在两项应用中均表现优异:(1)用于训练过程奖励模型(PRMs)以排序和聚合多个生成结果;(2)通过离线强化学习微调模型实现贪心解码。在ProcessBench上,SPARE仅需约16%的训练样本即可实现跨分布高效泛化,优于人类标注及其它合成训练基线。同时,其性能与基于MCTS的方法相当,但总词元数减少2.3倍。人工分析显示其与MCTS在精确率-召回率上具有互补性,提示可构建集成方法。这些结果确立了SPARE作为大模型推理中自动过程监督的实用且可扩展方案。
原文摘要 · Abstract (English)
Process or step-wise supervision has played a crucial role in advancing complex multi-step reasoning capabilities of Large Language Models (LLMs). However, efficient, high-quality automated process annotation remains a significant challenge. To address this, we introduce Single-Pass Annotation with Reference-Guided Evaluation (SPARE), a novel structured framework that enables efficient per-step annotation by jointly aligning solution steps to reference solutions and determine its accuracy with explicit reasoning in single generation. We demonstrate SPARE's effectiveness across four diverse datasets spanning mathematical reasoning (GSM8K, MATH), multi-hop question answering (MuSiQue-Ans), and spatial reasoning (SpaRP), showing consistent improvements in two applications: (1) training Process Reward Models (PRMs) for ranking and aggregating multiple generations, and (2) fine-tuning models via offline reinforcement learning for greedy decoding. On ProcessBench, SPARE demonstrates data-efficient out-of-distribution generalization, using only $\sim$16% of training samples compared to human-labeled and other synthetically trained baselines. Additionally, it achieves competitive performance with MCTS-based methods while offering 2.3$\times$ speedup in terms of total token count. Manual analysis reveals complementary precision-recall characteristics with MCTS approaches, suggesting potential for ensemble methods. These results establish SPARE as a practical and scalable solution for automatic process supervision in LLM reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。