研究人员推出了一种新颖的注意力机制 Kwai Summary Attention (KSA),旨在解决大型语言模型中标准 softmax 注意力的二次时间复杂度问题。KSA 旨在通过将历史上下文压缩成可学习的摘要 token 来维持 KV 缓存与序列长度之间的线性关系。这种方法试图在内存成本与有效保留长距离依赖性之间取得平衡,为现有方法(如减少 KV 缓存或使用对 KV 缓存友好的架构)提供了替代方案。 AI
影响 引入了一种新的注意力机制,以降低长上下文 LLM 的计算成本。
排序理由 介绍 LLM 新颖注意力机制的学术论文。
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →