Zyphra 提出分层注意力 ZHA,实现亚二次复杂度稀疏注意力
Zyphra 提出分层注意力 ZHA,通过构建并搜索 KV cache 的压缩树为每个查询挑选相关 token,在合适 top-p 阈值下实现亚二次复杂度。ZHA 只需用几十亿 token 的 KL 蒸馏即可把已有预训练模型适配过来,在 Qwen3.6-35B-A3B 上仅新增 81,920 个可训练参数;RULER-13 256K 上宽树达 96.2%、深树 93.3%,LongBench v2 深树 63.4% 高于稠密注意力的 58.0%,SWE-bench Verified 上与稠密基线相当。其 GPU 性能模型预测在约 48K token 时快于稠密注意力、约 480K token 时快于 MSA,一千万 token 上下文下相对稠密注意力约 32–35 倍加速。

A),这是一种稀疏注意力方法,它构建并搜索一棵分层树,以确定每个查询的相关 token 集合。ZHA 通过为路由选择合适的 top-p 阈值,实现了亚二次复杂度,优于现有那些仍维持二次索引器的稀疏注意力方法。我们证明,只需使用几十亿 token 的蒸馏阶段,就能将现有的预训练模型适配到 ZHA,并且 ZHA 在高稀疏度下仍能在具有挑战性的长上下文推理、智能体编码和检索基准上保持性能。我们的性能模型显示,在超长上下文中,ZHA 在速度和计算成本上大幅优于稀疏注意力方法。 为什么高效注意力如此困难? Transformer 最显著的特性之一,是能够从序列中的任意位置检索信息。 当语言模型生成下一个 token 时,注意力使其能够回顾此前出现的所有内容,并判断哪些信息片段是相关的。例如,编码模型可能需要从数千行之前检索某个函数定义,而推理模型可能需要重新审视长思维链中的某个中间结论。 关键在于,模型不必事先决定哪些信息会重要。每个新查询都可以独立检索它所需要的那部分上下文。 这种灵活性极其强大,但代价也很高昂。 在标准注意力中,每个 token 都要与之前的所有 token 进行比较。这导致
匹配字段正文


