让机器人用6.4万步经验快速适应新环境,无需专家示范
ReLIC: A Recipe for 64k Steps of In-Context Reinforcement Learning for Embodied AI
- 采用局部更新策略与Sink-KV机制,高效利用长期观察记忆
- 在未知房屋中完成多物体导航任务,性能超越多种元强化学习方法
- 仅靠自生成数据即可实现少样本模仿学习,适合具身智能研究者
具身智能体需通过整合长期经验快速适应新场景。例如,机器人进入陌生房屋时,初始阶段无法掌握物品位置,表现低效;随着经验积累,应能学习环境布局并记住物品位置,从而更高效完成新任务。为此,本文提出ReLIC,一种面向具身智能体的上下文强化学习方法。该方法使智能体在64,000步上下文经验下,通过全注意力机制实现快速适应,且训练过程基于自生成经验的强化学习。核心创新包括:提出一种新型在线强化学习策略“局部更新”(partial updates),以及支持长历史记录有效利用的Sink-KV机制。实验表明,ReLIC在未见过的房屋环境中执行多物体导航任务时,显著优于多种元强化学习基线模型。此外,尽管未使用专家演示,其仍具备少样本模仿学习能力。全面分析揭示:大规模强化学习训练、局部更新机制与Sink-KV的结合是实现有效上下文学习的关键。代码及全部实验可在https://github.com/aielawady/relic获取。
原文摘要 · Abstract (English)
Intelligent embodied agents need to quickly adapt to new scenarios by integrating long histories of experience into decision-making. For instance, a robot in an unfamiliar house initially wouldn't know the locations of objects needed for tasks and might perform inefficiently. However, as it gathers more experience, it should learn the layout of its environment and remember where objects are, allowing it to complete new tasks more efficiently. To enable such rapid adaptation to new tasks, we present ReLIC, a new approach for in-context reinforcement learning (RL) for embodied agents. With ReLIC, agents are capable of adapting to new environments using 64,000 steps of in-context experience with full attention while being trained through self-generated experience via RL. We achieve this by proposing a novel policy update scheme for on-policy RL called "partial updates'' as well as a Sink-KV mechanism that enables effective utilization of a long observation history for embodied agents. Our method outperforms a variety of meta-RL baselines in adapting to unseen houses in an embodied multi-object navigation task. In addition, we find that ReLIC is capable of few-shot imitation learning despite never being trained with expert demonstrations. We also provide a comprehensive analysis of ReLIC, highlighting that the combination of large-scale RL training, the proposed partial updates scheme, and the Sink-KV are essential for effective in-context learning. The code for ReLIC and all our experiments is at https://github.com/aielawady/relic
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。