长上下文 LLM 服务的分层上下文缓存

原题:Strata: Hierarchical Context Caching for Long Context Language Model Serving

问题与动机

长上下文复用要求把 KV-Cache 扩展到 GPU、CPU、SSD,但各层碎片化布局会把命中转成大量小传输;若调度器只看是否命中,不看加载完成时间,GPU 仍会阻塞。

关键观察 / 隐含假设

  • host/SSD 适合连续大 I/O,GPU 计算布局则适合模型访问,二者不必相同。
  • KV 加载可以与别的 prefill/decode 工作重叠。
  • 假设请求存在足够前缀复用,且层级带宽可被批量传输利用。

核心方法

Strata 用 GPU-assisted I/O 解耦 host 与 GPU 布局,将碎片重排放到设备侧并形成大传输;cache-aware scheduler 同时考虑命中位置和可用时间,以机会工作隐藏加载停顿。实现集成到 SGLang

实验与结果

在长上下文 serving workload 上,Strata 相对 vLLM-LMCache 吞吐最高提升 5×,相对 NVIDIA TensorRT-LLM 最高提升 3.75×(§7,图 13)。在未优化层级缓存中,KV transfer 可阻塞 74% prefill 时间;边界是具有前缀复用且 CPU/SSD cache 可命中的请求。

论断—证据表

论断证据边界置信度
命中不等于及时可用基线中 74% prefill 时间被传输阻塞§2
布局解耦与调度协同能恢复吞吐相对 LMCache 最高 5×§7

批判性分析

论证链条

论文把层级缓存问题从容量命中重新表述为传输粒度和 ready-time,并分别给出数据面与调度面机制。

假设压力测试

低复用、随机上下文、SSD 抖动或 decode 主导负载会削弱收益。

实验可信度

多强基线和生产 SGLang 实现增强可信度,但不同 SSD/PCIe 拓扑下的收益区间仍需展开。

局限与后续工作

  • 可探索多租户 cache fairness、写入放大、故障恢复,以及跨节点远程 KV 层级。

相关