Sampled Softmax

论文原文：On Using Very Large Target Vocabulary for Neural Machine Translation

我的研究方向是生成式文本自动摘要，在读2016年Nallapati那篇经典论文的时候，遇到了文中所谓的LVT(large vocabulary ‘trick’)问题，比较感兴趣就读了这篇论文的原文。由于基于神经网络的生成式文本摘要的原始模型就是从NMT迁移过来的，所以研究文本摘要有必要学好NMT模型。本文主要参考了这篇博客，但对其中的某些细节作出了更加具体的阐述。

1 问题简介

在NMT和相关系统中，一般情况下，词典构造方法是在target vocabulary中选择频率最高的K个词，超过这个数额的词，就表示为<UNK>。然而，如果一个句子当中的<UNK>;数量过多，性能会极具下降。本文提出了一种近似求解的算法，使得NMT系统可以使用一个较大的词表。并且在WMT14评测上进行了对比试验。

2 问题阐述

现在比较流行的NMT系统都是基于Encoder-Decoder架构的，简单来说，就是输入源语言的句子，Encoder将其编码到一个中间的隐状态h，然后Decoder再根据h生成最终的翻译句子。在训练时，根据机器生成的句子和参考句子最大化条件对数概率。
$${\theta }^{\ast }=\arg \underset{\theta }{\max }\sum _{n=1}^N\sum _{t=1}^{T_N}log(p(y_t^n|y_{<t}^n,x^n))$$
这其中$x$是输入Encoder的序列，$y$是Decoder的输出，$t$代表当前时间步，$Tn$可以理解为句子的总长，$n$为语料对的序号，$N$可以理解为训练数据的总数。这个公式可以这样理解：

最内层是一个条件概率，在输入Encoder的单词序列和当前时间步t之前Decoder输出的单词序列的条件下，输出为target单词的概率。取log概率，然后对每个时间步的条件概率进行累和得到当前语料对输入单词序列条件下输出单词序列是目标句子的概率。然后对所有的训练数据的概率进行累和，得到模型在全部训练集上，给出输入得到输出为目标句子的条件概率。在不考虑过拟合的前提下，显然我们希望这个概率越大越好。所以取max，而此时的参数即为argmax。

其实在对上述公式更原始的叙述中，第二步，对时间步的累和应该为累乘才更合理。在论文Neural Machine Translation by Jointly Learning to Align and Translate中是这样表示的：
$$p(y)=\prod _{t=1}^{T}p(y_t|y_1,...,y,c)$$
这里的$c$指的是Attention机制中的上下文向量(Context Vector)。将这个公式展开你将会得到类似隐式马尔可夫假设的推导过程中的某一步。马尔可夫假设输出只与前$k$个隐层状态有关，这个公式就是前$t-1$个输出$y$都考虑的情况，所以公式很像。可能论文作者只是想表达一下这个意思，这并不是重点。

继续说，Encoder中使用的是双向RNN，本文中使用的GRU作为RNN单元的非线性函数，而Decoder使用的是单向的GRU-RNN。Encoder在时间步t时的隐层状态ht，由前向和反向的时间步t时对应的隐层状态拼合而成。另外有一些论文中说时间步特指Decoder，本文中是通用的。这里的Decoder采用了Attention机制。

Attention机制 最重要的就是上下文向量，它是对Encoder编码后的所有隐层状态的一个加权和。原文中写的是convex sum，让我有些摸不到头脑，我觉得这里用weighted sum更合适些（和刚刚提到的另一篇论文一致）。公式如下：
$$c_i=\sum _{j=1}^{T_x}{\alpha }_{ij}{h}_j$$
其中对于每一个$h_j$的权重${\alpha }_{ij}$计算如下：
$${\alpha }_{ij}=\frac{exp(e_{ij})}{{\sum }_{k=1}^{T_x}exp(e_{ik})}$$
其中：
$$e_{ij}=a(h_t,z_{t-1})$$
其中$a$的具体实现是一个简单的前馈神经网络。权重${\alpha }_{ij}$根据Encoder的第$i$个隐层状态和Decoder的第$j-1$个隐层状态得到，是一个标量。Encoder可以认为是将输入的单词（一般为词向量）进行编码，得到包含有在整个词表上的概率分布信息的隐层状态。Attention对所有隐层状态加权求和，得到的是一个最终的概率分布，而这个概率分布更接近于权重大的那个隐层状态。相当于使得Decoder在解码的某一步，先根据上一步的解码状态和所有的输入隐层状态，得到对输入更关注的点，概率分布就会接近该点的翻译结果。这样就解决了对输入的长期遗忘问题。这个公式使用softmax也是为了使权重区分更加鲜明。

之前我们讨论的目标优化函数中的条件概率就可以表示如下：
$$p(y_t|y_{<t},x)=\frac{1}{Z}exp[w_{target}^T\varphi (y_{t-1},z_t,c_t)+b_{target}]$$
$$Z=\sum _{k:y_k\in V}exp[w_k^T\varphi (y_{t-1},z_t,c_t)+b_k]$$
这里的$\varphi$是一个仿射变换跟着一个非线性激活函数，原始的式子中target那里使用的是$t$，让人和时间步的$t$傻傻分不清。很明显这也是一个softmax，那么问题来了，对于分数的底$Z$，我们需要遍历整个词表才能计算得到。在词表很巨大时，这个Z就成为了模型的瓶颈。

注意这些公式并不是推导得到的，而是构造得到的，个人认为这种神经网络的假设是不能被演绎证明的。另外还要注意，对于词表过大，我们还听说过Beam search。Beam search和这个问题的区别：首先Beam search是用于解码阶段的而不是训练阶段，其次它是用贪婪的方式对全局最优解进行近似，而它计算局部最优的时候还是会面临这个问题。这两个问题是不同层次上的。

3 解决方案

下面是本文的重点。我们已经知道模型的瓶颈就在于，计算Z需要遍历整个词表。为了探讨问题的本质，作者对条件概率的公式取log后求了一下梯度。这里我给出梯度的推导过程：
$$logp(y_t|y_{<t},x)=log\frac{1}{Z}exp[w_{target}^T\varphi (y_{t-1},z_t,c_t)+b_{target}]=w_{target}^T\varphi (y_{t-1},z_t,c_t)+b_{target}-logZ$$
$$▽logp(y_t|y_{<t},x)=▽[w_{target}^T\varphi (y_{t-1},z_t,c_t)+b_{target}]-▽logZ$$
为了表示清晰，我们设：
$$\epsilon (y_j)=w_j^T\varphi (y_{t-1},z_t,c_t)+b_j$$
（上式在原论文中下标有误）那么有：
$$▽logp(y_t|y_{<t},x)=▽\epsilon (y_t)-▽logZ$$
$$=▽\epsilon (y_t)-\frac{1}{Z}▽Z$$
$$=▽\epsilon (y_t)-\frac{1}{Z}\sum _{k:y_k\in V}exp(\epsilon (y_k))▽\epsilon (y_k)$$
$$=▽\epsilon (y_t)-\sum _{k:y_k\in V}\frac{exp(\epsilon (y_k))}{Z}▽\epsilon (y_k)$$
$$=▽\epsilon (y_t)-\sum _{k:y_k\in V}p(y_k|y_{<t},x)▽\epsilon (y_k)$$
如果将$\epsilon$看做能量函数，则上式最终形式的第二项可以看做是能量函数的概率期望，即一个分布在整个词表上的期望意义上的梯度。于是可以用采样的方式，用样本的期望近似总体的期望，作者提出使用一个预先给定的分布$Q$，和一个抽样的子集$V’$来近似。那么上式中的第二项约等于：
$$\sum _{k:y_k\in V^{\prime }}\frac{{\omega }_k}{{\sum }_{k^{\prime }:y_{k^{\prime }}\in V^{\prime }}{\omega }_{k^{\prime }}}▽\epsilon (y_k)$$
其中：
$${\omega }_k=exp[\epsilon (y_k)-logQ(y_k)]$$
这里为了消去上式中的$-log$项，特意设分布$Q$为：
$$Q_i(y_k)=\frac{1}{|V_i^{\prime }|}if{y}_t\in V_i^{\prime }otherwise0$$
于是分母与分子中的$Q$值都相同，被消掉。这样我们就能近似地将那个引起瓶颈的式子表示为：
$$p(y_t|y_{<t},x)=\frac{exp[w_t^T\varphi (y_{t-1},z_t,c_t)+b_t]}{{\sum }_{k:y_k\in V^{\prime }}exp[w_k^T\varphi (y_{t-1},z_t,c_t)+b_k]}$$
关于抽样单词子集$V^{\prime }$的形成，我的理解是：首先设定一个大小阈值threshold t，在训练过程中将Decoder的目标，即给定的reference中的词汇加入V’中，剩余不够阈值的再将原始的target vocabulary中最频繁出现的依次添加进来。这样做就能够保证对目标语句中含有的单词的参数总能够更新。根据Nallapati引用这篇文章时的说法，应该是每一个mini-batch的语料一起训练（参见上文中提到的式子中的n），那么$V^{\prime }$就应该包含这个mini-batch样本中的所有单词，由此可见$V^{\prime }$的大小t也不能设的过小，但相对于几十万肯定是小多了。另外这么做给解码时选取$V^{\prime }$集合造成了困难，所以可以干脆解码时仍遍历词表。

有人说这和word2vec中的负采样有点像，但我觉得负采样还是和dropout更像一些。