研究人员开发了 Nexus Sampling,一种用于大型语言模型 KV 缓存驱逐的新颖方法,特别适用于长上下文和代理工作负载。这种无需训练的方法将 Nexus 评分与加权水库采样相结合,以保留可能因确定性 top-K 选择而丢失的重要 token。Nexus Sampling 在保留细微重要 token 方面理论上优于传统方法,并在 LongBench 等基准测试中实证达到了与密集注意力相当的性能,同时显著减少了缓存内存使用。 AI
影响 该方法可以显著减小 LLM 的内存占用,从而实现更复杂和更长上下文的应用。
排序理由 该集群包含一篇详细介绍 LLM 推理优化新方法的学术论文。
- arXiv
- Hugging Face
- KV cache
- LongBench: a bilingual, multitask benchmark for long context understanding
- Nexus Sampling
- Nexus scoring
- Weighted Reservoir Sampling from Distributed Streams
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →