一、概述

词嵌入(Word Embedding)是将自然语言表示的单词转换为计算机能够理解的向量或矩阵形式的技术。在Word Embedding之前,常用的方法有one-hot、n-gram、co-occurrence matrix,但是他们都有各自的缺点。Word Embedding是一种无监督的方法。

embeding层 维度 embedding维度设置_人工智能

Word Embedding的主要任务是输入收集到的词汇,然后获得它的向量表示,其大概过程如下:

embeding层 维度 embedding维度设置_机器学习_02

Word Embedding不能使用AutoEncoder来做是因为无法为AutoEncoder的输入找到一个包含可以学习的信息的表示。比如使用独热编码作为AutoEncoder的输入,由于独热编码的每一维度之间是相互独立的,且独热编码包含的信息量极少,因此AutoEncoder学不到一些有用的信息。

Word Embedding主要通过训练数据的上下文来学习文本的信息。Word Embedding假设不同的词如果处在同样的上下文位置的话,这两个词就会具有一定的相似性。

embeding层 维度 embedding维度设置_机器学习_03

二、词嵌入的方法

  1. Count based

Count based的方法的主要思想是认为如果两个词embeding层 维度 embedding维度设置_embeding层 维度_04embeding层 维度 embedding维度设置_机器学习_05频繁地同时出现,则其词向量embeding层 维度 embedding维度设置_embeding层 维度_06embeding层 维度 embedding维度设置_embeding层 维度_07会比较接近。

这种方法的典型模型是GloVe模型,这里不做重点介绍。

  1. Prediction based
  • 使用的网络结构

在多层神经网络中输入当前词embeding层 维度 embedding维度设置_神经网络_08的独热编码向量并输出每个维度表示embeding层 维度 embedding维度设置_embeding层 维度_04为每个词的概率的向量,然后取第一个隐藏层的输入作为词的编码向量,其过程如下图所示:

embeding层 维度 embedding维度设置_embeding层 维度_10

将获得的词的编码向量进行可视化以后可以看到类似的词会出现在比较接近的地方:

embeding层 维度 embedding维度设置_神经网络_11

  • 为什么第一个隐藏层的输入可以作为词的编码

在下面的训练文本中输入两个人名网络会预测到相同的输出“宣誓就职”,为了使得神经网络输出去同样的结果,神经网络就需要将两个人名经过权重的处理映射到接近的词向量,然后将这个词向量作为神经网络的输入。

embeding层 维度 embedding维度设置_人工智能_12

  • 共享参数——使用多个词预测下一个词

在实际应用中使用一个词来预测下一次个往往是比较困难的,一个词后面可能出现的词的可能结果有千千万万种,因此通常将多个词同时作为网络的输入,通常需要输入至少10个词才会训练出效果比较好的模型。

以输入两个词为例,假设输入的两个词向量为embeding层 维度 embedding维度设置_深度学习_13embeding层 维度 embedding维度设置_embeding层 维度_14,其长度均为embeding层 维度 embedding维度设置_深度学习_15,第一个隐藏层的输入为embeding层 维度 embedding维度设置_神经网络_16embeding层 维度 embedding维度设置_神经网络_17的长度为embeding层 维度 embedding维度设置_深度学习_18,则两个权重矩阵的形状为embeding层 维度 embedding维度设置_深度学习_19

在实际过程中需要使得embeding层 维度 embedding维度设置_embeding层 维度_20embeding层 维度 embedding维度设置_神经网络_21相等,即embeding层 维度 embedding维度设置_神经网络_22,则embeding层 维度 embedding维度设置_机器学习_23,这就是共享参数。对于为什么需要共享参数有以下两点解释:
①参数共享可以保证同样的词出现在不同位置时不会使得编码的结果不同;
②表示词的独热编码往往维数很高,共享参数可以降低参数的数量。

输入多个词的网络结构如下图所示,其中相同颜色的权重表示共享的参数:

embeding层 维度 embedding维度设置_机器学习_24

在训练时为了保证参数共享,将采用下种方式进行参数更新:

embeding层 维度 embedding维度设置_人工智能_25

以下为一个举例,预测下一个词时使用的是前两个词,然后将输出与预测的词的独热编码的交叉熵作为损失函数,通过最小化损失来训练模型:

embeding层 维度 embedding维度设置_深度学习_26

  • 其他变种

Continuous bag of word (CBOW) model

embeding层 维度 embedding维度设置_人工智能_27

Skip-gram

embeding层 维度 embedding维度设置_神经网络_28

  • 可视化

从训练得到的模型中获得词的编码向量以后进行可视化可以看到一些规律,比如国家和首都之间有一定的联系:

embeding层 维度 embedding维度设置_embeding层 维度_29

动词的三种时态之间有一定的规律:

embeding层 维度 embedding维度设置_机器学习_30

一类动物的词向量与某一特定品种的该类动物的词向量的差,以及一类职业的词向量与某一特定的该类职业的词向量的差会落在各自的位置:

embeding层 维度 embedding维度设置_深度学习_31

根据上述特性可以来回答类比的问题。比如我们知道embeding层 维度 embedding维度设置_人工智能_32,就可以回答以下问题:

embeding层 维度 embedding维度设置_人工智能_33

使embeding层 维度 embedding维度设置_机器学习_34,然后寻找与embeding层 维度 embedding维度设置_embeding层 维度_35最接近的词向量对应的词,大概率会找到答案“Germany”。

三、一些其他应用

  1. 多语种的Embedding

如果分别训练英文的材料和中文的材料,那么分别得到的结果就不会有任何联系,比如“apple”和“苹果”的词向量之间就不会有联系。这是因为词嵌入的方法只根据词的上下文来建立词的编码,如果不将中文和英文的资料放到一起训练就不会使相同意义的中英文词汇的词向量产生任何联系。但是如果我们在中英文分别训练出的词向量的基础上继续训练一个模型,使得中英文相同意义的词映射到同一个位置,就可以得到以下结果,这样的模型可以用来进行类似翻译这样的工作:

embeding层 维度 embedding维度设置_深度学习_36

  1. 对图像的Embedding

首先可以对各种类别的词进行词嵌入获得词的编码向量,然后可以训练一个模型使得该模型的输入是图片,输出一个和该图片类别的词向量同样维度的向量并且使得输出散落在类别的词向量周围。通过这样一种方式就可能实现当这个模型输入一张没有训练过的类别的图片时可以将该图片的编码向量映射到正确的类别的词向量周围,从而解决传统的分类方法只能够识别训练过的类别的图片的局限性。下图是上述过程的一个举例,在这里训练了车辆、马和狗的图片,当猫的图片输入到模型中时可以将其映射到“cat”的词向量周围:

embeding层 维度 embedding维度设置_机器学习_37

  1. 对词序列的Embedding

词序列可以指文档或一段话,可以将不定长的词向量映射到定长的编码向量,这个编码向量就代表了这个序列的语义,这种Embedding的方法可以解决Bag-of-word+AutuEncoder方法的局限性。Bag-of-word+AutuEncoder方法的过程如下图所示:

embeding层 维度 embedding维度设置_深度学习_38

Bag-of-word+AutuEncoder的局限性在于Bag-of-word忽略了词序列中词的顺序所包含的信息。举例来说下面两句话虽然包含同样的词却是完全不同的意思,一句是positive的,另一句是negative的:

embeding层 维度 embedding维度设置_embeding层 维度_39

参考资料

ref:自然语言处理(NLP)之Word Embedding
ref:GloVe模型