跳到正文
arXiv·· 14 天前AI 评分24

SharpDraft:用基数感知 Query 缩放加速长上下文投机解码

SharpDraft: Accelerating Long-Context Speculative Decoding with Cardinality-Aware Query Scaling

AI 导读

SharpDraft 是一种免训练方法,通过基数感知的 Query 缩放缓解注意力质量稀释问题,从而加速长上下文投机解码。该方法针对上下文增长时 softmax 在更多可见 Key 上归一化、导致高分 Key 上的注意力质量下降的现象,无需在线自适应的计算开销。

来源:arXiv · arxiv.org