Stanford 与 Berkeley 发布 TokenGate:长上下文预填充最高提速 11.3 倍

MundanePercentage674 · reddit · 2026-10-26

Stanford 与 UC Berkeley 研究者发布 TokenGate(arXiv 2610.26754),一种面向长上下文预填充的学习型 token 剪枝方法,思路是让 LLM 学会「略读」。

机制:低层(L 层以下)处理全部 N 个 token;学习到的门控只保留 top-K 文档 token(问题 token 永不剪枝);上层仅处理这 K 个 token。

结果:首 token 时间(TTFT)最高快 11.3 倍,峰值预填充内存降低 3.4 倍,比次优剪枝器快 6.4 倍。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →