735 字
4 分钟
RNN-Seq2Seq序列到序列模型
动机
前面三种RNN模式要求输入和输出序列等长,但机器翻译这类任务中输入输出长度往往不同,需要一种更灵活的架构。
序列到序列架构:编码器-解码器
Seq2Seq由两个RNN组成:
编码器(Encoder)/读取器(Reader):
- 读取输入序列
- 将整个输入序列压缩成一个上下文向量
- 通常取最后一个隐藏状态或者其简单的函数:
解码器(Decoder)/写入器(Writer):
- 以上下文向量 为条件
- 逐步生成输出序列
- 每一步的输入是上一步的输出(或训练时用Teacher Forcing)
训练目标: 最大化条件对数似然

Seq2Seq的瓶颈问题
基础Seq2Seq有一个明显缺陷:整个输入序列被压缩成固定长度的向量 ,当输入序列很长时, 很难保留所有关键信息,导致翻译质量下降。
注意力机制(Attention)
注意力机制正是为了解决上述瓶颈而提出的。核心思想是:
解码器在生成每个输出词时,不再只依赖固定的 ,而是动态地关注编码器不同时刻的隐藏状态。
假设encoder的隐藏状态为 ,decoder在生成第 个输出时,会计算一个上下文向量 ,它是编码器隐藏状态的加权和:
-
解码器在每一步计算当前状态与编码器各时刻隐藏状态的相关性权重:
- 其中 衡量解码器第 步与编码器第 步的匹配程度。
- score函数可以是点积、可学习的前馈网络(MLP)等。
-
然后用这些权重对编码器隐藏状态做加权求和,得到当前步专属的上下文向量:
-
最后,解码器在生成输出时使用 而不是固定的
这样解码器每一步都能有选择地聚焦在输入序列的不同部分,类似人类翻译时会反复回看原文的不同片段。
对比总结
| 基础Seq2Seq | 加入注意力机制 | |
|---|---|---|
| 上下文向量 | 固定的 ,只来自最后隐藏状态 | 动态的 ,每步重新计算 |
| 长序列表现 | 较差,信息易丢失 | 较好,可以直接访问任意位置 |
| 计算复杂度 | 低 | 较高(需计算所有位置的权重) |
| 可解释性 | 差 | 好(注意力权重可可视化) |
- 从“压缩”到“访问”
- Seq2Seq:压缩encoder的所有 到一个固定向量
- Attention:随时访问encoder的所有隐藏状态 ,动态计算
- 从“固定上下文”到“动态上下文”
- 从“链式依赖”到“直接连接”
- RNN路径 → Attention跳跃连接
RNN-Seq2Seq序列到序列模型
https://biscuit0613.github.io/posts/ml/rnn-seq2seq/