跳到正文
原文
arXiv · Machine Learning Theory· Jiawei Lin, Saibo Geng, Thomas Bourgeat·· 5 小时前AI 评分42

EchoPress:通过虚拟上下文重建实现查询无关的 KV Cache 剪枝

EchoPress: Query-Agnostic KV Cache Pruning via Virtual Context Reconstruction

AI 导读

EchoPress 是一种无需训练的 KV cache 剪枝方法,通过标准 prefill 阶段的 queries 和 keys 近似重建注意力,仅对第一个 chunk 重建以校准剩余上下文的重要性分数。

来源:arXiv · Machine Learning Theory · arxiv.org