赞
踩
用神经网络来训练语言模型的思想最早由百度 IDL (深度学习研究院)的徐伟提出[1],NNLM(Nerual Network Language Model)是这方面的一个经典模型,具体内容可参考 Bengio 2003年发表在JMLR上的论文。原文地址:
http://jmlr.org/papers/volume3/bengio03a/bengio03a.pdf
与传统的估算
p
(
w
t
∣
w
t
−
(
n
−
1
)
,
.
.
.
w
t
−
1
)
p(w_t|w_{t-(n-1)},...w_{t-1})
p(wt∣wt−(n−1),...wt−1)不同,NNLM模型直接通过一个神经网络结构对n元条件概率进行评估,NNLM模型的基本结构如下:
首先从语料库中搜集一系列长度为n的文本序列 ( w t , w t − 1 . . . w t − ( n − 1 ) ) (w_t,w_{t-1}...w_{t-(n-1)}) (wt,wt−1...wt−(n−1)),然后组成训练集D,我这里的理解是语料库就是我们在特定领域搜集的文本语料,同时还要有一个词典。有了训练数据和词典,下面就来看下模型是怎样进行前向传播的。
这里先对单个语句序列进行计算,也可以说是单个样本,比如: w 1 … w T w_1 …w_T w1…wT 其中 w t ∈ V w_t∈V wt∈V,这里的 V V V是所有单词的集合(即词典), V i V_i Vi表示词典中的第 i 个单词。
NNLM的目标是训练如下模型:
f
(
w
t
,
w
t
−
1
,
.
.
.
,
w
t
−
n
+
2
,
w
t
−
n
+
1
)
=
p
(
w
t
∣
w
1
t
−
1
)
f(w_t,w_{t−1},...,w_{t−n+2},w_{t−n+1})=p(w_t|w_1^{t−1})
f(wt,wt−1,...,wt−n+2,wt−n+1)=p(wt∣w1t−1)
其中
w
t
w_t
wt表示词序列中第t个单词,
w
1
t
−
1
w_1^{t−1}
w1t−1表示从第1个词到第t个词组成的子序列。模型需要满足的约束条件是:
上面模型的意思是当给定一段序列时,由其前面的(t-1)个词预测第n个词的概率。
限制条件一:即是通过网络得到的每个概率值都要大于0。
而对于第二个限制条件:因为我们的神经网络模型最终得到的输出是针对每t-1个词的输入来预测下一个,也即是第t个词是什么。因此模型的实际输出是一个向量,该向量的每一个分量依次对应下一个词为词典中某个词的概率。所以|v|维的概率值中必定有一个最大的概率,而其他的概率较小。
上面是模型的输入和输出,下面说下前向传播的过程:该模型可分为特征映射和计算条件概率分布两部分
特征映射:通过映射矩阵
C
∈
R
∣
V
∣
×
m
C \in R^{|V|×m}
C∈R∣V∣×m将输入的每个词映射为一个特征向量,
C
(
i
)
∈
R
m
C(i) \in R^m
C(i)∈Rm表示词典中第 i 个词对应的特征向量,其中 m 表示特征向量的维度。这里的映射可以是one-hot。然后将通过特征映射得到
C
(
w
t
−
n
+
1
)
,
.
.
.
,
C
(
w
t
−
1
)
C(w_{t−n+1}),...,C(w_{t−1})
C(wt−n+1),...,C(wt−1)合并成一个
(
n
−
1
)
m
(n−1)m
(n−1)m维的向量
(
C
(
w
t
−
n
+
1
)
,
.
.
.
,
C
(
w
t
−
1
)
)
(C(w_{t−n+1}),...,C(w_{t−1}))
(C(wt−n+1),...,C(wt−1)),因为每一个词是m维的,总共有n-1个词,所以总共有
(
n
−
1
)
m
(n−1)m
(n−1)m维。
也可以说是:一个从词汇表V到实数向量空间的映射C。通过这个映射得到每个单词的向量表示。
计算条件概率分布:通过一个函数 g(g是前馈或递归神经网络)将输入的词向量序列 ( C ( w t − n + 1 ) , . . . , C ( w t − 1 ) ) (C(w_{t−n+1}),...,C(w_{t−1})) (C(wt−n+1),...,C(wt−1))转化为一个概率分布 y ∈ R ∣ V ∣ y∈R^{|V|} y∈R∣V∣,所以这里的输出是|v|维的,和词典的维度是相同的。y中第 i 位表示词序列中第 n 个词是 Vi的概率,即: f ( i , w t − 1 , . . . , w t − n + 2 , w t − n + 1 ) = g ( i , C ( w t − n + 1 ) , . . . , C ( w t − 1 ) ) f(i,w_{t−1},...,w_{t−n+2},w_{t−n+1})=g(i,C(w_{t−n+1}),...,C(w_{t−1})) f(i,wt−1,...,wt−n+2,wt−n+1)=g(i,C(wt−n+1),...,C(wt−1))
下面详细介绍下神经网络的结构,网络输出层采用的是softmax函数,如下式所示:
p
(
w
t
∣
w
t
−
1
,
.
.
.
,
w
t
−
n
+
2
,
w
t
−
n
+
1
)
=
e
y
w
t
∑
i
e
y
i
p(w_t|w_{t−1},...,w_{t−n+2},w_{t−n+1})=\frac{e^{y_{w_t}}}{∑_ie^{y_i}}
p(wt∣wt−1,...,wt−n+2,wt−n+1)=∑ieyieywt
其中 y = b + W x + U t a n h ( d + H x ) y=b+Wx+Utanh(d+Hx) y=b+Wx+Utanh(d+Hx),模型的参数 θ = ( b , d , W , U , H , C ) \theta=(b,d,W,U,H,C) θ=(b,d,W,U,H,C)。 x = ( C ( w t − n + 1 ) , . . . , C ( w t − 1 ) ) x=(C(w_{t−n+1}),...,C(w_{t−1})) x=(C(wt−n+1),...,C(wt−1))是神经网络的输入。 W ∈ R ∣ V ∣ × ( n − 1 ) m W \in R^{|V|×(n−1)m} W∈R∣V∣×(n−1)m是可选参数,如果输入层与输出层没有直接相连(如图中绿色虚线所示),则可令W=0。
当输入层和输出层没有直接相连时:
H
∈
R
h
×
(
n
−
1
)
m
H \in R^{h×(n−1)m}
H∈Rh×(n−1)m是输入层到隐含层的权重矩阵,其中h表示隐含层神经元的数目。此时隐藏层运算:
h
×
(
n
−
1
)
m
h×(n−1)m
h×(n−1)m和
(
n
−
1
)
m
×
1
(n-1)m ×1
(n−1)m×1 相乘得到
h
×
1
h×1
h×1
U
∈
R
∣
V
∣
×
h
U∈R^{|V|×h}
U∈R∣V∣×h是隐含层到输出层的权重矩阵。
d
∈
R
h
d\in R^h
d∈Rh和
b
∈
R
∣
V
∣
b \in R^{|V|}
b∈R∣V∣分别是隐含层和输出层的偏置参数。这时的输出层运算:
∣
V
∣
×
h
|V|×h
∣V∣×h和
h
×
1
h×1
h×1相乘得到
∣
V
∣
×
1
|V|×1
∣V∣×1。
以上就是前向传播的过程。
模型的训练目标是最大化以下似然函数:
L
=
1
T
∑
t
l
o
g
f
(
w
t
,
w
t
−
1
,
.
.
.
,
w
t
−
n
+
2
,
w
t
−
n
+
1
;
θ
)
+
R
(
θ
)
L=\frac{1}{T} \sum_t logf(w_t,w_{t−1},...,w_{t−n+2},w_{t−n+1};θ)+R(θ)
L=T1∑tlogf(wt,wt−1,...,wt−n+2,wt−n+1;θ)+R(θ),其中
θ
\theta
θ为模型的所有参数,
R
(
θ
)
R(\theta)
R(θ)为正则化项。
使用梯度下降算法更新参数的过程如下:
θ
←
θ
+
λ
∂
l
o
g
p
(
w
t
∣
w
t
−
1
,
.
.
.
,
w
t
−
n
+
2
,
w
t
−
n
+
1
)
∂
θ
\theta←\theta+ \lambda \frac{∂logp(wt|wt−1,...,wt−n+2,wt−n+1)}{∂θ}
θ←θ+λ∂θ∂logp(wt∣wt−1,...,wt−n+2,wt−n+1),其中
λ
\lambda
λ为步长。
总结:由于NNLM模型使用了低维紧凑的词向量对上文进行表示,这解决了词袋模型带来的数据稀疏、语义鸿沟等问题。显然nnlm是一种更好的n元语言模型;另一方面在相似的上下文语境中,nnlm模型可以预测出相似的目标词,而传统模型无法做到这一点。
参考:
https://blog.csdn.net/u010089444/article/details/52624964?ref=myread
https://www.zhihu.com/question/32275069
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。