Skip to content
简体中文

切换主题

// ◎ Transformer 与注意力

环形注意力 Ring Attention

把超长序列切到多张卡上,按环传递 K/V 计算注意力。

Ring Attention 2023 前沿研究 核心

通俗解释

每张卡持有序列的一段,计算与本地 K/V 的注意力,同时把 K/V 传给下一张卡,通信与计算重叠。上下文长度可随设备数线性增长,是“上下文并行”的代表。

用在哪里:百万 token 级上下文训练

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

2022 FlashAttention2023 环形注意力 Ring Attent…

01原始论文

Ring Attention with Blockwise Transformers for Near-Infinite Context

Hao Liu, Matei Zaharia, Pieter Abbeel · 2023

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

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

先学

环形注意力 Ring Attention

学完可以学

  • 图谱里暂无后继

相关:YaRN 上下文扩展张量并行(Megatron)

FlashAttentionYaRN 上下文扩展张量并行(Megatron) 环形注意力 Ring Attention

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

03学习资料

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

04在课程里哪里学

给 AI 读 / 数据说明

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