← 全部深度解读
📖 深度解读 · INFRA

RACE: RoPE-Aligned Attention Compression Based on Query-Key Activation Energies

Irene Tenison, Abdulrahman Alabdulkareem, Daniel Karl I. Weidele, Nima Dehmamy, Dan Gutfreund · arXiv 2610.28385 · 发表 2026-10-20
3 分钟阅读时长 844字 4.30综合分 第 3 名10-25 Infra 日报
目录 · 6 节1问题2方法3核心贡献4实验5局限6实践意义
相关 5新颖 4实用 4证据 4时效 4

头维压缩 KV cache 常在推理时再还原全维,解码 FLOPs 反而涨。RACE 用校准集把预训练注意力改成原生降维:对 Q/K 共享投影并保持 RoPE 对齐,按 query-key 激活能量决定保留哪些能量方向,使 KV 存储与注意力算力一起降。50% KV 保留、256K 上下文时,总推理 FLOPs −47.8%,吞吐 1.79×,TTFT −43.2%,分配显存 −28.1%。相对 EigenAttention / PALU / Thin Keys / RAP,RACE 在「保留 RoPE、共享投影、降存储、降注意力 FLOPs、免推理还原」清单上全部勾选。头维压缩要原生降维,而不是推理时再拼回全维。

1. 问题

KV 变长拖垮显存与带宽;只压 token 或量化不够时,有人压头维。若解码期重建全维 K/V,省下的显存会换成 FLOPs 税。需要校准期一次性改成降维注意力。

全维 vs 压缩注意力(论文图)。
全维 vs 压缩注意力(论文图)。

2. 方法

共享投影把 Q/K 映到同一低维空间,旋转位置编码在低维上仍对齐;用激活能量选主方向,V 同步压缩。校准集约一次;推理不再上投影回全维。可与 token 减少、量化叠加(文中有组合实验)。

3. 核心贡献

把「压缩」从缓存格式问题提升为注意力算子本身的降维,且不破坏 RoPE 几何。省显存的同一套投影也应省注意力算力。

4. 实验

Llama/Qwen 系列在零样本、困惑度、LongBench 上对比。50% 保留 + 256K:FLOPs −47.8%,吞吐 1.79×,TTFT −43.2%,显存 −28.1%。对照表显示竞品常在「推理重建」或「不保 RoPE」上留缺口。

设置结果
KV 保留50%
上下文256K
总推理 FLOPs−47.8%
吞吐1.79×
TTFT−43.2%
GPU 显存−28.1%
精度曲线。
精度曲线。
FLOPs/吞吐。
FLOPs/吞吐。

5. 局限

校准集域偏移会影响能量方向;与激进量化叠用时要重扫保留比例;超长上下文的端到端速度还受别的算子影响。

6. 实践意义

服务侧若 KV 显存紧,优先试「原生降维注意力」而非「压完再还原」;保留 50% 作为默认扫描点,看 TTFT 与 LongBench 是否可接受。

← 全部深度解读 出现于报告:2026-10-25 · Infra #3 (4.30) 原文 ↗