跳到正文
arXiv cs.AI· Xiao Feng, Bo Han, Zhanke Zhou, Jiaqi Fan, Jiangchao Yao, Ka Ho Li, Dahai Yu, Michael Kwok-Po Ng·· 1 天前AI 评分35

RewardFlow:能否从状态图拓扑估计智能体强化学习中的动作级奖励?

RewardFlow: Can Action-Level Reward Be Estimated from State-Graph Topology in Agentic RL?

AI 导读

RewardFlow 通过共享状态把同一任务的多条轨迹连成有向状态图,从成功状态反向传播奖励信号,按最短路径距离分配状态势能,无需学习过程奖励模型或人工动作级标注即可估计动作级优势。

正文 · AI 翻译

查看 PDF

摘要:在家庭助理和网页购物等智能体任务中,大型语言模型(LLM)智能体在收到指示成功或失败的最终结果之前会执行多个动作。对于强化学习(RL)而言,这一结果掩盖了各个动作的贡献:一条购物轨迹可能找到了正确的商品,却未能完成购买。为解决这一信用分配问题,我们提出了 RewardFlow,它通过共享状态(按任务相关等价性或语义相似性分组的状态)将同一任务的多条轨迹连接起来。由此得到的有向状态图连接了观测到的轨迹片段。因此,失败轨迹中的一个动作可能到达某个状态,而从该状态出发另一条轨迹能够成功。它从任务成功的状态向后传播奖励信号,分配状态势值,这些势值随图中到成功状态的最短路径距离而递减。随后,势值差异会奖励那些缩短该距离的动作,使智能体在图中更接近成功。这些估计的奖励产生动作级优势,在策略优化过程中与来自结果奖励的轨迹级优势互补,且无需学习过程奖励模型或人工动作级标注。在 ALFWorld、WebShop、Sokoban 和搜索任务上的实验表明,在不同模型规模下均优于 RL 基线。在使用 3B 策略时,在 ALFWorld、WebShop 和 Sokoban 上,图构建和奖励估计每个训练步骤耗时 $1.47$-$2.39$ 秒(图 1)。代码可在 this https URL 获取。
评论: 被 NeurIPS 2026 接收
主题: 人工智能(cs.AI);计算与语言(cs.CL);机器学习(cs.LG)
引用为: arXiv:2603.18859 [cs.AI]
  (或 arXiv:2603.18859v3 [cs.AI] 用于此版本)
  https://doi.org/10.48550/arXiv.2603.18859

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Xiao Feng [查看邮箱]
[v1] 2026年3月19日星期四 13:07:12 UTC(1,915 KB)
[v2] 2026年5月28日星期四 06:59:57 UTC(1,737 KB)
[v3] 2026年11月12日星期四 03:57:23 UTC(1,446 KB)

来源:arXiv cs.AI · arxiv.org