趋近智
给定时间步 的 GRU 单元内,计算实际的隐藏状态 是最后一步。这个计算整合了更新门 ()、重置门 () 和候选隐藏状态 ()。这个状态表示 GRU 将传递到下一个时间步的信息。
最终隐藏状态 的计算中,更新门 起着核心作用。回想一下, 决定了之前隐藏状态 应保留多少,以及新的候选状态 应包含多少。GRU 通过直接插值机制来实现这一点。
最终隐藏状态 的公式是:
我们来分析这个公式:
: 此项计算应保留之前隐藏状态 的多少。更新门 的值介于 0 到 1 之间(因为使用了 Sigmoid 激活函数 (activation function))。如果 的某个值接近 0,则 中的对应值将接近 1,这意味着之前状态 中的对应元素在很大程度上得以保留。反之,如果 的某个值接近 1,则 中的对应值接近 0,从而有效地遗忘之前状态的该部分。符号 表示元素级乘法(哈达玛积)。
: 此项计算应包含候选隐藏状态 的多少。如果 的某个值接近 1,则候选状态 中的对应元素(包含为此时间步提议的新信息,受重置门影响)会被强烈包含。如果 的某个值接近 0,则候选状态中的对应元素大部分会被忽略。
加法 (+): 两个所得向量 (vector)进行元素级相加。这个加法完成了插值操作。最终隐藏状态 的每个元素都是之前状态 和候选状态 中对应元素的加权和,权重 (weight)由更新门 控制。
这种机制使得 GRU 单元能够动态调整从过去流向的信息量以及在每个时间步引入的新信息量。如果更新门认为之前状态仍然相关( 值较低),它可以使其在很大程度上保持不变地通过。如果它认定新信息更重要( 值较高),它会更多地包含候选状态。这种门控机制比 LSTM 的独立单元状态和输出门更简单,但提供了一种有效的方式来管理信息流并解决梯度消失问题。
以下图表说明了之前状态、候选状态和更新门如何组合形成最终隐藏状态:
最终隐藏状态 是通过对之前隐藏状态 和候选隐藏状态 进行插值计算而得出的。更新门 控制着这种插值的平衡。
所得向量 有两个作用:它是当前时间步 的 GRU 单元的输出(通常传递给后续层或用于预测),并且它成为下一个时间步 的“之前隐藏状态” 。这种循环连接使得 GRU 能够逐步处理序列,根据其遇到的序列元素来维护和更新其内部状态。
这部分内容有帮助吗?
© 2026 ApX Machine Learning用心打造