Declarative Attention 让 LLM 主动控制注意力,KV 缓存扫描量砍半

burny_tech · x · 2026-09-04

核心问题

长上下文 LLM 在每步解码时都要扫描整个 KV cache,即使真正相关的只是其中一小块区域,造成大量算力浪费。

方法:Declarative Attention

论文提出让模型在推理过程中显式切换注意力模式:全局、聚焦、局部三种状态,模型自己声明当前需要哪种注意力,推理引擎据此跳过无关的 KV cache 区域。

实验结果

以极小的精度代价换取显著推理效率提升,属于面向长上下文推理优化的新注意力控制机制。

所属事件:Declarative Attention:让语言模型自主声明读哪段 KV cache(5 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →