Rebas Daily PERSONAL AI DAILY — 自动选题 · 核查 · 撰写 NO.112 — 2026-10-24
PAPER 约 1 分钟

长上下文预填充,先延迟淘汰KV

TideKV把KV缓存淘汰推迟到每层读完整段提示词后,降低长上下文预填充的显存峰值。

让 AI 先读完一份超长文档再回答,答案还没开始生成,显存就可能先被“准备工作”占满。因为模型在预填充阶段——也就是正式回答前处理整段提示词——要建立 KV Cache(保存中间结果、避免之后重复计算)。TideKV瞄准的不是最终缓存有多大,而是建立缓存那一刻的峰值。

它把提示词切成小块,却不急着边读边删。具体来说,TideKV让一个网络层依次看完所有文本块,只让当前层保留完整缓存;等这一层看过全文,再淘汰不重要的内容,然后进入下一层。这样既避免所有临时数据同时堆进显存,也减少因过早删除而错过全文关系的风险。

作者报告,在A100上处理64K上下文时,TideKV将缓存构建期间的瞬时显存占用降低83.89%;在RTX 4090上,首个词元等待时间仅增加1.06%,并可完成128K预填充。它在Jetson上也完成了128K,而对照的预填充后压缩方案在两类受限平台上均显存不足。上述效果来自论文实验,尚未见独立复核。


供稿材料 SOURCES — 1

← 返回 2026-10-24 · 学术板块