HuggingFace Daily Papers(社区热门论文)
53AI 编辑部评分,满分 100

当注意力失明:ALiBi 位置编码中的数值失效问题

2026-08-04 08:00· 1天前
AI 导读

研究发现 ALiBi 位置编码的线性偏置缩放会因浮点精度下溢而失效,使大量注意力权重归零,导致注意力头部分失明。在 148M 参数解码器模型的预训练实验中,该失效模式会显著损害 token 检索能力,但对标准解码器基准影响较小。四种训练期缓解策略中,对数缩放距离在 passkey 检索上提升最稳定,且默认 ALiBi 斜率在 needle-in-a-haystack 检索中仍是强基线。

We identify a previously overlooked failure mode of ALiBi positional encoding: its linear bias scaling underflows floating-point precision, which zeroes out a large fraction of attention weights and renders the affected attention heads partially blind. We analyze this failure mode, characterize its impact, and examine four mitigation strategies. We further demonstrate its occurrence in state-of-the-art pretrained models based on ALiBi. Comprehensive pretraining experiments with 148M-parameter decoder models help us to disentangle its effects from out-of-context degradation. We find that ALiBi's failure mode can substantially impair token retrieval while having only a minor effect on standard decoder benchmarks. We propose four training-time mitigation strategies and evaluate them individually and in combinations, finding that log-scaled distances yield the most consistent improvements in passkey retrieval. Despite this problem, default ALiBi slopes remain a surprisingly strong baseline, particularly for needle-in-a-haystack retrieval. Based on these findings we provide concrete recommendations on how to train models with ALiBi.

来源:HuggingFace Daily Papers(社区热门论文) · arxiv.org

当注意力失明:ALiBi 位置编码中的数值失效问题

HuggingFace Daily Papers(社区热门论文)·2026-08-04 08:00·1天前
AI 导读

研究发现 ALiBi 位置编码的线性偏置缩放会因浮点精度下溢而失效,使大量注意力权重归零,导致注意力头部分失明。在 148M 参数解码器模型的预训练实验中,该失效模式会显著损害 token 检索能力,但对标准解码器基准影响较小。四种训练期缓解策略中,对数缩放距离在 passkey 检索上提升最稳定,且默认 ALiBi 斜率在 needle-in-a-haystack 检索中仍是强基线。

原文 · 保持原样,未翻译

We identify a previously overlooked failure mode of ALiBi positional encoding: its linear bias scaling underflows floating-point precision, which zeroes out a large fraction of attention weights and renders the affected attention heads partially blind. We analyze this failure mode, characterize its impact, and examine four mitigation strategies. We further demonstrate its occurrence in state-of-the-art pretrained models based on ALiBi. Comprehensive pretraining experiments with 148M-parameter decoder models help us to disentangle its effects from out-of-context degradation. We find that ALiBi's failure mode can substantially impair token retrieval while having only a minor effect on standard decoder benchmarks. We propose four training-time mitigation strategies and evaluate them individually and in combinations, finding that log-scaled distances yield the most consistent improvements in passkey retrieval. Despite this problem, default ALiBi slopes remain a surprisingly strong baseline, particularly for needle-in-a-haystack retrieval. Based on these findings we provide concrete recommendations on how to train models with ALiBi.

来源:HuggingFace Daily Papers(社区热门论文)· arxiv.org