arXiv:2503.00845cs.CLcs.AI2025-03KDD被引 23

用过程奖励提升大模型图推理能力,显著增强泛化性。

Rewarding Graph Reasoning Process makes LLMs more Generalized Reasoners

  • 用自动化算法生成细粒度推理步骤,构建大规模图推理数据集。
  • 在13个图推理任务中使Qwen2.5-7B准确率提升9%,跨领域有效。
  • 首次实现图推理奖励模型,适合研究通用推理与强化学习的学者。

尽管大语言模型(LLMs)取得显著进展,但提升其高级推理能力仍是关键挑战。过程奖励模型(PRMs)通过提供分步反馈,在数学推理中展现巨大潜力,但在更广泛推理领域应用仍受限,主要因人工构建分步监督成本过高。本文探索将PRMs应用于图推理问题——该领域需复杂多步推理,且可借助成熟图算法自动生成分步标注。我们提出GraphSILO,首个包含细粒度分步标签的图推理数据集,基于任务导向轨迹与蒙特卡洛树搜索(MCTS)自动生成详细推理步骤。基于此数据集,训练首个专为图推理设计的PRM——GraphPRM,评估其在推理时缩放与直接偏好优化(DPO)强化学习中的效果。实验表明,GraphPRM在13个图推理任务中显著提升LLM性能,使Qwen2.5-7B准确率提升9%,并成功迁移至新图推理数据集及数学问题求解等新领域。尤其在GSM8K和Math500上表现优异,证明图推理奖励具备跨领域适用性。研究结果表明PRMs在推动多样化领域推理方面具有巨大潜力,为更通用高效的LLMs铺平道路。

原文摘要 · Abstract (English)

Despite significant advancements in Large Language Models (LLMs), developing advanced reasoning capabilities in LLMs remains a key challenge. Process Reward Models (PRMs) have demonstrated exceptional promise in enhancing reasoning by providing step-wise feedback, particularly in the context of mathematical reasoning. However, their application to broader reasoning domains remains understudied, largely due to the high costs associated with manually creating step-level supervision. In this work, we explore the potential of PRMs in graph reasoning problems - a domain that demands sophisticated multi-step reasoning and offers opportunities for automated step-level data generation using established graph algorithms. We introduce GraphSILO, the largest dataset for graph reasoning problems with fine-grained step-wise labels, built using automated Task-oriented Trajectories and Monte Carlo Tree Search (MCTS) to generate detailed reasoning steps with step-wise labels. Building upon this dataset, we train GraphPRM, the first PRM designed for graph reasoning problems, and evaluate its effectiveness in two key settings: inference-time scaling and reinforcement learning via Direct Preference Optimization (DPO). Experimental results show that GraphPRM significantly improves LLM performance across 13 graph reasoning tasks, delivering a 9% gain for Qwen2.5-7B and demonstrating transferability to new graph reasoning datasets and new reasoning domains like mathematical problem-solving. Notably, GraphPRM enhances LLM performance on GSM8K and Math500, underscoring the cross-domain applicability of graph-based reasoning rewards. Our findings highlight the potential of PRMs in advancing reasoning across diverse domains, paving the way for more versatile and effective LLMs.

图推理过程奖励大模型泛化能力

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