论文

基于解释性世界模型的强化学习信用分配

Beyond Temporal Proximity: Explanatory World Models for Credit Assignment in Reinforcement Learning

精选理由

斯坦福团队新论文EWM用语言模型解释解决RL长期奖励分配问题,在多个基准测试中大幅提升性能。

研究人员提出解释性世界模型(EWM),用于解决强化学习中奖励信号衰减问题。该框架使用前沿语言模型生成解释,通过保守分层操作符递归重新分配奖励。在Crafter和WordCraft基准测试中,EWM显著优于PPO和LaRe算法,尤其在奖励延迟最大时提升最明显。在KuaiRand推荐基准和工业规模生成推荐系统中,EWM也展现出优越性能。

原文 · arXiv cs.AI

Beyond Temporal Proximity: Explanatory World Models for Credit Assignment in Reinforcement Learning

Generative Recommenders increasingly use Reinforcement Learning (RL) to optimize outcomes that may appear hundreds or thousands of interactions after a decision. Standard RL propagates reward according to temporal proximity rather than explicit relevance, attenuating the signal assigned to distant but important actions and creating a source of sample inefficiency. We introduce Explanatory World Models (EWM), an attribution framework that uses explanations to identify the earlier events that plausibly contributed to a realized outcome. We instantiate EWM with a frontier language model and use its explanations as learning signals using a conservative hierarchical operator that recursively redistributes reward while preserving the undiscounted episode return. To investigate what constitutes a useful explanation, we operationalize three complementary variants inspired by Halpern's actual causation, Lombrozo's explanatory simplicity, and Deutsch's hard-to-vary explanations. We evaluate EWM on two controlled long-horizon RL benchmarks, a public sequential-recommendation benchmark, and an industry-scale generative recommender. On Crafter and WordCraft, EWM substantially outperforms PPO and LaRe, with its largest gains when rewards are most delayed. On KuaiRand, EWM outperforms SFT and LaRe offline and A2C online. Finally, in an offline evaluation of an industry generative recommender, EWM considerably improves Recall and increases behavioral proxies. These results suggest that explanations can provide structured and reusable credit for learning from delayed outcomes.