让机器人智能决定何时记忆、记什么,提升复杂任务适应力。
Gated Memory Policy: In-Context Memorization and Adaptation
- 用可学习的门控机制控制记忆调用时机,只在必要时激活历史信息。
- 引入轻量交叉注意力构建高效记忆表征,在非马尔可夫任务上提升30.1%成功率。
- 通过扩散噪声注入增强对错误历史的鲁棒性,适合真实场景部署。
机器人操作任务的记忆需求差异大,从无需记忆的马尔可夫任务到需在单次试验中记忆历史或基于多次试验结果进行上下文适应的非马尔可夫任务不等。令人惊讶的是,简单延长视觉动作策略的观察历史常导致性能显著下降,源于分布偏移和过拟合。为此,我们提出门控记忆策略(Gated Memory Policy, GMP),一种能学习何时调用记忆及调用内容的视觉动作策略。为学习调用时机,GMP采用可学习的记忆门机制,仅在必要时激活历史上下文,提升鲁棒性与响应速度;为高效学习调用内容,GMP引入轻量级交叉注意力模块,构建有效的潜在记忆表示;为进一步增强鲁棒性,GMP在历史动作中注入扩散噪声,缓解训练与推理阶段对噪声或错误历史的敏感性。在我们提出的非马尔可夫基准测试集MemMimic上,GMP相比长历史基线平均成功率达提升30.1%,同时在RoboMimic的马尔可夫任务上保持竞争力。所有代码、数据及野外部署说明均公开于项目网站https://gated-memory-policy.github.io/。
原文摘要 · Abstract (English)
Robotic manipulation tasks exhibit varying memory requirements, ranging from Markovian tasks that require no memory to non-Markovian tasks that demand in-context memorization of historical information within a single trial or in-context adaptation based on the outcomes of multiple past trials. Surprisingly, simply extending observation histories of a visuomotor policy often leads to a significant performance drop due to distribution shift and overfitting. To address these issues, we propose Gated Memory Policy (GMP), a visuomotor policy that learns both when to recall memory and what to recall. To learn when to recall memory, GMP employs a learned memory gate mechanism that selectively activates history context only when necessary, improving robustness and reactivity. To learn what to recall efficiently, GMP introduces a lightweight cross-attention module that constructs effective latent memory representations. To further enhance robustness, GMP injects diffusion noise into historical actions, mitigating sensitivity to noisy or inaccurate histories during both training and inference. On our proposed non-Markovian benchmark MemMimic, GMP achieves a 30.1% average success rate improvement over long-history baselines, while maintaining competitive performance on Markovian tasks in RoboMimic. All code, data and in-the-wild deployment instructions are available on our project website https://gated-memory-policy.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。