研究人员开发了一种名为增减KV缓存(WnW)的新颖方法,以优化用于长篇语音处理的大语言模型的内存使用。该技术将KV缓存头分为锚点、潮汐和固定角色,允许锚点头保留在GPU上以进行实时重要性监控。潮汐头利用CPU驻留存储,并根据锚点头分数选择性地调用,而固定头则保留较小的GPU子集。在3B参数模型上使用LibriSpeech-Long数据集进行的实验表明,WnW在仅使用音频令牌通常所需的GPU内存的20%的情况下,保持了接近全缓存的准确性,显著优于仅预填充的压缩方法。 AI
影响 这种WnW KV缓存方法可以显著降低处理长音频输入的LLM的计算成本,从而实现更高效、更易于访问的语音AI应用。
排序理由 该集群包含一篇详细介绍LLM优化新方法的 ist 研究论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →