arXiv:2510.07151cs.LGcs.AI2025-10被引 4

为长时序强化学习设计可更新重写的外部记忆模块

ELMUR: External Layer Memory with Update/Rewrite for Long-Horizon RL Problems

  • 在每一层引入结构化外部记忆,通过双向交叉注意力交互
  • 在百万步迷宫任务中实现100%成功率,比基线提升70%整体表现
  • 适合需要长期记忆的机器人控制与稀疏奖励场景

现实世界中的机器人代理需在部分可观测和长时域环境下决策,关键线索可能在很久前出现却影响后续判断。现有方法多依赖即时信息,缺乏对历史经验的利用。标准循环或Transformer模型难以保持和利用长程依赖:上下文窗口限制历史长度,而简单的记忆扩展在规模和稀疏性下失效。我们提出ELMUR(带更新/重写的外部层内存),一种具有结构化外部记忆的Transformer架构。每层维护记忆嵌入,通过双向交叉注意力与之交互,并利用最近最少使用(LRU)记忆模块进行替换或凸混合更新。ELMUR将有效时域扩展至注意力窗口的10万倍,在合成T-Maze任务中,长达百万步的通道任务达到100%成功率。在POPGym上,超过半数任务优于基线;在MIKASA-Robo稀疏奖励操作任务中,性能接近翻倍,23项任务中有21项取得最佳成功率,所有任务平均成功率提升约70%。结果表明,结构化的层内外部记忆为部分可观测下的决策提供了一种简单且可扩展的方案。代码与项目页:https://elmur-paper.github.io/

原文摘要 · Abstract (English)

Real-world robotic agents must act under partial observability and long horizons, where key cues may appear long before they affect decision making. However, most modern approaches rely solely on instantaneous information, without incorporating insights from the past. Standard recurrent or transformer models struggle with retaining and leveraging long-term dependencies: context windows truncate history, while naive memory extensions fail under scale and sparsity. We propose ELMUR (External Layer Memory with Update/Rewrite), a transformer architecture with structured external memory. Each layer maintains memory embeddings, interacts with them via bidirectional cross-attention, and updates them through an Least Recently Used (LRU) memory module using replacement or convex blending. ELMUR extends effective horizons up to 100,000 times beyond the attention window and achieves a 100% success rate on a synthetic T-Maze task with corridors up to one million steps. In POPGym, it outperforms baselines on more than half of the tasks. On MIKASA-Robo sparse-reward manipulation tasks with visual observations, it nearly doubles the performance of strong baselines, achieving the best success rate on 21 out of 23 tasks and improving the aggregate success rate across all tasks by about 70% over the previous best baseline. These results demonstrate that structured, layer-local external memory offers a simple and scalable approach to decision making under partial observability. Code and project page: https://elmur-paper.github.io/.

强化学习长时序决策外部记忆机器人控制

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