KAIST 与 DeepMind 提出声明式注意力,让模型自己决定读哪些上下文

coallaoh · x · 2026-10-18

KAIST AI 与 Google DeepMind 研究者(一作 Namgyu Ho、Tal Schuster 等)发布 Declarative Attention(DA),一种让语言模型「自己声明注意力范围」的稀疏注意力协议。

核心思路:不再用外部代理分数预测哪些 token 重要(每步仍是 O(N)),而是让模型在思维链中以纯文本标记三种模式——<global>(读全部上下文)、<focus>(只读指定区域)、<local>(只看最近输出),推理引擎像解析工具调用一样据此屏蔽大部分 KV cache 读取。

实验结果(15 个长上下文任务、zero-shot、离线模型 Gemma-4 31B 与 Qwen-3.6 27B):

作者还展望了在 agent 场景配合专门后训练的潜力,以及模型可控的 KV cache offloading 作为「可逆上下文压缩」的路径。代码已开源。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →