模型目标:进行词的向量化表示,使得向量之间尽可能多地蕴含语义和语法的信息

  • 输入:语料库
  • 输出:词向量

方法概述:首先基于语料库构建词的共现矩阵,然后基于共现矩阵和GloVe模型学习词向量。
学习:GloVe模型

统计共现矩阵

设共现矩阵为X,其元素为Xi,jX_{i,j}
Xi,jX_{i,j}的意义为:在整个语料库中,单词i和单词j共同出现在一个窗口中的次数
举个栗子:
设有语料库:

i love you but you love him i am sad

这个小小的语料库只有1个句子,涉及到7个单词:i、love、you、but、him、am、sad。
如果我们采用一个窗口宽度为5(左右长度都为2)的统计窗口,那么就有以下窗口内容:
学习:GloVe模型
窗口0、1长度小于5是因为中心词左侧内容少于2个,同理窗口8、9长度也小于5。
以窗口5为例说明如何构造共现矩阵:
中心词为love,语境词为but、you、him、i;则执行:
学习:GloVe模型
使用窗口将整个语料库遍历一遍,即可得到共现矩阵X。

使用GloVe模型训练词向量

模型公式

先看模型,代价函数长这个样子:
学习:GloVe模型
viv_ivjv_j是单词i和单词j的词向量,bib_ibjb_j是两个标量(作者定义的偏差项),f是权重函数(具体函数公式及功能下一节介绍),N是词汇表的大小(共现矩阵维度为N∗N)。
可以看到,GloVe模型没有使用神经网络的方法

模型怎么来的

那么作者为什么这么构造模型呢?首先定义几个符号:
学习:GloVe模型
其实就是矩阵单词i那一行的和;
学习:GloVe模型
条件概率,表示单词k出现在单词i语境中的概率;
学习:GloVe模型
两个条件概率的比率。
作者的灵感是这样的:
作者发现,ratioi,j,kratio_{i,j,k}这个指标是有规律的,规律统计在下表:
学习:GloVe模型
很简单的规律,但是有用。

思想:假设我们已经得到了词向量,如果我们用词向量viv_ivjv_jvkv_k通过某种函数计算ratioi,j,kratio_{i,j,k},能够同样得到这样的规律的话,就意味着我们词向量与共现矩阵具有很好的一致性,也就说明我们的词向量中蕴含了共现矩阵中所蕴含的信息

设用词向量viv_ivjv_jvkv_k计算ratioi,j,kratio_{i,j,k}的函数为g(vi,vj,vk)g(v_i,v_j,v_k)(我们先不去管具体的函数形式),那么应该有:
学习:GloVe模型
即:
学习:GloVe模型
即二者应该尽可能地接近;
很容易想到用二者的差方来作为代价函数:
学习:GloVe模型
但是仔细一看,模型中包含3个单词,这就意味着要在N∗N∗N的复杂度上进行计算,太复杂了,最好能再简单点。
现在我们来仔细思考g(vi,vj,vk)g(v_i,v_j,v_k),或许它能帮上忙;

作者的脑洞是这样的:

  • 要考虑单词i和单词j之间的关系,那g(vi,vj,vk)g(v_i,v_j,v_k)中大概要有这么一项吧:vivjv_i−v_j;嗯,合理,在线性空间中考察两个向量的相似性,不失线性地考察,那么vivjv_i−v_j大概是个合理的选择;
  • ratioi,j,kratio_{i,j,k}是个标量,那么g(vi,vj,vk)g(v_i,v_j,v_k)最后应该是个标量啊,虽然其输入都是向量,那內积应该是合理的选择,于是应该有这么一项吧:(vivj)Tvk(v_i−v_j)^Tv_k
  • 然后作者又往(vivj)Tvk(v_i−v_j)^Tv_k的外面套了一层指数运算exp(),得到最终的g(vi,vj,vk)g(v_i,v_j,v_k)=exp((vivj)Tvk(v_i−v_j)^Tv_k);
    最关键的第3步,为什么套了一层exp()?

套上之后,我们的目标是让以下公式尽可能地成立:
学习:GloVe模型
即:
学习:GloVe模型
即:
学习:GloVe模型
即:
学习:GloVe模型
然后就发现找到简化方法了:只需要让上式分子对应相等,分母对应相等,即:
学习:GloVe模型
然而分子分母形式相同,就可以把两者统一考虑了,即:
学习:GloVe模型
本来我们追求:
学习:GloVe模型
现在只需要追求:
学习:GloVe模型
两边取个对数:
学习:GloVe模型
那么代价函数就可以简化为:
学习:GloVe模型
现在只需要在N∗N的复杂度上进行计算,而不是N∗N∗N,现在关于为什么第3步中,外面套一层exp()就清楚了,正是因为套了一层exp(),才使得差形式变成商形式,进而等式两边分子分母对应相等,进而简化模型。

然而,出了点问题。
仔细看这两个式子:
学习:GloVe模型
log(Pi,j)log(P_{i,j})不等于log(Pj,i)log(P_{j,i})但是viTvjv_i^Tv_j等于vjTviv_j^Tv_i;即等式左侧不具有对称性,但是右侧具有对称性。

数学上出了问题。
补救一下好了。
现将代价函数中的条件概率展开:
学习:GloVe模型
即为:
学习:GloVe模型
将其变为:
学习:GloVe模型
即添了一个偏差项bjb_j,并将log(Xi)log(X_i)吸收到偏差项bib_i中。
于是代价函数就变成了:
学习:GloVe模型
具体权重函数应该是怎么样的呢?
首先应该是非减的,其次当词频过高时,权重不应过分增大,作者通过实验确定权重函数为:
学习:GloVe模型
到此,整个模型就介绍完了。

说明

复制链接

相关文章: