FlashMemory-DeepSeek-V4: Lightning Index Ultra-Long Context via Lookahead Sparse Attention
常规 LLM 解码时把完整 KV cache 常驻显存,超长上下文 serving 时显存成为硬瓶颈。论文提出 Lookahead Sparse Attention (LSA):基于 DeepSeek-V4 架构构建一个 Neural Memory Indexer,主动预测未来会用到的上下文、只把 query-critical 的 KV chunk 保留在显存。关键是 backbone-free 解耦训练——把 indexer 建成标准 dual-encoder,用常规检索训练框架独立训练,全程不把巨大的 backbone 载入显存。最终把平均物理 KV cache 占用压到全量基线的 13.5%,在 500K 上下文下 KV 抑制超过 90%,同时在 LongBench-v2、LongMemEval、RULER 上平均精度还 +0.6%。
INSIGHT把'该看哪些历史 token'建模成一个可独立训练的检索问题,是这篇的关键拆解:indexer 不依赖 backbone 前向、用 dual-encoder + 标准 retrieval loss 训练,训练成本和主模型彻底脱钩。对搜索团队的启示是 KV 稀疏化可以直接套用我们熟悉的召回/双塔范式来做,而不必改动主模型本身。13.5% 的 KV 占用、500K 下 >90% 抑制说明长上下文 serving 的显存账可以显著重算。局限是方案绑定 DeepSeek-V4 架构,迁移到其它模型需要重训 indexer。