斯坦福团队推出 TokenGate:学会略读上下文,prefill 最快提速 11 倍

CShorten30 · x · 2026-10-21

Stanford 与 UC Berkeley(Liana Patel、Ion Stoica、Matei Zaharia、Carlos Guestrin 等)发布 Token-Gated Language Models 研究:训练模型学会「略读」上下文,在浅层后仅条件性前传重要 token,实现新维度上的条件计算——类比从稠密模型到 MoE 的转变,只是稀疏化对象从参数换成了上下文 token。

关键结果(微调 Qwen3-1.7B,精度对齐标准 prefill):

方法上在 Layer L 处用 Gate Module 保留 K 个 token,上层以 Token-Gated Attention 剪枝前传。代码开源于 github.com/token-gate-llms/TokenGate。

所属事件:斯坦福等推出 TokenGate,长上下文 prefill 最高提速 11 倍(3 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →