arXiv:2602.14386cs.CL2026-02被引 1

让大模型一次生成多个词,提升复杂推理能力

Beyond Token-Level Policy Gradients for Complex Reasoning with Large Language Models

  • 把连续K个词当作一个整体动作优化,而非逐词决策
  • 在数学和编程任务上优于传统逐词优化方法
  • 适合需要连贯逻辑的复杂推理场景

现有自回归语言模型的策略梯度方法通常逐个选择后续词作为动作。虽然在多数生成任务中有效,但在复杂推理任务中可能不充分,因为单个语义决策常跨越多个词(如定义变量或组合方程)。这导致逐词优化与推理的块级本质之间存在潜在错配。为此,我们提出多词策略梯度优化(MPO),将连续K个词视为统一的语义动作。该块级视角使方法能捕捉推理轨迹的组合结构,并支持对连贯的高层次目标进行优化。在数学推理和编程基准上的实验表明,MPO优于标准的逐词策略梯度基线,揭示了逐词策略梯度在复杂推理中的局限性,推动未来研究向超越逐词粒度的方向发展。

原文摘要 · Abstract (English)

Existing policy-gradient methods for auto-regressive language models typically select subsequent tokens one at a time as actions in the policy. While effective for many generation tasks, such an approach may not fully capture the structure of complex reasoning tasks, where a single semantic decision is often realized across multiple tokens--for example, when defining variables or composing equations. This introduces a potential mismatch between token-level optimization and the inherently block-level nature of reasoning in these settings. To bridge this gap, we propose Multi-token Policy Gradient Optimization (MPO), a framework that treats sequences of K consecutive tokens as unified semantic actions. This block-level perspective enables our method to capture the compositional structure of reasoning trajectories and supports optimization over coherent, higher-level objectives. Experiments on mathematical reasoning and coding benchmarks show that MPO outperforms standard token-level policy gradient baselines, highlight the limitations of token-level policy gradients for complex reasoning, motivating future research to look beyond token-level granularity for reasoning-intensive language tasks.

大模型推理策略梯度代码生成

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