Kimi 团队发布 Attention Residuals:用注意力替代残差连接

burny_tech · x · 2026-10-27

Kimi Team 发布论文《Attention Residuals》,指出现代 LLM 标准的 PreNorm 残差连接以固定单位权重累积所有层输出,导致隐状态随深度不受控增长、稀释每层贡献。AttnRes 改用 softmax 注意力让每层以可学习、依赖输入的权重选择性聚合此前各层的表示。为降低大规模训练中对所有前层做注意力的显存与通信开销,提出 Block AttnRes:把层分块、只对块级表示做注意力,配合基于缓存的流水线通信和两阶段计算,可作为标准残差的近似零开销替代。Scaling law 实验显示各模型规模下收益一致,消融验证了内容依赖的深度选择的必要性。开发者 burnytech 正尝试把 AttnRes 加进 General Physics Transformer(GPhT),配图表格显示:加入 Block AttnRes 后辐射层 12 步 NMSE 从 0.39277 降到 0.28917,Full AttnRes 为 0.31735,loss 收敛更快。

原文链接 →

「模型」频道最新

更多「模型」频道 AI 资讯 →