研究人员开发了一种名为联合子空间稀疏注意力(JSSA)的新方法,以提高大型语言模型长上下文解码的效率。JSSA通过学习共享的低维查询-键子空间以获得更好的排序来解决令牌级选择的挑战,并实现了一个无收集操作的内核以实现高效的令牌检索。该方法在保持高精度的同时显著提高了解码吞吐量,与在128K上下文下进行全键值(KV)注意力相比,在NVIDIA B200 GPU上实现了高达6.43倍的性能提升。 AI
影响 该方法可以显著降低运行具有长上下文的大型语言模型的计算成本,使其更易于访问和更高效。
排序理由 该集群描述了一篇关于改进LLM效率的新颖方法的最新研究论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →