Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
CoolBran
d2l-zh
提交
a2da5585
D
d2l-zh
项目概览
CoolBran
/
d2l-zh
与 Fork 源项目一致
从无法访问的项目Fork
通知
3
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
d2l-zh
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
a2da5585
编写于
11月 20, 2021
作者:
R
Rachel Hu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
self attention polish
上级
17467851
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
82 addition
and
22 deletion
+82
-22
chapter_attention-mechanisms/self-attention-and-positional-encoding.md
...tion-mechanisms/self-attention-and-positional-encoding.md
+82
-22
未找到文件。
chapter_attention-mechanisms/self-attention-and-positional-encoding.md
浏览文件 @
a2da5585
# 自注意力和位置编码
:label:
`sec_self-attention-and-positional-encoding`
在深度学习中,我们经常使用卷积神经网络(CNN)或循环神经网络(RNN)对序列进行编码。现在想象一下,有了注意力机制之后,我们将词元序列输入注意力池化中,以便同一组词元同时充当查询、键和值。具体来说,每个查询都会关注所有的键-值对并生成一个注意力输出。由于查询、键和值来自同一组输入,因此被称为
*自注意力*
(self-attention :cite:
`Lin.Feng.Santos.ea.2017,Vaswani.Shazeer.Parmar.ea.2017`
,也被称为
*内部注意力*
(intra-attention) :cite:
`Cheng.Dong.Lapata.2016,Parikh.Tackstrom.Das.ea.2016,Paulus.Xiong.Socher.2017`
。在本节中,我们将讨论使用自注意力进行序列编码,包括使用序列的顺序作为补充信息。
在深度学习中,我们经常使用卷积神经网络(CNN)或循环神经网络(RNN)对序列进行编码。
想象一下,有了注意力机制之后,我们将词元序列输入注意力池化中,
以便同一组词元同时充当查询、键和值。
具体来说,每个查询都会关注所有的键-值对并生成一个注意力输出。
由于查询、键和值来自同一组输入,因此被称为
*自注意力*
(self-attention)
:cite:
`Lin.Feng.Santos.ea.2017,Vaswani.Shazeer.Parmar.ea.2017`
,
也被称为
*内部注意力*
(intra-attention) :cite:
`Cheng.Dong.Lapata.2016,Parikh.Tackstrom.Das.ea.2016,Paulus.Xiong.Socher.2017`
。
在本节中,我们将使用自注意力进行序列编码,以及如何使用序列的顺序作为补充信息。
```
{.python .input}
from d2l import mxnet as d2l
...
...
@@ -29,11 +36,17 @@ import tensorflow as tf
## [**自注意力**]
给定一个由词元组成的输入序列$
\m
athbf{x}_1,
\l
dots,
\m
athbf{x}_n$,其中任意$
\m
athbf{x}_i
\i
n
\m
athbb{R}^d$($1
\l
eq i
\l
eq n$)。该序列的自注意力输出为一个长度相同的序列$
\m
athbf{y}_1,
\l
dots,
\m
athbf{y}_n$,其中:
给定一个由词元组成的输入序列$
\m
athbf{x}_1,
\l
dots,
\m
athbf{x}_n$,
其中任意$
\m
athbf{x}_i
\i
n
\m
athbb{R}^d$($1
\l
eq i
\l
eq n$)。
该序列的自注意力输出为一个长度相同的序列
$
\m
athbf{y}_1,
\l
dots,
\m
athbf{y}_n$,其中:
$$
\m
athbf{y}_i = f(
\m
athbf{x}_i, (
\m
athbf{x}_1,
\m
athbf{x}_1),
\l
dots, (
\m
athbf{x}_n,
\m
athbf{x}_n))
\i
n
\m
athbb{R}^d$$
根据 :eqref:
`eq_attn-pooling`
中定义的注意力池化函数$f$。下面的代码片段是基于多头注意力对一个张量完成自注意力的计算,张量的形状为(批量大小, 时间步的数目或词元序列的长度, $d$)。输出与输入的张量形状相同。
根据 :eqref:
`eq_attn-pooling`
中定义的注意力池化函数$f$。
下面的代码片段是基于多头注意力对一个张量完成自注意力的计算,
张量的形状为(批量大小,时间步的数目或词元序列的长度,$d$)。
输出与输入的张量形状相同。
```
{.python .input}
num_hiddens, num_heads = 100, 5
...
...
@@ -75,28 +88,61 @@ attention(X, X, X, valid_lens, training=False).shape
让我们比较下面几个架构,目标都是将由$n$个词元组成的序列映射到另一个长度相等的序列,其中的每个输入词元或输出词元都由$d$维向量表示。具体来说,我们将比较的是卷积神经网络、循环神经网络和自注意力这几个架构的计算复杂性、顺序操作和最大路径长度。请注意,顺序操作会妨碍并行计算,而任意的序列位置组合之间的路径越短,则能更轻松地学习序列中的远距离依赖关系 :cite:
`Hochreiter.Bengio.Frasconi.ea.2001`
。
![
比较卷积神经网络(填充词元被忽略)、循环神经网络和自注意力三种架构
。
](
../img/cnn-rnn-self-attention.svg
)
![
比较卷积神经网络(填充词元被忽略)、循环神经网络和自注意力三种架构
](
../img/cnn-rnn-self-attention.svg
)
:label:
`fig_cnn-rnn-self-attention`
考虑一个卷积核大小为$k$的卷积层。我们将在后面的章节中提供关于使用卷积神经网络处理序列的更多详细信息。目前,我们只需要知道,由于序列长度是$n$,输入和输出的通道数量都是$d$,所以卷积层的计算复杂度为$
\m
athcal{O}(knd^2)$。如 :numref:
`fig_cnn-rnn-self-attention`
所示,卷积神经网络是分层的,因此为有$
\m
athcal{O}(1)$个顺序操作,最大路径长度为$
\m
athcal{O}(n/k)$。例如,$
\m
athbf{x}_1$和$
\m
athbf{x}_5$处于 :numref:
`fig_cnn-rnn-self-attention`
中卷积核大小为3的双层卷积神经网络的感受野内。
当更新循环神经网络的隐状态时,$d
\t
imes d$权重矩阵和$d$维隐状态的乘法计算复杂度为$
\m
athcal{O}(d^2)$。由于序列长度为$n$,因此循环神经网络层的计算复杂度为$
\m
athcal{O}(nd^2)$。根据 :numref:
`fig_cnn-rnn-self-attention`
,有$
\m
athcal{O}(n)$个顺序操作无法并行化,最大路径长度也是$
\m
athcal{O}(n)$。
在自注意力中,查询、键和值都是$n
\t
imes d$矩阵。考虑 :eqref:
`eq_softmax_QK_V`
中缩放的”点-积“注意力,其中$n
\t
imes d$矩阵乘以$d
\t
imes n$矩阵,然后输出的$n
\t
imes n$矩阵乘以$n
\t
imes d$矩阵。因此,自注意力具有$
\m
athcal{O}(n^2d)$计算复杂性。正如我们在 :numref:
`fig_cnn-rnn-self-attention`
中看到的那样,每个词元都通过自注意力直接连接到任何其他词元。因此,有$
\m
athcal{O}(1)$个顺序操作可以并行计算,最大路径长度也是$
\m
athcal{O}(1)$。
总而言之,卷积神经网络和自注意力都拥有并行计算的优势,而且自注意力的最大路径长度最短。但是因为其计算复杂度是关于序列长度的二次方,所以在很长的序列中计算会非常慢。
考虑一个卷积核大小为$k$的卷积层。
我们将在后面的章节中提供关于使用卷积神经网络处理序列的更多详细信息。
目前,我们只需要知道,由于序列长度是$n$,输入和输出的通道数量都是$d$,
所以卷积层的计算复杂度为$
\m
athcal{O}(knd^2)$。
如 :numref:
`fig_cnn-rnn-self-attention`
所示,
卷积神经网络是分层的,因此为有$
\m
athcal{O}(1)$个顺序操作,
最大路径长度为$
\m
athcal{O}(n/k)$。
例如,$
\m
athbf{x}_1$和$
\m
athbf{x}_5$处于
:numref:
`fig_cnn-rnn-self-attention`
中卷积核大小为3的双层卷积神经网络的感受野内。
当更新循环神经网络的隐状态时,
$d
\t
imes d$权重矩阵和$d$维隐状态的乘法计算复杂度为$
\m
athcal{O}(d^2)$。
由于序列长度为$n$,因此循环神经网络层的计算复杂度为$
\m
athcal{O}(nd^2)$。
根据 :numref:
`fig_cnn-rnn-self-attention`
,
有$
\m
athcal{O}(n)$个顺序操作无法并行化,最大路径长度也是$
\m
athcal{O}(n)$。
在自注意力中,查询、键和值都是$n
\t
imes d$矩阵。
考虑 :eqref:
`eq_softmax_QK_V`
中缩放的”点-积“注意力,
其中$n
\t
imes d$矩阵乘以$d
\t
imes n$矩阵。
之后输出的$n
\t
imes n$矩阵乘以$n
\t
imes d$矩阵。
因此,自注意力具有$
\m
athcal{O}(n^2d)$计算复杂性。
正如我们在 :numref:
`fig_cnn-rnn-self-attention`
中看到的那样,
每个词元都通过自注意力直接连接到任何其他词元。
因此,有$
\m
athcal{O}(1)$个顺序操作可以并行计算,
最大路径长度也是$
\m
athcal{O}(1)$。
总而言之,卷积神经网络和自注意力都拥有并行计算的优势,
而且自注意力的最大路径长度最短。
但是因为其计算复杂度是关于序列长度的二次方,所以在很长的序列中计算会非常慢。
## [**位置编码**]
:label:
`subsec_positional-encoding`
在处理词元序列时,循环神经网络是逐个的重复地处理词元的,而自注意力则因为并行计算而放弃了顺序操作。为了使用序列的顺序信息,我们通过在输入表示中添加
*位置编码*
(positional encoding)来注入绝对的或相对的位置信息。位置编码可以通过学习得到也可以直接固定得到。接下来,我们描述的是基于正弦函数和余弦函数的固定位置编码 :cite:
`Vaswani.Shazeer.Parmar.ea.2017`
。
在处理词元序列时,循环神经网络是逐个的重复地处理词元的,
而自注意力则因为并行计算而放弃了顺序操作。
为了使用序列的顺序信息,我们通过在输入表示中添加
*位置编码*
(positional encoding)来注入绝对的或相对的位置信息。
位置编码可以通过学习得到也可以直接固定得到。
接下来,我们描述的是基于正弦函数和余弦函数的固定位置编码
:cite:
`Vaswani.Shazeer.Parmar.ea.2017`
。
假设输入表示$
\m
athbf{X}
\i
n
\m
athbb{R}^{n
\t
imes d}$包含一个序列中$n$个词元的$d$维嵌入表示。位置编码使用相同形状的位置嵌入矩阵$
\m
athbf{P}
\i
n
\m
athbb{R}^{n
\t
imes d}$输出$
\m
athbf{X} +
\m
athbf{P}$,矩阵第$i$行、第$2j$列和$2j$列上的元素为:
假设输入表示$
\m
athbf{X}
\i
n
\m
athbb{R}^{n
\t
imes d}$
包含一个序列中$n$个词元的$d$维嵌入表示。
位置编码使用相同形状的位置嵌入矩阵
$
\m
athbf{P}
\i
n
\m
athbb{R}^{n
\t
imes d}$输出$
\m
athbf{X} +
\m
athbf{P}$,
矩阵第$i$行、第$2j$列和$2j$列上的元素为:
$$
\b
egin{aligned} p_{i, 2j} &=
\s
in
\l
eft(
\f
rac{i}{10000^{2j/d}}
\r
ight),
\\
p_{i, 2j+1} &=
\c
os
\l
eft(
\f
rac{i}{10000^{2j/d}}
\r
ight).
\e
nd{aligned}$$
:eqlabel:
`eq_positional-encoding-def`
乍一看,这种基于三角函数的设计看起来很奇怪。在解释这个设计之前,让我们先在下面的
`PositionalEncoding`
类中实现它。
乍一看,这种基于三角函数的设计看起来很奇怪。
在解释这个设计之前,让我们先在下面的
`PositionalEncoding`
类中实现它。
```
{.python .input}
#@save
...
...
@@ -159,7 +205,10 @@ class PositionalEncoding(tf.keras.layers.Layer):
return self.dropout(X, **kwargs)
```
在位置嵌入矩阵$
\m
athbf{P}$中,[
**行代表词元在序列中的位置,列代表位置编码的不同维度**
]。在下面的例子中,我们可以看到位置嵌入矩阵的第$6$列和第$7$列的频率高于第$8$列和第$9$列。第$6$列和第$7$列之间的偏移量(第$8$列和第$9$列相同)是由于正弦函数和余弦函数的交替。
在位置嵌入矩阵$
\m
athbf{P}$中,
[
**行代表词元在序列中的位置,列代表位置编码的不同维度**
]。
在下面的例子中,我们可以看到位置嵌入矩阵的第$6$列和第$7$列的频率高于第$8$列和第$9$列。
第$6$列和第$7$列之间的偏移量(第$8$列和第$9$列相同)是由于正弦函数和余弦函数的交替。
```
{.python .input}
encoding_dim, num_steps = 32, 60
...
...
@@ -194,15 +243,20 @@ d2l.plot(np.arange(num_steps), P[0, :, 6:10].T, xlabel='Row (position)',
### 绝对位置信息
为了明白沿着编码维度单调降低的频率与绝对位置信息的关系,让我们打印出$0, 1,
\l
dots, 7$的[
**二进制表示**
]形式。正如我们所看到的,每个数字、每两个数字和每四个数字上的比特值在第一个最低位、第二个最低位和第三个最低位上分别交替。
为了明白沿着编码维度单调降低的频率与绝对位置信息的关系,
让我们打印出$0, 1,
\l
dots, 7$的[
**二进制表示**
]形式。
正如我们所看到的,每个数字、每两个数字和每四个数字上的比特值
在第一个最低位、第二个最低位和第三个最低位上分别交替。
```
{.python .input}
#@tab all
for i in range(8):
print(f'{i}
in binary is
{i:>03b}')
print(f'{i}
的二进制是:
{i:>03b}')
```
在二进制表示中,较高比特位的交替频率低于较低比特位,与下面的热图所示相似,只是位置编码通过使用三角函数[
**在编码维度上降低频率**
]。由于输出是浮点数,因此此类连续表示比二进制表示法更节省空间。
在二进制表示中,较高比特位的交替频率低于较低比特位,
与下面的热图所示相似,只是位置编码通过使用三角函数[
**在编码维度上降低频率**
]。
由于输出是浮点数,因此此类连续表示比二进制表示法更节省空间。
```
{.python .input}
P = np.expand_dims(np.expand_dims(P[0, :, :], 0), 0)
...
...
@@ -226,9 +280,15 @@ d2l.show_heatmaps(P, xlabel='Column (encoding dimension)',
### 相对位置信息
除了捕获绝对位置信息之外,上述的位置编码还允许模型学习得到输入序列中相对位置信息。这是因为对于任何确定的位置偏移$
\d
elta$,位置$i +
\d
elta$处的位置编码可以线性投影位置$i$处的位置编码来表示。
除了捕获绝对位置信息之外,上述的位置编码还允许模型学习得到输入序列中相对位置信息。
这是因为对于任何确定的位置偏移$
\d
elta$,位置$i +
\d
elta$处
的位置编码可以线性投影位置$i$处的位置编码来表示。
这种投影的数学解释是,令$
\o
mega_j = 1/10000^{2j/d}$,对于任何确定的位置偏移$
\d
elta$, :eqref:
`eq_positional-encoding-def`
中的任何一对$(p_{i, 2j}, p_{i, 2j+1})$都可以线性投影到$(p_{i+
\d
elta, 2j}, p_{i+
\d
elta, 2j+1})$:
这种投影的数学解释是,令$
\o
mega_j = 1/10000^{2j/d}$,
对于任何确定的位置偏移$
\d
elta$,
:eqref:
`eq_positional-encoding-def`
中的任何一对
$(p_{i, 2j}, p_{i, 2j+1})$都可以线性投影到
$(p_{i+
\d
elta, 2j}, p_{i+
\d
elta, 2j+1})$:
$$
\b
egin{aligned}
&
\b
egin{bmatrix}
\c
os(
\d
elta
\o
mega_j) &
\s
in(
\d
elta
\o
mega_j)
\\
-
\s
in(
\d
elta
\o
mega_j) &
\c
os(
\d
elta
\o
mega_j)
\\
\e
nd{bmatrix}
...
...
@@ -245,7 +305,7 @@ $2\times 2$投影矩阵不依赖于任何位置的索引$i$。
*
在自注意力中,查询、键和值都来自同一组输入。
*
卷积神经网络和自注意力都拥有并行计算的优势,而且自注意力的最大路径长度最短。但是因为其计算复杂度是关于序列长度的二次方,所以在很长的序列中计算会非常慢。
*
为了使用序列的顺序信息,我们可以通过在输入表示中添加位置编码来注入绝对的或相对的位置信息。
*
为了使用序列的顺序信息,我们可以通过在输入表示中添加位置编码
,
来注入绝对的或相对的位置信息。
## 练习
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录