卷积自编码器擅长捕获图像等网格状数据的空间层级,但许多重要数据集具有固有的序列结构。时间序列数据(股票价格、传感器读数)、自然语言文本、音频信号和视频都是元素顺序具有重要意义的例子。标准的馈送式自编码器,甚至在固定窗口上运行的卷积自编码器,都难以有效建模这些时间依赖性。
循环神经网络 (neural network)(RNN)可以结合到自编码器架构中,从而形成循环自编码器(RAE)。其基本思路是将编码器和解码器中的馈送层替换为循环层,例如简单RNN、LSTM(长短期记忆)或GRU(门控循环单元 (GRU))。
循环自编码器架构
RAE通常遵循熟悉的编码器-瓶颈-解码器结构,但已适用于序列:
-
编码器: 循环网络(例如LSTM)逐步处理输入序列x=(x1,x2,...,xT)。在每个时间步t,RNN根据当前输入xt和上一个隐藏状态ht−1更新其隐藏状态ht:
ht=fenc(ht−1,xt)
最终的隐藏状态hT(或有时是所有隐藏状态的函数)用作捕获整个输入序列信息的压缩表示或上下文 (context)向量 (vector)z。这个向量z代表瓶颈层。
-
瓶颈: 固定大小的上下文向量z=hT。此向量旨在概括输入序列的基本信息。
-
解码器: 另一个循环网络(通常在架构上与编码器相似)接收上下文向量z并生成输出序列x^=(x^1,x^2,...,x^T′)。解码器通常用上下文向量z初始化(例如,h0′=z),并逐个元素生成序列,通常将先前生成的元素作为下一个步骤的输入反馈(或在训练期间使用教师强制):
ht′=fdec(ht−1′,x^t−1)
x^t=gdec(ht′)
目标通常是重建原始输入序列,因此通常T′=T,目标是x^≈x。
该结构与机器翻译或文本摘要中常用的序列到序列(Seq2Seq)模型非常相似,但在此处以无监督方式应用于表示学习或重建。
循环自编码器图示。编码器处理输入序列(x1,...,xT),生成一个上下文向量z。解码器利用z生成重建序列(x^1,...,x^T)。虚线表示依赖关系或初始化。
LSTM和GRU的优点
尽管简单的RNN可以捕获时间模式,但它们通常受梯度消失或梯度爆炸问题影响,使得学习长距离依赖性变得困难。对于大多数应用,RAE中更青睐LSTM或GRU。它们的门控机制允许它们在更长的时间尺度上选择性地记忆或遗忘信息,从而产生更好的序列数据表示。
实现考量
- 单元选择: LSTM或GRU单元通常比简单RNN更推荐。
- 双向性: 对于编码时(例如特征提取)过去和未来元素的上下文 (context)均有益的任务,编码器中可以使用双向RNN。这会双向处理序列,并连接两个方向的隐藏状态。
"* 变长序列: 序列数据通常具有不同的长度。将序列填充到最大长度以及使用掩码层(以在计算和损失计算期间忽略填充步骤)是必要的。"
- 训练: 教师强制,即解码器使用真实的前一个元素xt−1作为输入而不是生成的x^t−1,通常在训练期间用于稳定学习。课程学习方法可能会逐渐降低使用教师强制的概率。
- 损失函数 (loss function): 重建损失通常在序列中逐元素计算,然后聚合(例如,求和或平均)。对于数值序列,每个时间步的均方误差(MSE)很常见。对于类别序列(如文本),每个时间步的交叉熵损失是适用的。
循环自编码器应用
RAE特别适用于:
- 无监督特征提取: 上下文 (context)向量 (vector)z可以用作变长序列的固定大小特征表示,适合作为下游分类器或聚类算法的输入。
- 时间序列异常检测: 通过在正常序列上训练RAE,可以通过观察未见序列与所学模式的偏离所导致的高重建误差来检测异常。
- 序列压缩/重建: 重建输入序列的核心任务。
- 数据去噪: 类似于去噪自编码器,RAE可以训练从受损输入中重建干净序列。
局限性
基本RAE架构的主要局限是固定大小的瓶颈z。对于非常长的序列,将所有必要信息压缩到这个单一向量 (vector)中可能具有挑战性,可能导致信息丢失。高级Seq2Seq技术,例如注意力机制 (attention mechanism),允许解码器回顾特定的编码器隐藏状态,可以缓解此问题,但这超出了标准RAE的定义范围。
总之,循环自编码器扩展了自编码器理念,通过利用循环神经网络 (neural network) (RNN)的能力来有效建模序列数据。它们为无监督表示学习、异常检测以及涉及时间序列、文本和其他有序数据的重建任务提供了一种有价值的工具。