arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.23592cs.ROcs.AIcs.SE
收藏

KEEP: 一种面向高效具身规划的KV缓存中心化内存管理系统

KEEP: A KV-Cache-Centric Memory Management System for Efficient Embodied Planning

Institute for Artificial Intelligence(人工智能研究院);School of Integrated Circuits, Peking University(集成电路学院,北京大学);Shenzhen Institute of Artificial Intelligence(深圳人工智能研究院);Microsoft Research(微软研究院);Beijing Advanced Innovation Center for Integrated Circuits(北京集成电路先进创新中心)

Zebin Yang, Tong Xie, Baotong Lu, Shaoshan Liu, Bo Yu, Meng Li

AI总结:

本文提出KEEP系统,通过静态动态内存构建算法、多跳内存重计算算法和层平衡内存加载,提升具身规划效率,实验表明在ALFRED数据集上速度提升2.68倍,优于CacheBlend方法。

摘要:

记忆增强型大语言模型(LLMs)在复杂和长视距具身规划中表现出色。通过跟踪过往经验与环境状态,内存使LLMs保持全局视角,从而避免重复探索。然而,现有方法常将内存存储为原始文本,导致提示过长和预填充延迟高。虽然可以存储和重用KV缓存,但频繁的KV缓存更新严重削弱了效率。本文提出KEEP,一种面向高效具身规划的KV缓存中心化内存管理系统。KEEP有三个关键创新:(1)静态动态内存构建算法通过混合粒度内存组减少KV缓存重新计算;(2)多跳内存重计算算法动态识别不同内存组间的跨注意力,并迭代重建内存交互;(3)层平衡内存加载消除不同层间的KV缓存加载和跨注意力计算不平衡。大量实验结果表明,与基于文本的内存方法相比,KEEP在ALFRED数据集上实现了2.68倍的速度提升,精度损失可忽略不计。与KV重计算方法CacheBlend(EuroSys'25)相比,KEEP在成功率上提高了4.13%,时间到第一个token(TTFT)减少了1.90倍。我们的代码可在https://github.com/PKU-SEC-Lab/KEEP_Embodied_Memory上获得。

英文摘要:

Memory-augmented Large Language Models (LLMs) have demonstrated remarkable capability for complex and long-horizon embodied planning. By keeping track of past experiences and environmental states, memory enables LLMs to maintain a global view, thereby avoiding repetitive exploration. However, existing approaches often store the memory as raw text, leading to excessively long prompts and high prefill latency. While it is possible to store and reuse the KV caches, the efficiency benefits are greatly undermined due to frequent KV cache updates. In this paper, we propose KEEP, a KV-cache-centric memory management system for efficient embodied planning. KEEP features 3 key innovations: (1) a Static-Dynamic Memory Construction algorithm that reduces KV cache recomputation by mixed-granularity memory group; (2) a Multi-hop Memory Re-computation algorithm that dynamically identifies important cross-attention among different memory groups and reconstructs memory interactions iteratively; (3) a Layer-balanced Memory Loading that eliminates unbalanced KV cache loading and cross-attention computation across different layers. Extensive experimental results have demonstrated that KEEP achieves 2.68x speedup with negligible accuracy loss compared with text-based memory methods on ALFRED dataset. Compared with the KV re-computation method CacheBlend (EuroSys'25), KEEP shows 4.13% success rate improvement and 1.90x time-to-first-token (TTFT) reduction. Our code is available on https://github.com/PKU-SEC-Lab/KEEP_Embodied_Memory.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑