arXiv:2606.18954cs.CL2026-06被引 2

用图结构共享相似推理路径,提升大模型推理效率与准确性

GraphPO: Graph-based Policy Optimization for Reasoning Models

论文配图:GraphPO: Graph-based Policy Optimization for Reasoning Models
图 1 · 摘自论文原文
  • 将推理过程建模为有向无环图,合并语义相同的路径
  • 在相同预算下,比链式和树式方法准确率提升5%-12%
  • 适合需要高效推理的智能体与复杂问题求解场景

基于可验证奖励的强化学习(RLVR)已成为提升大型推理模型能力的标准范式。传统方法独立采样回答并仅根据最终答案优化策略,存在两个局限:一是独立采样导致中间推理步骤重复,浪费计算;二是最终答案奖励稀疏,难以定位有效步骤。树形方法通过共享前缀、比较同前缀分支提供细粒度信号,但仍独立扩展分支。当不同分支到达相似推理状态时无法共享信息,重复探索,且仅做局部比较,导致优势估计方差较高。为此,我们提出GraphPO(基于图的策略优化),将轨迹表示为有向无环图,推理步骤作为边,从推理路径中总结的语义状态作为节点。GraphPO将语义等价的推理路径归入等价类,允许共享后缀,并将预算重新分配至多样化探索。同时,为进入边分配效率优势,为出边分配正确性优势,从而在提升推理效率的同时,从结果中衍生过程监督。理论证明GraphPO可降低优势估计方差,增强推理效率。在三个LLM上进行的实验表明,在相同令牌或响应预算下,GraphPO在推理与智能体搜索基准上均持续优于链式与树式基线。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has become a standard paradigm for enhancing the capability of large reasoning models. RLVR typically samples responses independently and optimizes the policy using from final answers. This paradigm has two limitations. First, independently responses often contain similar intermediate reasoning steps, causing redundant exploration and wasted computation. Second, sparse final-answer rewards make it hard to identify useful steps. Tree-based methods partly address this problem by sharing prefixes and comparing branches from the same prefix to provide fine-grained signals. However, tree branches are still expanded independently. When different branches reach similar reasoning states, they cannot share information and repeat similar exploration. Moreover, tree-based methods ignore such dispersion and only perform local comparisons within separate branches, which can lead to higher variance in advantage estimation. To address this challenge, we propose GraphPO (Graph-based Policy Optimization), a novel RL framework that represents rollouts as a directed acyclic graph, with reasoning steps as edges and semantic states summarized from the reasoning paths as nodes. GraphPO merges semantically equivalent reasoning paths into equivalence classes, allowing them to share suffixes and reallocating budget away from redundant expansions to diverse exploration. Furthermore, we assign efficiency advantages to incoming edges and correctness advantages to outgoing edges, thereby improving inference efficiency while deriving process supervision from outcome. Theory shows that GraphPO reduces advantage-estimation variance and enhances reasoning efficiency. Experiments on three LLMs across reasoning and agentic search benchmarks show that GraphPO consistently outperforms chain- and tree-based baselines with the same token budgets or response budgets.

强化学习推理优化图神经网络

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