为机器人任务规划设计双记忆系统,提升长期与短期记忆协同能力
KARMA: Augmenting Embodied AI Agents with Long-and-short Term Memory Systems
- 构建长时(3D场景图)与短时(物体状态变化)双记忆模块
- 在AI2-THOR中任务成功率提升1.3倍至2.3倍,效率提高3.4倍至62.7倍
- 可直接部署于真实机器人平台,适合复杂家务任务场景
执行连贯的长序列家庭任务的具身智能体常因上下文记忆不足导致效率低下和错误。为此,我们提出KARMA,一种融合长期与短期记忆模块的创新记忆系统,通过记忆增强提示提升大语言模型在具身代理中的规划能力。长期记忆以3D场景图形式记录环境全局信息,短期记忆动态追踪物体位置与状态变化。该双记忆结构使代理能检索过往场景经验,从而提升任务规划的准确性和效率。短期记忆采用自适应替换策略,确保关键信息保留而冗余数据被剔除。相较于现有具身智能体,本系统在AI2-THOR模拟器中,复合任务与复杂任务的成功率分别提升1.3倍和2.3倍,任务执行效率提升3.4倍和62.7倍。此外,KARMA具备即插即用特性,可无缝部署于移动操作平台等真实机器人系统。实验视频见https://youtu.be/4BT7fnw9ehs,代码开源于https://github.com/WZX0Swarm0Robotics/KARMA/tree/master。
原文摘要 · Abstract (English)
Embodied AI agents responsible for executing interconnected, long-sequence household tasks often face difficulties with in-context memory, leading to inefficiencies and errors in task execution. To address this issue, we introduce KARMA, an innovative memory system that integrates long-term and short-term memory modules, enhancing large language models (LLMs) for planning in embodied agents through memory-augmented prompting. KARMA distinguishes between long-term and short-term memory, with long-term memory capturing comprehensive 3D scene graphs as representations of the environment, while short-term memory dynamically records changes in objects' positions and states. This dual-memory structure allows agents to retrieve relevant past scene experiences, thereby improving the accuracy and efficiency of task planning. Short-term memory employs strategies for effective and adaptive memory replacement, ensuring the retention of critical information while discarding less pertinent data. Compared to state-of-the-art embodied agents enhanced with memory, our memory-augmented embodied AI agent improves success rates by 1.3x and 2.3x in Composite Tasks and Complex Tasks within the AI2-THOR simulator, respectively, and enhances task execution efficiency by 3.4x and 62.7x. Furthermore, we demonstrate that KARMA's plug-and-play capability allows for seamless deployment on real-world robotic systems, such as mobile manipulation platforms.Through this plug-and-play memory system, KARMA significantly enhances the ability of embodied agents to generate coherent and contextually appropriate plans, making the execution of complex household tasks more efficient. The experimental videos from the work can be found at https://youtu.be/4BT7fnw9ehs. Our code is available at https://github.com/WZX0Swarm0Robotics/KARMA/tree/master.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。