将 LLM 的 KV 缓存卸载到 NVMe 驱动器仅在给定提示长度、模型大小和 GPU 的情况下,驱动器的持续带宽超过重新计算成本时才有利。一项对 Llama 3.2 3B 模型的研究表明,随着提示长度的增加,盈亏平衡带宽会显著降低,对于更长的上下文,更快的 NVMe 驱动器变得更具优势。由于其每 token 的 KV 字节增长速度慢于其预填充计算需求,因此更大的模型通常从卸载中受益更多。 AI
影响 优化 KV 缓存卸载可以提高大型语言模型(尤其是在更长上下文的情况下)的推理速度和效率。
排序理由 详细介绍 LLM 推理优化技术性能分析的技术论文。[lever_c_demoted from research: ic=1 ai=1.0]
- arXiv 2609.11744
- bfloat16
- Fio Linux Tool
- Llama 3.2:3b
- NVIDIA H100
- NVM Express
- PCI Express 5.0
- py-kvcache
- Qwen3-4B
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →