arXiv:2503.13551cs.CLcs.AI2025-03ACL被引 16

提出分层奖励模型,让大模型推理更可靠且省数据标注成本。

Towards Hierarchical Multi-Step Reward Models for Enhanced Reasoning in Large Language Models

  • 分层评估每一步和连续步骤的推理质量,提升逻辑连贯性判断。
  • 在PRM800K等数据集上表现更稳定,错误修正后仍能正确评分。
  • 用轻量压缩策略降低标注成本,适合需要高效训练的场景。

近期研究表明,大语言模型通过监督微调或强化学习具备强大推理能力。然而,主流的流程奖励模型(PRM)存在奖励欺骗问题,难以准确识别最优中间步骤,且推理过程标注成本高,难获大规模高质量数据。为此,我们提出分层奖励模型(HRM),可同时在细粒度与粗粒度层面评估单步及连续推理步骤。HRM在错误步骤被自我反思修正后仍能保持准确评估,显著提升多步推理一致性判断。为降低数据生成成本,引入轻量级数据增强策略HNC,将树结构中两个连续推理步骤合并。结合蒙特卡洛树搜索生成的轨迹应用HNC,有效提升训练数据多样性与鲁棒性,仅引入可控噪声且计算开销极小。在PRM800K上的实验证明,HRM与HNC组合比PRM更具稳定性与可靠性。跨领域评估在MATH500和GSM8K上也显示其强泛化能力与任务适应性。

原文摘要 · Abstract (English)

Recent studies show that Large Language Models (LLMs) achieve strong reasoning capabilities through supervised fine-tuning or reinforcement learning. However, a key approach, the Process Reward Model (PRM), suffers from reward hacking, making it unreliable in identifying the best intermediate step. In addition, the cost of annotating reasoning processes for reward modeling is high, making large-scale collection of high-quality data challenging. To address this, we propose a novel reward model approach called the Hierarchical Reward Model (HRM), which evaluates both individual and consecutive reasoning steps at both fine-grained and coarse-grained levels. HRM excels at assessing multi-step reasoning coherence, especially when flawed steps are later corrected through self-reflection. To further reduce the cost of generating training data, we introduce a lightweight and effective data augmentation strategy called Hierarchical Node Compression (HNC), which merges two consecutive reasoning steps into one within the tree structure. By applying HNC to MCTS-generated reasoning trajectories, we enhance the diversity and robustness of HRM training data while introducing controlled noise with minimal computational overhead. Empirical results on the PRM800K dataset show that HRM, together with HNC, provides more stable and reliable evaluations than PRM. Furthermore, cross-domain evaluations on the MATH500 and GSM8K datasets demonstrate HRM's strong generalization and robustness across a variety of reasoning tasks.

推理增强奖励模型数据压缩大模型

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