多层 RNN原理以及实现
数学原理多层 RNN 的核心思想是堆叠多个 RNN 层,每一层的输出作为下一层的输入,从而逐层提取更高条理的抽象特性。
1. 单层 RNN 的数学表示
起首,单层 RNN 的计算过程如下。对于一个时间步 t t t,单层 RNN 的隐藏状态 h t h_t ht 和输出 y t y_t yt 可以表示为:
h t = activation ( W i h x t + b i h + W h h h t − 1 + b h h ) h_t = \text{activation}(W_{ih} x_t + b_{ih} + W_{hh} h_{t-1} + b_{hh}) ht=activation(Wihxt+bih+Whhht−1+bhh)
y t = W h o h t + b h o y_t = W_{ho} h_t + b_{ho} yt=Whoht+bho
其中:
[*] x t x_t xt 是时间步 t t t 的输入。
[*] h t h_t ht 是时间步 t t t 的隐藏状态。
[*] h t − 1 h_{t-1} ht−1 是时间步 t − 1 t-1 t−1 的隐藏状态。
[*] W i h W_{ih} Wih、 W h h W_{hh} Whh、 W h o W_{ho} Who 是权重矩阵。
[*] b i h b_{ih} bih、 b h h b_{hh} bhh、 b h o b_{ho} bho 是偏置项。
[*] activation \text{activation} activation 是激活函数(如 tanh \tanh tanh 或 ReLU \text{ReLU} ReLU)。
2. 多层 RNN 的数学表示
假设我们有一个 L L L 层的 RNN,每一层的隐藏状态为 h t ( l ) h_t^{(l)} ht(l),其中 l l l 表示第 l l l 层, t t t 表示时间步。多层 RNN 的计算过程如下:
(1) 第一层( l = 1 l = 1 l=1)
第一层的输入是原始输入序列 x t x_t xt,隐藏状态 h t ( 1 ) h_t^{(1)} ht(1) 的计算公式为:
h t ( 1 ) = activation ( W i h ( 1 ) x t + b i h ( 1 ) + W h h ( 1 ) h t − 1 ( 1 ) + b h h ( 1 ) ) h_t^{(1)} = \text{activation}(W_{ih}^{(1)} x_t + b_{ih}^{(1)} + W_{hh}^{(1)} h_{t-1}^{(1)} + b_{hh}^{(1)}) ht(1)=activation(Wih(1)xt+bih(1)+Whh(1)ht−1(1)+bhh(1))
其中:
[*] W i h ( 1 ) W_{ih}^{(1)} Wih(1)、 W h h ( 1 ) W_{hh}^{(1)} Whh(1) 是第一层的权重矩阵。
[*] b i h ( 1 ) b_{ih}^{(1)} bih(1)、 b h h ( 1 ) b_{hh}^{(1)} bhh(1) 是第一层的偏置项。
(2) 第 l l l 层( l > 1 l > 1 l>1)
第 l l l 层的输入是第 l − 1 l-1 l−1 层的输出 h t ( l − 1 ) h_t^{(l-1)} ht(l−1),隐藏状态 h t ( l ) h_t^{(l)} ht(l) 的计算公式为:
h t ( l ) = activation ( W i h ( l ) h t ( l − 1 ) + b i h ( l ) + W h h ( l ) h t − 1 ( l ) + b h h ( l ) ) h_t^{(l)} = \text{activation}(W_{ih}^{(l)} h_t^{(l-1)} + b_{ih}^{(l)} + W_{hh}^{(l)} h_{t-1}^{(l)} + b_{hh}^{(l)}) ht(l)=activation(Wih(l)ht(l−1)+bih(l)+Whh(l)ht−1(l)+bhh(l))
其中:
[*] W i h ( l ) W_{ih}^{(l)} Wih(l)、 W h h ( l ) W_{hh}^{(l)} Whh(l) 是第 l l l 层的权重矩阵。
[*] b i h ( l ) b_{ih}^{(l)} bih(l)、 b h h ( l ) b_{hh}^{(l)} bhh(l) 是第 l l l 层的偏置项。
(3) 输出层
最后一层(第 L L L 层)的输出 h t ( L ) h_t^{(L)} ht(L) 作为整个网络的输出 y t y_t yt:
y t = W h o h t ( L ) + b h o y_t = W_{ho} h_t^{(L)} + b_{ho} yt=Whoht(L)+bho
其中:
[*] W h o W_{ho} Who、 b h o b_{ho} bho 是输出层的权重矩阵和偏置项。
3. 多层 RNN 的数据流向
以下是一个 L L L 层 RNN 的数据流向的数学描述:
(1) 输入序列
输入序列为 x 1 , x 2 , … , x T x_1, x_2, \dots, x_T x1,x2,…,xT,其中 T T T 是序列长度。
(2) 初始化隐藏状态
每一层的初始隐藏状态 h 0 ( l ) h_0^{(l)} h0(l) 通常初始化为零或随机值:
h 0 ( l ) = 0 或 h 0 ( l ) ∼ N ( 0 , σ 2 ) h_0^{(l)} = \mathbf{0} \quad \text{或} \quad h_0^{(l)} \sim \mathcal{N}(0, \sigma^2) h0(l)=0或h0(l)∼N(0,σ2)
(3) 时间步 t t t 的计算
对于每个时间步 t t t,从第一层到第 L L L 层依次计算隐藏状态:
[*] 第一层:
h t ( 1 ) = activation ( W i h ( 1 ) x t + b i h ( 1 ) + W h h ( 1 ) h t − 1 ( 1 ) + b h h ( 1 ) ) h_t^{(1)} = \text{activation}(W_{ih}^{(1)} x_t + b_{ih}^{(1)} + W_{hh}^{(1)} h_{t-1}^{(1)} + b_{hh}^{(1)}) ht(1)=activation(Wih(1)xt+bih(1)+Whh(1)ht−1(1)+bhh(1))
[*] 第 l l l 层( l > 1 l > 1 l>1):
h t ( l ) = activation ( W i h ( l ) h t ( l − 1 ) + b i h ( l ) + W h h ( l ) h t − 1 ( l ) + b h h ( l ) ) h_t^{(l)} = \text{activation}(W_{ih}^{(l)} h_t^{(l-1)} + b_{ih}^{(l)} + W_{hh}^{(l)} h_{t-1}^{(l)} + b_{hh}^{(l)}) ht(l)=activation(Wih(l)ht(l−1)+bih(l)+Whh(l)ht−1(l)+bhh(l))
[*] 输出:
y t = W h o h t ( L ) + b h o y_t = W_{ho} h_t^{(L)} + b_{ho} yt=Whoht(L)+bho
(4) 序列输出
终极,整个序列的输出为 y 1 , y 2 , … , y T y_1, y_2, \dots, y_T y1,y2,…,yT。
4. 多层 RNN 的特点
(1) 逐层抽象
[*]每一层 RNN 可以看作是对输入序列的差异条理的抽象。
[*]较低层捕捉局部和细节信息,较高层捕捉全局和语义信息。
(2) 参数共享
[*]每一层的参数(权重矩阵和偏置项)在时间步之间共享。
[*]差异层的参数是独立的。
(3) 梯度流传
[*]在反向流传时,梯度会通过期间步和层数流传。
[*]由于梯度消散或爆炸题目,练习深层 RNN 大概会比较困难。
可视化原理
下面是一个可视化的结构显示图:其中每一层神经元都要有两个方向的输出,一个是向本时间步的下一层传送,另一个是向下一个时间步的本层传送。而且,每一个神经元都有两个权重矩阵。注意:下方右图仅仅是逻辑上睁开的数据流,其中差异世间步上的同一层,用的是同一个权重矩阵。
https://i-blog.csdnimg.cn/direct/3d0b8e179aac4188a6939c0f1d6ada76.jpeg#pic_center
代码实现
1. 示例任务
假设有一个简单的任务:
[*]处理一个长度为 4 的序列
[*]批次大小为 2
[*]每个时间步的输入特性维度为 3
[*]希望使用一个 2 层的单向 RNN
[*]隐藏状态维度为 5。
2. 输入数据
输入句子
[*]句子 1: “I love PyTorch”
[*]句子 2: “RNN is fun”
输入数据的形状
[*]序列长度 (seq_len): 4(假设每个单词是一个时间步)
[*]批次大小 (batch_size): 2
[*]输入特性维度 (input_size): 3(假设每个单词用一个 3 维向量表示)
具体输入数据
import torch
# 输入数据形状: (seq_len, batch_size, input_size)
input_data = torch.tensor([
# 时间步 1
[,# 句子 1 的第一个单词
], # 句子 2 的第一个单词
# 时间步 2
[,# 句子 1 的第二个单词
], # 句子 2 的第二个单词
# 时间步 3
[,# 句子 1 的第三个单词
], # 句子 2 的第三个单词
# 时间步 4
[,# 句子 1 的第四个单词
]# 句子 2 的第四个单词
])
print("Input shape:", input_data.shape)# 输出: torch.Size()
3. 初始隐藏状态
初始隐藏状态的形状
[*]RNN 层数 (num_layers): 2
[*]方向数 (num_directions): 1(单向 RNN)
[*]批次大小 (batch_size): 2
[*]隐藏状态维度 (hidden_size): 5
具体初始隐藏状态
# 初始隐藏状态形状: (num_layers * num_directions, batch_size, hidden_size)
h0 = torch.zeros(2, 2, 5)# 2层RNN,批次大小为2,隐藏状态维度为5
print("h0 shape:", h0.shape)# 输出: torch.Size()
4. 界说 RNN 模子
import torch.nn as nn
# 定义 RNN
rnn = nn.RNN(
input_size=3,# 输入特征维度
hidden_size=5, # 隐藏状态维度
num_layers=2,# RNN 层数
batch_first=False# 输入形状为 (seq_len, batch_size, input_size)
)
5. 前向流传
计算输出
# 前向传播
output, hn = rnn(input_data, h0)
print("Output shape:", output.shape)# 输出: torch.Size()
print("hn shape:", hn.shape) # 输出: torch.Size()
输出解析
[*] output:
[*]形状为 (seq_len, batch_size, hidden_size),即 (4, 2, 5)。
[*]包含了每个时间步的隐藏状态。
[*]例如,output 是第一个时间步的隐藏状态,output[-1] 是最后一个时间步的隐藏状态。
[*] hn:
[*]形状为 (num_layers, batch_size, hidden_size),即 (2, 2, 5)。
[*]包含了最后一个时间步的隐藏状态。
[*]例如,hn 是第一层的终极隐藏状态,hn 是第二层的终极隐藏状态。
6. 具体输出值
output 的值
print("Output (所有时间步的隐藏状态):")
print(output)
输出示例:
tensor([[[ 0.1234,0.5678,0.9101,0.1121,0.3141],
[ 0.4151,0.6171,0.8191,0.0212,0.2232]],
[[ 0.4252,0.6272,0.8292,0.0313,0.2333],
[ 0.4353,0.6373,0.8393,0.0414,0.2434]],
[[ 0.4454,0.6474,0.8494,0.0515,0.2535],
[ 0.4555,0.6575,0.8595,0.0616,0.2636]],
[[ 0.4656,0.6676,0.8696,0.0717,0.2737],
[ 0.4757,0.6777,0.8797,0.0818,0.2838]]],
grad_fn=<StackBackward>)
hn 的值
print("hn (最后一个时间步的隐藏状态):")
print(hn)
输出示例:
tensor([[[ 0.4656,0.6676,0.8696,0.0717,0.2737],
[ 0.4757,0.6777,0.8797,0.0818,0.2838]],
[[ 0.4858,0.6878,0.8898,0.0919,0.2939],
[ 0.4959,0.6979,0.8999,0.1020,0.3040]]],
grad_fn=<StackBackward>)
batch_first=True时
以下是一个具体的例子,展示当 batch_first=True 时,PyTorch 中 torch.nn.RNN 的输入、输出以及参数的作用。
任务
假设有一个简单的任务:
[*]处理一个长度为 4 的序列
[*]批次大小为 2
[*]每个时间步的输入特性维度为 3
[*]希望使用一个 2 层的单向 RNN
[*]隐藏状态维度为 5
[*]而且设置 batch_first=True。
2. 输入数据
输入句子
[*]句子 1: “I love PyTorch”
[*]句子 2: “RNN is fun”
输入数据的形状
[*]批次大小 (batch_size): 2
[*]序列长度 (seq_len): 4(假设每个单词是一个时间步)
[*]输入特性维度 (input_size): 3(假设每个单词用一个 3 维向量表示)
具体输入数据
import torch
# 输入数据形状: (batch_size, seq_len, input_size)
input_data = torch.tensor([
# 句子 1
[,# 第一个单词
,# 第二个单词
,# 第三个单词
], # 第四个单词
# 句子 2
[,# 第一个单词
,# 第二个单词
,# 第三个单词
]# 第四个单词
])
print("Input shape:", input_data.shape)# 输出: torch.Size()
3. 初始隐藏状态
初始隐藏状态的形状
[*]RNN 层数 (num_layers): 2
[*]方向数 (num_directions): 1(单向 RNN)
[*]批次大小 (batch_size): 2
[*]隐藏状态维度 (hidden_size): 5
具体初始隐藏状态
# 初始隐藏状态形状: (num_layers * num_directions, batch_size, hidden_size)
h0 = torch.zeros(2, 2, 5)# 2层RNN,批次大小为2,隐藏状态维度为5
print("h0 shape:", h0.shape)# 输出: torch.Size()
4. 界说 RNN 模子
import torch.nn as nn
# 定义 RNN
rnn = nn.RNN(
input_size=3,# 输入特征维度
hidden_size=5, # 隐藏状态维度
num_layers=2,# RNN 层数
batch_first=True# 输入形状为 (batch_size, seq_len, input_size)
)
5. 前向流传
计算输出
# 前向传播
output, hn = rnn(input_data, h0)
print("Output shape:", output.shape)# 输出: torch.Size()
print("hn shape:", hn.shape) # 输出: torch.Size()
输出解析
[*] output:
[*]形状为 (batch_size, seq_len, hidden_size),即 (2, 4, 5)。
[*]包含了每个时间步的隐藏状态。
[*]例如,output 是第一个句子的所有时间步的隐藏状态,output 是第二个句子的所有时间步的隐藏状态。
[*] hn:
[*]形状为 (num_layers, batch_size, hidden_size),即 (2, 2, 5)。
[*]包含了最后一个时间步的隐藏状态。
[*]例如,hn 是第一层的终极隐藏状态,hn 是第二层的终极隐藏状态。
6. 具体输出值
output 的值
print("Output (所有时间步的隐藏状态):")
print(output)
输出示例:
tensor([[[ 0.1234,0.5678,0.9101,0.1121,0.3141],
[ 0.4252,0.6272,0.8292,0.0313,0.2333],
[ 0.4454,0.6474,0.8494,0.0515,0.2535],
[ 0.4656,0.6676,0.8696,0.0717,0.2737]],
[[ 0.4151,0.6171,0.8191,0.0212,0.2232],
[ 0.4353,0.6373,0.8393,0.0414,0.2434],
[ 0.4555,0.6575,0.8595,0.0616,0.2636],
[ 0.4757,0.6777,0.8797,0.0818,0.2838]]],
grad_fn=<TransposeBackward0>)
hn 的值
print("hn (最后一个时间步的隐藏状态):")
print(hn)
输出示例:
tensor([[[ 0.4656,0.6676,0.8696,0.0717,0.2737],
[ 0.4757,0.6777,0.8797,0.0818,0.2838]],
[[ 0.4858,0.6878,0.8898,0.0919,0.2939],
[ 0.4959,0.6979,0.8999,0.1020,0.3040]]],
grad_fn=<StackBackward>)
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页:
[1]