长上下文竞赛的瓶颈,正在从“能不能塞进去”转向“塞进去之后算不算得起”。Zyphra 给出的答案是:不改模型权重,只换注意力结构。

发生了什么

Zyphra 公布了名为 Zyphra Hierarchical Attention(ZHA)的方法,核心是一种树索引的稀疏注意力。它的做法不是从头训练一个新模型,而是对已冻结的 LLM 做改造:权重保持不动,通过约 4B token 的蒸馏,让模型学会在新的稀疏注意力路径下工作。

按官方说法,改造后的模型在推理类评测和 256K 长度的检索任务上,表现与原始稠密注意力相当。标题里“true subquadratic”的措辞,指向的是让注意力复杂度真正低于二次方,而不只是在某个长度区间里做近似。

为什么重要

稠密注意力的二次方成本,是长上下文落地最硬的一道墙。业界过去几年主要走三条路:一是改架构从头预训练(如各类线性注意力、状态空间模型),代价是丢掉现有生态;二是训练时的稀疏化,需要动预训练流程;三是推理侧的 KV cache 压缩、分块、检索式外挂,属于工程补丁,模型本身的能力边界没变。

ZHA 的位置比较特别——它把“架构升级”做成了对存量模型的后处理。冻结权重意味着不需要重跑预训练,也不需要重新对齐;4B token 的蒸馏量相对主流预训练规模是轻量的。如果这条路可复现,那么“把手上已有的开源模型升级成亚二次方注意力”就变成一个可批量执行的工序,而不是一次重新造模型的豪赌。

这也解释了它为什么值得关注:真正的价值不在某个 benchmark 分数,而在于架构演进与模型资产的解耦。

影响与看点

对开发者而言,最直接的意义是长上下文推理的显存与延迟账本可能被重算。如果稀疏注意力在 256K 级别仍能保持检索质量,长文档问答、代码库级理解、多轮长会话这类场景的部署门槛会下降。

对模型厂商,这构成一种新的竞争维度:不只是“我的模型多强”,还有“我的模型多容易被改造”。一个权重可被冻结改造的模型,生命周期会被拉长。

需要保持克制的几点:摘要只提到推理与 256K 检索两类评测,通用能力、指令跟随、多语言等维度是否同样无损,尚不清楚;4B token 蒸馏的具体数据构成与训练成本也未披露;稀疏注意力在真实服务中的吞吐收益,取决于 kernel 实现,而非论文层面的复杂度符号。

一句判断:ZHA 更像是一个方向性的证明——注意力架构可以像固件一样被刷进已有模型,而不是必须与权重一同诞生。这条路若被更多团队验证,长上下文的成本曲线会比预期更早下移。