现代循环神经网络6:编码器-解码器架构

一、编码器-解码器架构概述

在许多实际应用中(例如机器翻译),我们的输入和输出都是长度可变的序列。传统的神经网络难以直接处理这种变长输入输出问题。编码器-解码器(Encoder-Decoder)架构正是为此而设计的。它由两个主要组件构成:

  1. 编码器(Encoder): 接收一个长度可变的输入序列,并将其转换为一个固定形状的"编码状态"。
  2. 解码器(Decoder): 接收编码状态,并逐步生成长度可变的输出序列。

举个简单例子,对于英语到法语的机器翻译,输入序列可能是:

  • 英文输入:
    X = ( They , are , watching , . ) X = (\text{They}, \text{are}, \text{watching}, \text{.}) X=(They,are,watching,.)

编码器将这个序列编码为一个固定的状态 s s s(可以理解为一个向量),表示为:

  • 编码状态:
    s = f ( X ) s = f(X) s=f(X)

其中, f f f 表示编码器内部的映射函数。然后,解码器基于这个状态 s s s,逐个生成输出词元,得到翻译后的序列:

  • 法文输出:
    Y = ( Ils , regardent , . ) Y = (\text{Ils}, \text{regardent}, \text{.}) Y=(Ils,regardent,.)

如图所示,整个过程可以直观地理解为将一个变长序列"压缩"为一个定长的状态,再由这个状态"解压缩"出另一个变长序列。


二、编码器

编码器的核心任务是将输入序列转换为一个固定形状的编码状态。假设输入序列为
X = ( x 1 , x 2 , ... , x T ) X = (x_1, x_2, \ldots, x_T) X=(x1,x2,...,xT)

那么编码器可以看作是一个函数 f f f,输出一个状态向量
s = f ( X ) s = f(X) s=f(X)

在深度学习中,常用循环神经网络(RNN)、长短时记忆网络(LSTM)或门控循环单元(GRU)来实现编码器,因为它们能够处理序列数据。

下面是一个基于 PyTorch 框架的编码器接口示例代码:

python 复制代码
class Encoder(nn.Module):
    """编码器-解码器架构的基本编码器接口"""

    def __init__(self, **kwargs):
        super(Encoder, self).__init__(**kwargs)

    def forward(self, X, *rgs):
        # X 为输入序列(长度可变)
        # 此处应实现编码逻辑,将 X 转换为固定形状的编码状态
        raise NotImplementedError

在这个接口中,任何继承 Encoder 的模型都需要实现 forward 方法。数学上可以认为这个过程为:
s = f ( X ) s = f(X) s=f(X)

三、解码器

解码器负责将固定形状的编码状态转换为长度可变的输出序列。其核心在于:

  1. 初始化状态:将编码器输出转换为解码器的初始状态。
  2. 逐步生成输出:在每个时间步,解码器根据当前状态和上一步输出生成下一个词元。

例如,在序列生成过程中,我们可以在时间步 t t t 用下面两个公式描述状态更新与输出生成:
h t = tanh ⁡ ( W   y t − 1 + U   h t − 1 + b ) h_t = \tanh(W\, y_{t-1} + U\, h_{t-1} + b) ht=tanh(Wyt−1+Uht−1+b)
y t = s o f t m a x ( V   h t + c ) y_t = \mathrm{softmax}(V\, h_t + c) yt=softmax(Vht+c)

其中:

  • h t h_t ht 表示当前时刻的隐藏状态,
  • y t − 1 y_{t-1} yt−1 是上一步生成的词元,
  • W , U , V W, U, V W,U,V 为参数矩阵,
  • b , c b, c b,c 为偏置项。

下面给出一个 PyTorch 实现的解码器接口示例代码:

python 复制代码
class Decoder(nn.Module):
    """编码器-解码器架构的基本解码器接口"""

    def __init__(self, **kwargs):
        super(Decoder, self).__init__(**kwargs)

    def init_state(self, enc_outputs, *args):
        raise NotImplementedError

    def forward(self, X, state):
        raise NotImplementedError

四、合并编码器和解码器

编码器与解码器虽然是两个独立的组件,但在整个序列转换模型中必须协同工作。整体流程如下:

  1. 编码阶段 :输入序列 X X X 经过编码器处理,生成编码输出 enc_outputs
  2. 状态初始化 :将 enc_outputs 通过解码器的 init_state 方法转换为初始解码状态 dec_state
  3. 解码阶段 :解码器根据输入(如目标序列的部分信息)和 dec_state 逐步生成输出序列。

整个过程可以用下面的公式来描述:
y = D e c o d e r ( d e c _ X , i n i t _ s t a t e ( E n c o d e r ( e n c _ X ) ) ) y = \mathrm{Decoder}(dec\_X, \mathrm{init\_state}(\mathrm{Encoder}(enc\_X))) y=Decoder(dec_X,init_state(Encoder(enc_X)))

下面是一个整合编码器和解码器的 PyTorch 接口示例代码:

python 复制代码
class EncoderDecoder(nn.Module):
    """编码器-解码器架构的基类"""

    def __init__(self, encoder, decoder, **kwargs):
        super(EncoderDecoder, self).__init__(**kwargs)
        self.encoder = encoder
        self.decoder = decoder

    def forward(self, enc_X, dec_X, *args):
        # enc_X:输入序列
        # dec_X:解码器接收的输入(例如目标序列的部分)
        enc_outputs = self.encoder(enc_X, *args)
        dec_state = self.decoder.init_state(enc_outputs, *args)
        return self.decoder(dec_X, dec_state)

五、小结

本文介绍了编码器-解码器架构,这种架构在处理序列转换问题(如机器翻译)中具有广泛的应用。主要内容回顾如下:

  • 编码器:将变长输入序列

    X = ( x 1 , x 2 , ... , x T ) X = (x_1, x_2, \ldots, x_T) X=(x1,x2,...,xT)

    转换为固定形状的编码状态
    s = f ( X ) s = f(X) s=f(X)

  • 解码器:接收编码状态,利用状态更新公式

    h t = tanh ⁡ ( W   y t − 1 + U   h t − 1 + b ) h_t = \tanh(W\, y_{t-1} + U\, h_{t-1} + b) ht=tanh(Wyt−1+Uht−1+b)

    和输出生成公式
    y t = s o f t m a x ( V   h t + c ) y_t = \mathrm{softmax}(V\, h_t + c) yt=softmax(Vht+c)

    逐步生成输出序列。

  • 整合:整个模型通过编码器输出与解码器初始状态的衔接实现完整的序列转换过程,即

    y = D e c o d e r ( d e c _ X , i n i t _ s t a t e ( E n c o d e r ( e n c _ X ) ) ) y = \mathrm{Decoder}(dec\_X, \mathrm{init\_state}(\mathrm{Encoder}(enc\_X))) y=Decoder(dec_X,init_state(Encoder(enc_X)))

这种架构不仅为后续更复杂的循环神经网络(例如带注意力机制的模型)奠定了基础,也为实现机器翻译等任务提供了清晰的模块化设计思路。希望通过本文的讲解,大家能对深度学习中的序列转换模型有更直观的理解,并尝试阅读和实现相关代码。

相关推荐
空堂与归1 小时前
如何用Python实现深度学习?PyTorch框架核心与实践
pytorch·python·深度学习
richard_first1 小时前
第5章 Attention 的思想
人工智能·深度学习·transformer
阿图灵1 小时前
OpenCV 图像操作六件套:读取、裁剪、缩放、旋转、通道分割与保存
图像处理·人工智能·python·深度学习·opencv·计算机视觉
浪兎兎1 小时前
【深度学习】(四)案例:基于PyTorch的全连接神经网络实现手机价格区间预测
pytorch·深度学习·神经网络
sel_92 小时前
【Pytorch】PyTorch 深度学习框架详解:从安装到实战
人工智能·pytorch·深度学习·算法·语言模型
ZPC82102 小时前
moveit2_servo cmakelist 拆解
人工智能·深度学习·机器人
HyperAI超神经11 小时前
【Triton 教程】triton_language.fdiv
人工智能·深度学习·triton
遥感知识服务12 小时前
从局部阈值、双极化到暗地表剔除:NASA OPERA DSWx-S1全球动态水体算法拆解
大数据·人工智能·深度学习·神经网络·算法·机器学习
jeffsonfu13 小时前
循环神经网络(RNN)详解:处理序列数据的自然选择
人工智能·深度学习
Web3&Basketball15 小时前
vLLM部署开源大模型实战:显存、命令与成本核算
人工智能·深度学习·大模型·ai技术·vllm