NLP基础:n-gram语言模型和神经网络语言模型


语言模型是自然语言处理中的重要技术,假设一段长度为TT的文本中的词依次为w1,w2,,wTw_1, w_2, \ldots, w_T,语言模型将计算该序列的概率:
P(w1,w2,,wT). P(w_1, w_2, \ldots, w_T).
语言模型有助于提升自然语言处理任务的效果,例如在语音识别任务中,给定一段“厨房里食油用完了”的语音,有可能会输出“厨房里食油用完了”和“厨房里石油用完了”这两个读音完全一样的文本序列。合适的语言模型能够判断出前者的概率大于后者的概率,于是可以得到正确的“厨房里食油用完了”这个文本序列。

语言模型的计算

我们可以把文本看作一段离散的时间序列w1,w2,,wTw_1, w_2, \ldots, w_T,假设每个词是按时间先后顺序依次生成的,那么在离散的时间序列中,wtw_t1tT1 \leq t \leq T)可看作在时间步(time step)tt的输出或标签。于是,对于一个句子而言,有:
P(w1,w2,,wT)=t=1TP(wtw1,,wt1). P(w_1, w_2, \ldots, w_T) = \prod_{t=1}^T P(w_t \mid w_1, \ldots, w_{t-1}).
例如,一段含有4个词的文本序列的概率:

P(w1,w2,w3,w4)=P(w1)P(w2w1)P(w3w1,w2)P(w4w1,w2,w3). P(w_1, w_2, w_3, w_4) = P(w_1) P(w_2 \mid w_1) P(w_3 \mid w_1, w_2) P(w_4 \mid w_1, w_2, w_3).

n-gram 语言模型

如果一个句子特别长,那么计算和存储多个词共同出现的概率的复杂度会呈指数级增加。

由此引入 n-gram 语言模型,n-gram 是一种基于统计模型的算法。它的基本思想是将文本里面的内容按照字节进行大小为n的滑动窗口操作,形成了长度为n的字节片段序列,每一个字节片段称为gram。

n-gram 模型基于马尔可夫假设,第n个词的出现只与前面 n-1 个词相关,而与其它任何词都不相关,整句的概率就是各个词出现概率的乘积:
P(w1,w2,,wT)t=1TP(wtwt(n1),,wt1). P(w_1, w_2, \ldots, w_T) \approx \prod_{t=1}^T P(w_t \mid w_{t-(n-1)}, \ldots, w_{t-1}) .
这些概率可以通过直接从语料中统计 n 个词同时出现的次数得到。常用的是 n=2 的Bi-Gram和 n=3 的Tri-Gram。

例如,长度为4的序列w1,w2,w3,w4w_1, w_2, w_3, w_4在Bi-Gram和Tri-Gram中的概率分别为:

P(w1,w2,w3,w4)=P(w1)P(w2w1)P(w3w2)P(w4w3),P(w1,w2,w3,w4)=P(w1)P(w2w1)P(w3w1,w2)P(w4w2,w3). \begin{aligned} P(w_1, w_2, w_3, w_4) &= P(w_1) P(w_2 \mid w_1) P(w_3 \mid w_2) P(w_4 \mid w_3) ,\\ P(w_1, w_2, w_3, w_4) &= P(w_1) P(w_2 \mid w_1) P(w_3 \mid w_1, w_2) P(w_4 \mid w_2, w_3) . \end{aligned}

nn较小时,nn元语法往往并不准确。然而,当nn较大时,nn元语法需要计算并存储大量的词频和多词相邻频率。

对于Bi-Gram而言,有:
p(wtwt1)=c(wt1,wt)wtc(wt1,wt) p(w_t|w_{t-1}) = \frac{c(w_{t-1},w_t)}{\sum_{w_t}c(w_{t-1},w_t)}
t=1t=1时,为了使上式有意义,在句子的开头加上<BOS><BOS>,即对于所有的句子w0=<BOS>w_0 = <BOS>。同时,给每个句子加上一个结束符<EOS><EOS>,加上结束符的目的是使得所有的字符串的概率之和sp(s)=1\sum_sp(s)=1

举个例子,我们有下面的语料数据:
NLP基础:n-gram语言模型和神经网络语言模型

用计算最大似然估计的方法计算概率:
p(wiwi1)=c(wi1wi)wic(wi1wi) p(w_i|w_{i-1}) = \frac{c(w_{i-1}w_i)}{\sum_{w_i}c(w_{i-1}w_i)}
其中,函数 c 表示个数统计。

用计算最大似然估计的方法计算概率p(BROWREADABOOK)p(BROW \quad \quad READ \quad A \quad BOOK),具体过程如下:
P(BROW<BOS>)=c(<BOS>BROW)wc(<BOS>w)=13P(READBROW)=c(BROWREAD)wc(BROWw)=11P(AREAD)=c(READA)wc(READw)=23P(BOOKA)=c(ABOOK)wc(Aw)=12P(<EOS>BOOK)=c(BOOK<EOS>)wc(BOOKw)=12 \begin{aligned} &P(BROW |<BOS>) = \frac{c(<BOS>\quad BROW )}{\sum_{w}c(<BOS> \quad w)} = \frac{1}{3} \\ & P(READ |BROW) = \frac{c(BROW \quad READ )}{\sum_{w}c(BROW \quad w)} = \frac{1}{1} \\ & P(A |READ) = \frac{c(READ \quad A )}{\sum_{w}c(READ \quad w)} = \frac{2}{3} \\ & P(BOOK |A) = \frac{c( A \quad BOOK )}{\sum_{w}c(A \quad w)} = \frac{1}{2} \\ & P(<EOS>|BOOK ) = \frac{c( BOOK \quad <EOS> )}{\sum_{w}c(BOOK \quad w)} = \frac{1}{2} \end{aligned}
所以:
p(BROW  READ  A  BOOK)=P(BROW<BOS>)×P(READBROW)×P(AREAD)×P(BOOKA)×P(<EOS>BOOK)=13×1×23×12×12  0.06 \begin{aligned} &p(BROW \; READ\; A\; BOOK)\\ = &P(BROW |<BOS>)\times P(READ |BROW) \times P(A |READ)\\&\times P(BOOK |A)\times P(<EOS>|BOOK ) \\ =&\frac{1}{3} \times 1 \times\frac{2}{3}\times\frac{1}{2}\times\frac{1}{2}\\ \approx & \;0.06 \end{aligned}

n-gram 平滑技术

在自然语言处理处理中,难免会遇到未登录词(OOV),即测试集中出现了训练集中未出现过的词,或者未出现过的子序列,导致语言模型计算出的概率为0。使用平滑技术的目的就是在条件概率为0时,防止整个句子的概率为0,人为按一定规则,给0条件概率一定的值。常见的平滑方法有:

  • 加法平滑方法
  • 古德-图灵(Good-Turing)估计法
  • Katz平滑方法
  • Jelinek-Mercer平滑方法
  • Witten-Bell平滑方法
  • 绝对减值法

具体内容参考宗成庆老师的《统计自然语言处理 第二版》

神经网络语言模型(NNLM)

基本思想

神经网络语言模型先给每个词赋予一个词向量,利用神经网络去建模当前词出现的概率与其前 n-1 个词之间的约束关系,即通过神经网络去学习词的向量表示。结构如下图所示:

NLP基础:n-gram语言模型和神经网络语言模型

具体而言,假设当前词出现的概率只依赖于前n1n-1个词。即:
p(wtw1wt1)=p(wtwtn+1wt1) p(w_t|w_1\ldots w_{t-1}) = p(w_t|w_{t-n+1}\ldots w_{t-1})
假设VV表示所有N个词的集合,wtVw_t \in V,存在一个V×m|V|\times m的参数矩阵C,矩阵的每一行代表每一个词的特征向量,m代表每个特征向量由m个维度。C(wt)C(w_t)表示第 t 个词wtw_t的对应的特征向量。

如上图所示,将句子的前 n-1 个词特征对应的向量C(wtn+1)C(wt1)C(w_{t-n+1})\ldots C(w_{t-1})作为神经网络的输入,输出当前的词是wtw_t的概率。计算过程如下:
y=Wx+Utanh(d+Hx)+b y = W x + U tanh(d+Hx) + b
其中:
x=(C(wt1),C(wt2),,C(wtn+1)) x = (C(w_{t-1}),C(w_{t-2}),\ldots ,C(w_{t-n+1}))
经过上式得到yw=(yw,1,yw,2,,yw,N)Ty_w = (y_{w,1},y_{w,2},\ldots,y_{w,N})^T ,其中,将ywy_w进行softmax后,yw,iy_{w,i} 表示当上下文为context(w)context(w)时,下一个词恰好是词典中第ii个词的概率,即:
P(wcontext(w))=eyw,iwi=1Neyw,i P(w|context(w)) = \frac{e^{y_{w,i_w}}}{\sum_{i=1}^N e^{y_{w,i}}}
其中,iwi_w表示词在词典中的索引。

计算集中在隐藏层和输出层之间的矩阵向量运算,以及输出层上的softmax归一化运算。该模型的参数是:
θ=(b,d,W,U,C) \theta = (b,d,W,U,C)
损失函数是:
L=1Ttlogp^(wt=iwtn+1wt1)+R(θ) L = -\frac{1}{T} \sum_t \log \hat p(w_t=i|w_{t-n+1}\ldots w_{t-1}) + R(\theta)
$ R(\theta)$是正则化项,用于控制过拟合。

有了损失函数,我们可以用梯度下降法,在给定学习率η\eta的条件下,进行参数更新:
θθηLθ \theta \leftarrow \theta - \eta \frac{\partial L}{\partial \theta}
直至收敛,得到一组最优参数。

神经网络语言模型小结

只要词的表征足够好,就能具有比 n-gram 更好的泛化能力,从而很大程度地降低了数据稀疏带来的问题。不足之处是仅包含了有限的前文信息。

优点:(1) 长距离依赖,具有更强的约束性;(2) 避免了数据稀疏所带来的OOV问题;(3) 好的词表征能够提高模型泛化能力。

缺点:(1) 模型训练时间长;(2) 可解释性较差。

神经网络语言模型与 n-gram 模型相比,

1,相似的词具有相似的向量。对 n-gram 模型而言,如果语料中S1=“A dog is running in the room”出现了10000次,而S2=“A cat is running in the room”只出现了1次,p(S1)肯定会远大于p(S2)。而实际上,dog 和 cat 是相似的存在,神经网络语言模型所得到的词向量距离会比较近,表示两个词十分相似。

2,用向量表示的词自带平滑化功能(由p(w∣Context(w))∈(0,1)p(w|Context(w))∈(0,1)p(w∣Context(w))∈(0,1)不会为零),不再需要像 n-gram 那样进行额外处理了。

语言模型评价指标—困惑度

通常用困惑度作为语言模型的评价指标。

设存在文本序列W=(w1,w2,,wN)W=(w_1,w_2,\ldots,w_N),则每个词的平均交叉熵为:
H(W)=1NlogP(w1,w2,,wN) H(W) = -\frac{1}{N}\log P(w_1,w_2,\ldots,w_N)
显然,交叉熵越小,得到的概率模型越接近真实分布,进一步我们定义困惑度为:
Preplexity(W)=2H(W)=1P(w1,w2,,wN)N Preplexity(W) = 2^{H(W)} = \sqrt[N]{\frac{1}{ P(w_1,w_2,\ldots,w_N)}}
由公式可知:困惑度越小,说明所建模的语言模型越精确。

参考文章

  1. A Neural Probabilistic Language Model

  2. 深入理解语言模型 Language Model

  3. 6.1语言模型

  4. 语言模型基础