WUSH-KV: KV Cache Quantization with Data-Adaptive Transforms
WUSH-KV:数据自适应变换的 KV Cache 低比特量化
🧭 LLM 推理优化 · KV Cache 量化 ★ 深度解读
Institute of Science and Technology Austria (ISTA) · ETH Zürich · Red Hat AI
用 WUSH 数据自适应可逆变换对 KV cache 做低比特量化,2-bit 下 Qwen3-8B 全面优于 OSCAR
📌 工作参考价值(4 点)
- 对需要长上下文/高并发 LLM 服务的推荐与 Agent 系统,KV cache 内存是主要瓶颈。可借鉴将可逆变换与量化解耦:校准阶段按每个 head 的 Gram 矩阵(缓存张量自身统计)+ Hessian(下游消费者敏感度)构建变换,允许非正交各向异性缩放,比固定 Hadamard/随机旋转在 2-bit 下显著降低误差。类似思路可用于用户/物品 embedding 缓存或特征存储的低比特压缩,关键是 Hessian 选下游任务对扰动的敏感度。
- 值侧变换可在离线折叠进权重(V→WV、WO),在线零额外开销;键侧变换因 RoPE 无法折叠,放在 RoPE 之后,每新增 token 只多做一次 d×d 矩阵乘。工程上优先折叠 value 侧,key 侧用固定在线 transform,避免逐位置补偿。
- 保留 sink token 和最近 token 的 FP16 窗口(例如 sink=64, recent=256)并用批量量化 flush(chunk 8/16)可在大幅降低平均位宽时保护长上下文和生成质量;该策略与具体量化器无关,可直接迁移。
- 校准只需 128 条序列、一次完成,跨 bitwidth 复用;量化器可选 QuEST 或 OSCAR-style percentile-clipped affine,证明 WUSH 与 QuEST 搭配接近最优。在 SGLang 中集成时,可沿用 OSCAR 的 cache 管理策略,不必为 WUSH 重新调参。