跳到正文
arXiv · cs.AI · Zhiqi Li, Bo Zhu · 8 小时前模型评分39

dKV Forcing:用可微 KV 缓存分离生成与记忆表示

dKV Forcing: Separating Generation and Memory Representations through a Differentiable KV Cache

导读

研究提出 dKV Forcing(dKV 指可微 KV),在共享主干上为写入方加低秩残差,把记忆写入与生成分开,并通过未来读取反传梯度训练。该方法用持久可微页替代训练时缓存,一次遍历即可算出完整记忆梯度,相比两遍重建减少 28% 常驻内存、缩短 20.8% 训练周期,生成器更新开销仅 7%。

生成六维分析
分类置信度

中 · 待人工复核

  • 只有单一媒体来源,需要人工复核。

来源级别:媒体与个人

事件类型:研究发布

证据摘录

  • “We propose dKV Forcing (dKV denotes differentiable KV) to learn a dedicated memory representation from its future use”

来源:arXiv · cs.AI · arxiv.org