LSTM 式共享缓存 Transformer:递归推理下的 KV 缓存匹配实验

BlackHC · x · 2026-10-06

前 DeepMind 研究员 BlackHC 的一项小型研究项目:构建一个 LSTM 风格的绑定(tied)Transformer,全模型只共享一份核心 KV 缓存,核心问题是如何针对模型在推理时递归产生的缓存历史进行训练。工作建立在 LCKV(末层缓存+KV 匹配)、CLA(跨层复用)和 Huginn(递归深度)之上,并用 LSTM 单元测试缓存历史失配与修复。

所属事件:BlackHC 离职首作:共享 KV cache 的绑定 Transformer(8 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →