研究人员开发了联合子空间稀疏注意力(JSSA),这是一种提高长上下文语言模型解码效率的新方法。JSSA通过学习共享的低维查询-键子空间以获得更好的排序,并实现无收集操作(gather-free)的内核以高效检索令牌,从而解决了令牌级选择的挑战。该方法在保持与全KV注意力相当的准确性的同时,显著提高了解码吞吐量,在128K上下文的NVIDIA B200 GPU上实现了高达6.43倍的提升。 AI
影响 该方法可以显著降低LLM处理长上下文的计算成本,从而实现更高效和可扩展的应用。
排序理由 该条目描述了一篇详细介绍改进LLM效率的新颖方法的最新技术论文。[lever_c_demoted from research: ic=1 ai=1.0]
在 Hugging Face Daily Papers 阅读 →
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →