DashAttention: Differentiable and Adaptive Sparse Hierarchical Attention
DashAttention:可微分自适应稀疏层级注意力
🧭 自适应稀疏层级注意力 · α-entmax 路由 ★ 深度解读
Tsinghua University · Instituto Superior Técnico, Universidade de Lisboa · Carnegie Mellon University · Sapienza University of Rome · University of Edinburgh
用 α-entmax 替代 top-k 实现可微分、自适应块选择,在长上下文任务中同时提升精度与速度。
📌 工作参考价值(4 点)
- 在推荐系统的长用户行为序列中,可借鉴分块摘要 + α-entmax 路由的设计:先学习每个块的轻量表示,再用 α-entmax 动态选择相关块,替代固定 top-k,使模型能按需关注任意数量的历史块,提升长序列建模的灵活性。
- α-entmax 的可微稀疏性让块选择能够端到端训练,避免硬截断带来的梯度断裂,适合需要可学习稀疏门控的生成式推荐(如 Semantic ID 序列建模)或 Agent 记忆检索模块。
- 两阶段“粗筛+细读”范式具有工程复用价值:第一阶段用廉价摘要快速评分,第二阶段仅在保留块内计算完整 token 级注意力,可在推理时大幅降低计算量,同时保持精度,适用于工业级推荐模型的服务优化。
- 论文的局部注意力摘要初始化技巧(从均值池化平滑过渡)及 GPU 友好的 Triton 实现策略(块摘要缓存、位掩码压缩、单次融合 pass)可直接启发推荐系统在处理超长序列时的特征工程和系统实现。