Skip to content
简体中文

切换主题

// ◎ Transformer 与注意力

ALiBi 线性偏置

按距离给注意力分数加线性惩罚,代替位置编码。

ALiBi (Attention with Linear Biases) 2021 进阶 扩展

通俗解释

不加位置向量,而是让越远的 token 注意力分数扣得越多。训练短、推理长时外推更好。

用在哪里:BLOOM、MPT

时间轴 · 前置 → 当前 → 后继

2017 Transformer 与…2021 ALiBi 线性偏置

01原始论文

Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolation

Ofir Press, Noah A. Smith, Mike Lewis · 2021

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

左边先学;右边是学完它之后可以学的。

先学

ALiBi 线性偏置

学完可以学

  • 图谱里暂无后继

相关:旋转位置编码 RoPE

Transformer 与自注…旋转位置编码 RoPE ALiBi 线性偏置

左:前置 · 右:后继 · 虚线:相关。看完整关系图 →

03学习资料

官方文档、图解、课程和参考实现,链接都核对过。

04在课程里哪里学

给 AI 读 / 数据说明

数据更新于:AI 算法图谱与课程体系(北京时间)