arXiv:2506.00539cs.CL2025-06NeurIPS被引 7

通过意图空间聚合奖励,解决语言智能体的稀疏奖励问题

ARIA: Training Language Agents with Intention-Driven Reward Aggregation

  • 将自然语言动作映射到低维意图空间,聚类相似动作并共享奖励
  • 在4个下游任务上平均提升9.95%性能,显著降低策略梯度方差
  • 适合需要复杂语言决策的开放环境智能体训练,如谈判与问答

大语言模型使智能体能够通过自由语言交互实现复杂推理与决策。然而,在开放语言动作环境(如谈判或提问游戏)中,动作空间可建模为词元的联合分布,导致动作空间呈指数级增长。在此空间采样动作易引发极端奖励稀疏性,造成巨大奖励方差,阻碍有效强化学习。为此,我们提出ARIA,一种在意图空间聚合奖励的方法,以实现高效且有效的语言智能体训练。ARIA旨在将高维联合词元分布空间中的自然语言动作投影至低维意图空间,使语义相似的动作被聚类并共享奖励。这种意图感知的奖励聚合通过密集化奖励信号降低奖励方差,促进更优策略优化。大量实验表明,ARIA不仅显著降低策略梯度方差,还在四个下游任务上平均实现9.95%的性能提升,持续优于离线与在线强化学习基线。

原文摘要 · Abstract (English)

Large language models (LLMs) have enabled agents to perform complex reasoning and decision-making through free-form language interactions. However, in open-ended language action environments (e.g., negotiation or question-asking games), the action space can be formulated as a joint distribution over tokens, resulting in an exponentially large action space. Sampling actions in such a space can lead to extreme reward sparsity, which brings large reward variance, hindering effective reinforcement learning (RL). To address this, we propose ARIA, a method that Aggregates Rewards in Intention space to enable efficient and effective language Agents training. ARIA aims to project natural language actions from the high-dimensional joint token distribution space into a low-dimensional intention space, where semantically similar actions are clustered and assigned shared rewards. This intention-aware reward aggregation reduces reward variance by densifying reward signals, fostering better policy optimization. Extensive experiments demonstrate that ARIA not only significantly reduces policy gradient variance, but also delivers substantial performance gains of an average of 9.95% across four downstream tasks, consistently outperforming offline and online RL baselines.

语言智能体强化学习奖励聚合意图空间

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