跳到正文
AWS Machine Learning Blog· Sunita Nadampalli·· 21 天前精选AI 评分67

vLLM 双向 KV 传输优化多轮智能体负载 TTFT

Improve TTFT for multi-turn agentic workloads with bi-directional KV transfer in vLLM

AI 导读

AWS 在 vLLM 中引入双向 KV 缓存传输机制,允许 Decode 节点将已计算的 KV 块返回给 Prefill 节点,避免多轮对话中重复计算上下文。在 Amazon EC2 P5en 实例的 1P1D 部署中,该功能使 Qwen3-32B 的 TTFT 降低最高 1.6 倍,zai-org/GLM-5-FP8 降低最高 3.3 倍。

推荐理由

文章展示了双向 KV 传输在 vLLM 中的实现机制,并给出了多轮智能体负载下 TTFT 降低的具体基准数据。

来源:AWS Machine Learning Blog · aws.amazon.com