word2vec （CBOW分层softmax负采样）

Posted 2022-04-13 liyuxia713

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了word2vec （CBOW分层softmax负采样）相关的知识，希望对你有一定的参考价值。

本文介绍

wordvec的概念
语言模型训练的两种模型CBOW+skip gram
word2vec 优化的两种方法：层次softmax+负采样
gensim word2vec默认用的模型和方法

未经许可，不要转载。

机器学习的输入都是数字，而NLP都是文字；为了让机器学习应用在NLP上，需要把文字转换为数字，把文字嵌入到数学空间。

1. 词表示：

词的独热表示：onehot （词之间是孤立的）
- onehot:
  - 思想：假设词表大小为N，则每个单字表示为N维向量；每个单字只有1位为1，其他为0；茫茫0海中一个1
  - 缺点：词之间是孤立的；维度太大
词的分布式表示：（能描述词之间的语义关系）
- 基于矩阵的分布式表示
- 基于聚类的分布式表示
- 基于神经网络的分布表示，词嵌入 word embedding
  - 将01表示改为浮点数表述；降维
  - word2vec是用神经网络训练语言模型（NNLM）过程中得到的参数.
其他概念
- 语言模型：就是一段文字成为句子的概率；经常用的ngram, n=2或3

2. NNLM 神经网络语言模型

用神经网络训练语言模型，常见有两种；用上下文预测term(CBOW)，用term预测上下文（skip-gram），结构一般是简单三层网络，一个输入层、一个隐含层、一个输出层。

2.1 CBOW 连续词袋模型

技术图片

变量：词表大小V, 要嵌入到N维空间中，C/2是窗口大小，即上下文取几个term(不算当前term)；- N一般取50~300
步骤：参考自 https://www.zhihu.com/question/44832436
- input layer：窗口内的C个词的onehot表示
- input layer -> hidden layer：（V维表示降到N维表示）
  - 通过权重矩阵\$W_V\\times N\$ ,将 \$V \\times C\$映射为 $N \\times C $
- hidden layer + 激活函数：C个词表示为1个词
  - word2vec中激活函数，用了简单取平均；
- hidden layer -> output layer：1个N维词还原到高维V维中；
  - 通过权重矩阵\$W_N\\times V^'\$ ,将 \$N \\times 1\$映射为 \$V\\times 1\$
- output layer + softmax 激活函数，将值归一化到0~1之间 y
- 用BP+梯度下降优化cost function y和真实y之间的距离，迭代优化参数 \$W、W^'\$
- 收敛result：y是V维向量，每个元素取值0~1, 将最大元素值，还原为onehot编码，就是最终结果了。

2.2 skip-gram 语言模型

思路同CBOW, 只是输入是1个词，输出是C个词；
一般用BP训练得到参数，预测用一次前向传播；

3. word2vec用的优化方法

word2vec结合了CBOW, skip-gram的方法训练得到参数\$W\$, 但在计算中做了很多优化；
可以看到，NNLM计算中，两个问题导致计算量大；

词表维度大；
softmax计算量大；

下面介绍两种优化方法；

3.1 hierarchical softmax 层次softmax

/?hī(?)?r?rkik?l/

huffman: 带权路径长度最短为目标，可以用log(V)长度的编码表示

带权路径长度WPL：
- 图a： 5 * 2 + 7 * 2 + 2 * 2 +13 * 2 = 54
- 图b：5 * 3 + 2 * 3 + 7 * 2 + 13 * 1 = 48 （huffman, 带权路径长度较小)
- 参考自： https://zhuanlan.zhihu.com/p/56139075
层次softmax
softmax需要对每个词语都计算输出概率，并进行归一化，计算量很大；
进行softmax的目的是多分类，那么是否可以转成多个二分类问题呢，如SVM思想？从而引入了层次softmax
- 示意图
- 为什么有效？
  - 1）用huffman编码做词表示
  - 2）把N分类变成了log(N)个2分类。如要预测的term（足球）的编码长度为4，则可以把预测为‘足球‘，转换为4次二分类问题，在每个二分类上用二元逻辑回归的方法（sigmoid）；
  - 3）逻辑回归的二分类中，sigmoid函数导数有很好的性质，\$\\sigma^'(x) = \\sigma(x)(1-\\sigma(x))\$
  - 4）采用随机梯度上升求解二分类，每计算一个样本更新一次误差函数
- 参考自：http://flyrie.top/2018/10/31/Word2vec_Hierarchical_Softmax/
- gensim的word2vec 默认已经不采用分层softmax了, 因为$log_2 1000=10 $也挺大的；如果huffman的根是生僻字，则分类次数更多；
  
  3.2 高频词抽样+负采样

为了解决NNLM中softmax过程的计算性能问题，通常有几种优化技巧：

将常见的单词组合（word pairs）或者词组作为单个“words”来处理
对高频次单词进行抽样来减少训练样本的个数
对优化目标采用“negative sampling”方法，这样每个训练样本的训练只会更新一小部分的模型权重，从而降低计算负担

参考自：https://www.jianshu.com/p/56554a63410f

3.2.1 对高频词抽样

主要思想是少训练没有区分度的高频term；
对于我们在训练原始文本中遇到的每一个单词，它们都有一定概率被我们从文本中删掉，而这个被删除的概率与单词的频率有关。词频越高（stopword），抽样越少；

抽样率 \$P(w_i) = (\\sqrt\\fracZ(w_i)t+1)\\times \\fractZ(w_i) = \\sqrt\\fractZ(w_i) + \\fractZ(w_i)\$ ，其中
- \$Z(w_i)\$是词在语料中的出现概率，反比关系，越高频词抽的越少；
- \$t\$是设定的阈值，正比关系，\$t\$越大，不同频率单词的采样概率差异越大； gensim word2vec中默认值是0.001
- 抽样率直观理解
```
import numpy as np
import matplotlib.pyplot as plt

x = np.array(range(1, 100)) / 1000.0
t1 = 0.001
t2 = 0.0001
y1 = np.power(t1/x, 0.5) + t1/x
y2 = np.power(t2/x, 0.5) + t2/x
plt.plot(x, y1, '-', label='t=0.001')
plt.plot(x, y2, '-', label='t=0.0001')
plt.legend()
```
  3.2.2 负采样 Negative sampling
问题：在NNLM中对于每个样本w, 都要更新一次hidden->output的参数阵\$W^'_N\\times V\$，计算量很大；比如词表大小V=10000, N=300, 那么权重矩阵有300万个参数；
思想：如果每次只更新目标词（正例 positive word）和少数几个负例（negative words）, 那么计算量会显著减少；比如如果我们每次只更新M=1+5个，则更新了6*300=1800, 1800/300万=0.06%；计算量减少了3个数量级。
negative words如何选择：根据词表中出现的概率，出现概率越高，被选作负样本的概率越高；
- \$P(w_i) = \\fracf(w_i)^3/4\\sum_j=0^vf(w_j)^3/4\$, 其中\$f(w_j)\$是出现的概率，3/4是经验值，比线性关系更弱化了些；

4. gensim word2vec默认参数（CBOW+负采样）

默认用了CBOW模型，采用高频词抽样+负采样进行优化；

from gensim.models import Word2Vec
word2vec.Word2Vec(sentences=None, size=100, alpha=0.025, window=5, min_count=5, max_vocab_size=None, sample=0.001, seed=1, workers=3, min_alpha=0.0001, sg=0, hs=0, negative=5, ns_exponent=0.75, cbow_mean=1,  iter=5, null_word=0, trim_rule=None, sorted_vocab=1, batch_words=10000, compute_loss=False, callbacks=(), max_final_vocab=None)