跳到正文
Hugging Face Daily Papers·· 3 天前精选AI 评分54

JSSA:面向长上下文解码的联合子空间稀疏注意力

Joint Subspace Sparse Attention for Efficient Long-Context Decoding

AI 导读

论文提出联合子空间稀疏注意力(JSSA),通过学习共享低维 query-key 子空间提升 token 级选择质量,并设计免 gather 的 token 级索引注意力 kernel,支持 GQA KV-cache 布局下每个 KV head 独立选择。

推荐理由

论文给出免训练的稀疏注意力方案,读者可了解长上下文解码吞吐与精度如何兼顾。

来源:Hugging Face Daily Papers · huggingface.co