arXiv:2602.03560cs.CLcs.AI2026-02被引 8

HySparse通过混合稀疏与全连接注意力,大幅降低显存占用并提升模型性能。

HySparse: A Hybrid Sparse Attention Architecture with Oracle Token Selection and KV Cache Sharing

  • 用全连接层作为精确的令牌选择器,指导稀疏层关注关键信息。
  • 在80B MoE模型中仅用5层全连接注意力,显存减少近10倍且性能更优。
  • 支持缓存复用,兼顾计算效率与内存节省,适合大模型部署。

本文提出混合稀疏注意力(HySparse),将全连接注意力层与多个稀疏注意力层交替设计。该架构巧妙利用前序全连接层的输出,直接确定稀疏层的令牌选择和键值缓存。相比传统方法依赖额外代理预测重要性,HySparse以全连接层为精确‘预言机’,避免复杂性和次优表现;同时,其稀疏层可复用全连接层的键值缓存,实现计算与内存双重优化。我们在7B稠密模型和80B MoE模型上评估,结果表明:在所有设置下,HySparse均优于全连接注意力和混合SWA基线。尤其在含49层的80B MoE模型中,仅5层使用全连接注意力,仍实现显著性能提升,且键值缓存存储量减少近10倍。

原文摘要 · Abstract (English)

This work introduces Hybrid Sparse Attention (HySparse), a new architecture that interleaves each full attention layer with several sparse attention layers. While conceptually simple, HySparse strategically derives each sparse layer's token selection and KV caches directly from the preceding full attention layer. This architecture resolves two fundamental limitations of prior sparse attention methods. First, conventional approaches typically rely on additional proxies to predict token importance, introducing extra complexity and potentially suboptimal performance. In contrast, HySparse uses the full attention layer as a precise oracle to identify important tokens. Second, existing sparse attention designs often reduce computation without saving KV cache. HySparse enables sparse attention layers to reuse the full attention KV cache, thereby reducing both computation and memory. We evaluate HySparse on both 7B dense and 80B MoE models. Across all settings, HySparse consistently outperforms both full attention and hybrid SWA baselines. Notably, in the 80B MoE model with 49 total layers, only 5 layers employ full attention, yet HySparse achieves substantial performance gains while reducing KV cache storage by nearly 10x.

稀疏注意力模型压缩缓存复用大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。