简单易懂的自动编码器

大家好，又见面了，我是你们的朋友全栈君。

作者：叶虎

编辑：田旭

引言

自动编码器是一种无监督的神经网络模型，它可以学习到输入数据的隐含特征，这称为编码(coding)，同时用学习到的新特征可以重构出原始输入数据，称之为解码(decoding)。从直观上来看，自动编码器可以用于特征降维，类似主成分分析PCA，但是其相比PCA其性能更强，这是由于神经网络模型可以提取更有效的新特征。除了进行特征降维，自动编码器学习到的新特征可以送入有监督学习模型中，所以自动编码器可以起到特征提取器的作用。作为无监督学习模型，自动编码器还可以用于生成与训练样本不同的新数据，这样自动编码器（变分自动编码器，VariationalAutoencoders）就是生成式模型。

本文将会讲述自动编码器的基本原理以及常用的自动编码器模型：堆栈自动编码器(StackedAutoencoder)。后序的文章会讲解自动编码器其他模型：去噪自动编码器(DenoisingAutoencoder)，稀疏自动编码器(SparseAutoencoder)以及变分自动编码器。所有的模型都会使用Tensorflow进行编程实现。

自动编码器原理

自动编码器的基本结构如图1所示，包括编码和解码两个过程：

简单易懂的自动编码器

图1自动编码器的编码与解码

自动编码器是将输入进行编码，得到新的特征，并且希望原始的输入能够从新的特征简单易懂的自动编码器重构出来。编码过程如下：

简单易懂的自动编码器

可以看到，和神经网络结构一样，其编码就是线性组合之后加上非线性的激活函数。如果没有非线性的包装，那么自动编码器就和普通的PCA没有本质区别了。利用新的特征，可以对输入简单易懂的自动编码器重构，即解码过程：

简单易懂的自动编码器

我们希望重构出的和简单易懂的自动编码器尽可能一致，可以采用最小化负对数似然的损失函数来训练这个模型：

简单易懂的自动编码器

对于高斯分布的数据，采用均方误差就好，而对于伯努利分布可以采用交叉熵，这个是可以根据似然函数推导出来的。一般情况下，我们会对自动编码器加上一些限制，常用的是使简单易懂的自动编码器，这称为绑定权重（tiedweights），本文所有的自动编码器都加上这个限制。有时候，我们还会给自动编码器加上更多的约束条件，去噪自动编码器以及稀疏自动编码器就属于这种情况，因为大部分时候单纯地重构原始输入并没有什么意义，我们希望自动编码器在近似重构原始输入的情况下能够捕捉到原始输入更有价值的信息。

堆栈自动编码器

前面我们讲了自动编码器的原理，不过所展示的自动编码器只是简答的含有一层，其实可以采用更深层的架构，这就是堆栈自动编码器或者深度自动编码器，本质上就是增加中间特征层数。这里我们以MNIST数据为例来说明自动编码器，建立两个隐含层的自动编码器，如图2所示：

简单易懂的自动编码器

图2堆栈自动编码器架构

对于MNIST来说，其输入是28*28=784维度的特征，这里使用了两个隐含层其维度分别为300和150，可以看到是不断降低特征的维度了。得到的最终编码为150维度的特征，使用这个特征进行反向重构得到重建的特征，我们希望重建特征和原始特征尽量相同。由于MNIST是0,1量，可以采用交叉熵作为损失函数，TF的代码核心代码如下：

（左右滑动，查看完整代码）

n_inputs = 28*28
n_hidden1 = 300
n_hidden2 = 150

# 定义输入占位符：不需要y
X = tf.placeholder(tf.float32, [None, n_inputs])

# 定义训练参数
initializer = tf.contrib.layers.variance_scaling_initializer()
W1 = tf.Variable(initializer([n_inputs, n_hidden1]), name="W1")
b1 = tf.Variable(tf.zeros([n_hidden1,]), name="b1")
W2 = tf.Variable(initializer([n_hidden1, n_hidden2]), name="W2")
b2 = tf.Variable(tf.zeros([n_hidden2,]), name="b2")
W3 = tf.transpose(W2, name="W3")
b3 = tf.Variable(tf.zeros([n_hidden1,]), name="b3")
W4 = tf.transpose(W1, name="W4")
b4 = tf.Variable(tf.zeros([n_inputs,]), name="b4")

# 构建模型
h1 = tf.nn.sigmoid(tf.nn.xw_plus_b(X, W1, b1))
h2 = tf.nn.sigmoid(tf.nn.xw_plus_b(h1, W2, b2))
h3 = tf.nn.sigmoid(tf.nn.xw_plus_b(h2, W3, b3))
outputs = tf.nn.sigmoid(tf.nn.xw_plus_b(h3, W4, b4))

# 定义loss
loss = -tf.reduce_mean(tf.reduce_sum(X * tf.log(outputs) +
                    (1 - X) * tf.log(1 - outputs), axis=1))
train_op = tf.train.AdamOptimizer(1e-02).minimize(loss)

当训练这个模型后，你可以将原始MNIST的数字手写体与重构出的手写体做个比较，如图3所示，上面是原始图片，而下面是重构图片，基本上没有差别了。尽管我们将维度从784降低到了150，得到的新特征还是抓取了原始特征的核心信息。

简单易懂的自动编码器

图3原始图片（上）与重构图片对比（下）

有一点，上面的训练过程是一下子训练完成的，其实对于堆栈编码器来说，有时候会采用逐层训练方式。直白点就是一层一层地训练：先训练X->h1的编码，使h1->X的重构误差最小化；之后再训练h1->h2的编码，使h2->h1的重构误差最小化。其实现代码如下：

（左右滑动，查看完整代码）

# 构建模型
h1 = tf.nn.sigmoid(tf.nn.xw_plus_b(X, W1, b1))
h1_recon = tf.nn.sigmoid(tf.nn.xw_plus_b(h1, W4, b4))
h2 = tf.nn.sigmoid(tf.nn.xw_plus_b(h1, W2, b2))
h2_recon = tf.nn.sigmoid(tf.nn.xw_plus_b(h2, W3, b3))
outputs = tf.nn.sigmoid(tf.nn.xw_plus_b(h2_recon, W4, b4))

learning_rate = 1e-02
# X->h1
with tf.name_scope("layer1"):
    layer1_loss = -tf.reduce_mean(tf.reduce_sum(X * tf.log(h1_recon) +
                            (1-X) * tf.log(1-h1_recon), axis=1))
    layer1_train_op = tf.train.AdamOptimizer(learning_rate).minimize(layer1_loss,
                                            var_list=[W1, b1, b4])

# h1->h2
with tf.name_scope("layer2"):
    layer2_loss = -tf.reduce_mean(tf.reduce_sum(h1 * tf.log(h2_recon) +
                  (1 - h1) * tf.log(1 - h2_recon), axis=1))
    layer2_train_op = tf.train.AdamOptimizer(learning_rate).minimize(layer2_loss,
                      var_list=[W2, b2, b3])