KV Cache Offloading for Context-Intensive Tasks
AI 摘要
研究KV缓存卸载在上下文密集型任务中的性能下降问题,并提出改进方案。
主要贡献
- 发现KV缓存卸载在上下文密集型任务中性能下降
- 创建Text2JSON基准测试,用于评估上下文密集型任务
- 分析性能下降的原因并提出改进策略
方法论
通过构建Text2JSON基准测试,并在Llama 3和Qwen 3模型上评估KV缓存卸载策略的性能,分析原因并提出改进方案。
原文摘要
With the growing demand for long-context LLMs across a wide range of applications, the key-value (KV) cache has become a critical bottleneck for both latency and memory usage. Recently, KV-cache offloading has emerged as a promising approach to reduce memory footprint and inference latency while preserving accuracy. Prior evaluations have largely focused on tasks that do not require extracting large amounts of information from the context. In this work, we study KV-cache offloading on context-intensive tasks: problems where the solution requires looking up a lot of information from the input prompt. We create and release the Text2JSON benchmark, a highly context-intensive task that requires extracting structured knowledge from raw text. We evaluate modern KV offloading on Text2JSON and other context-intensive tasks and find significant performance degradation on both Llama 3 and Qwen 3 models. Our analysis identifies two key reasons for poor accuracy: low-rank projection of keys and unreliable landmarks, and proposes a simpler alternative strategy that significantly improves accuracy across multiple LLM families and benchmarks. These findings highlight the need for a comprehensive and rigorous evaluation of long-context compression techniques.