Xin Luo

Stem模块提升LLM稀疏注意力效率

研究人员推出了一种名为Stem的新型模块，旨在提高大型语言模型（LLMs）中稀疏注意力机制的效率。Stem解决了自注意力机制的计算瓶颈，特别是在预填充阶段的长上下文场景中。它通过从信息流的角度重新思考因果注意力，采用Token Position-Decay策略和Output-Aware Metric来优先处理重要Token并保留累积依赖关系，从而实现这一目标。