跳到正文
SemiAnalysis· @SemiAnalysis_ · X·· 3 小时前AI 评分37
AI 导读

SemiAnalysis 指出稀疏注意力只缓解显存带宽瓶颈,不解决容量问题:稀疏计算仍需在上下文中保留并选中全部 token,最终还得加载进 HBM。HiCache 的思路是把 KV cache 从 HBM 卸载到更大容量、更低带宽的存储层,计算时只加载最相关的 token;HiSparse 则套用常规缓存规则,把最相关的部分加载回 HBM。

正文

Sparse attention fixes memory bandwidth, not memory capacity.

“The whole idea of HiCache is trying to offload KV cache from the HBM to a lower tier, higher capacity but lower bandwidth memory, and only load the most relevant set of tokens during computation.”

“Sparse attention doesn’t mitigate the memory capacity bottleneck. It’s a technique that helps a lot with memory bandwidth.”

“During the sparse attention computation, you still need to select the tokens. So you still have to keep all the tokens in context, and you cannot discard them. At some point, it still has to be loaded into HBM.”

“That’s where HiSparse becomes useful, because it can apply the typical caching rules so that you could load the most relevant parts into HBM.”

来源:SemiAnalysis · x.com