Softplus Attention with Re-weighting Boosts Length Extrapolation in Large Language Models
Softplus注意力与重加权提升大语言模型的长度外推能力
机构 * University of California, Berkeley(加州大学伯克利分校)
专题命中 长上下文与记忆 :large language model(title,abstract);language model(title,abstract);foundation model(abstract);分类 cs.CL、cs.AI、cs.LG
AI总结 提出一种两阶段注意力机制,用Softplus和l1归一化替代Softmax,并引入基于不变熵的动态缩放因子和重加权机制,以提升数值稳定性、缓解注意力下沉现象,并显著改善长度外推性能。
Comments Accepted by ICML 2026