VFold: Symmetry-Aware Cross-Layer Value Cache Compression
VFold:对称性感知的跨层 Value Cache 压缩
🧭 LLM 推理 KV cache 压缩
Johns Hopkins University · George Mason University
利用 value cache 跨层对称性做无损合并,不改模型结构,可与量化/剪枝组合进一步压缩
📌 工作参考价值(4 点)
- 适合长上下文 Agent/RAG/LLM 排序服务:在不改模型结构、不重新训练的前提下,于 decode 前对 value cache 做跨层对称合并,可直接降低显存占用,线上可插拔。
- 可与 KV cache 量化或 key cache pruning 组合:先做 VFold 式 value 合并,再做高压缩比量化/剪枝,能在 memory 受限条件下换更大 batch 或更长上下文;需用 LongBench/RULER/多轮工具调用等业务指标回归。
- 工程实现上只需读取 value cache 并做轻量融合/索引,对注意力结构无侵入,适合已经冻结模型参数的线上推理框架(如 vLLM、TensorRT-LLM)外挂使用。
- 可离线检测各层 value 表示的余弦相似度/CKA,定位高冗余层,再决定合并范围;若用 LLM 做序列推荐或用户行为编码,类似跨层冗余也可能存在,值得做一次成本极低的探测。