arXiv:2605.14558cs.LGcs.AI2026-05被引 3

让大模型更专注行动决策,提升智能体强化学习效果

Resolving Action Bottleneck: Agentic Reinforcement Learning Informed by Token-Level Energy

论文配图:Resolving Action Bottleneck: Agentic Reinforcement Learning Informed by Token-Level Energy
图 1 · 摘自论文原文
  • 根据令牌能量重新加权,突出行动令牌的训练信号
  • 在4个环境中表现优于传统方法,最高提升65.2个百分点
  • 无需额外计算开销,适合部署于资源受限场景

智能体强化学习利用多轮轨迹训练大语言模型,其中包含长篇推理与短时环境动作。现有策略梯度方法(如PPO、GRPO)对轨迹中每个令牌一视同仁,导致信用分配不均。本文从能量模型视角出发,发现令牌训练信号主要集中在动作令牌上,即使其仅占轨迹极小比例。该现象称为“行动瓶颈”。为此提出简单有效的重加权方法ActFocus,降低推理令牌梯度权重,并引入基于能量的再分配机制,进一步增强高不确定性动作令牌的权重。在四个环境及不同模型规模下,ActFocus显著优于PPO和GRPO,最终步性能分别提升最高达65.2和63.7个百分点,且无额外运行时间或内存开销。

原文摘要 · Abstract (English)

Agentic reinforcement learning trains large language models using multi-turn trajectories that interleave long reasoning traces with short environment-facing actions. Common policy-gradient methods, such as PPO and GRPO, treat each token in a trajectory equally, leading to uniform credit assignment. In this paper, we critically demonstrate that such uniform credit assignment largely misallocates token-level training signals. From an energy-based modeling perspective, we show that token-level training signals, quantified by their correlations with reward variance of different rollouts sampled from a given prompt, concentrate sharply on action tokens rather than reasoning tokens, even though action tokens account for only a small fraction of the trajectory. We refer to this phenomenon as the Action Bottleneck. Motivated by this observation, we propose an embarrassingly simple token reweighting approach, ActFocus, that downweights gradients on reasoning tokens, along with an additional energy-based redistribution mechanism that further increases the weights on action tokens with higher uncertainty. Across four environments and different model sizes, ActFocus consistently outperforms PPO and GRPO, yielding final-step gains of up to 65.2 and 63.7 percentage points, respectively, without any additional runtime or memory cost.

强化学习大模型智能体动作优化

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