Word2Vec（理论、公式和代码）

最新推荐文章于 2025-10-24 10:15:00 发布

原创最新推荐文章于 2025-10-24 10:15:00 发布 · 1.2k 阅读

23 ·

CC 4.0 BY-SA版权

文章标签：

#word2vec #人工智能 #自然语言处理

Word2Vec是一种解决词袋模型局限性的文本表示技术，通过学习词向量捕捉词间语义关系。文章介绍了Skip-Gram模型和CBOW架构，并展示了Python中gensim库的实现及其在NLP任务中的应用。

Word2Vec是一种广泛使用的文本表示技术，可以将词汇表中的每个词转换为一个高维空间中的向量。这些词向量能够捕捉词之间的语义关系，如同义词或语境上相似的词会在向量空间中彼此接近。

提出契机

Word2Vec的提出主要是为了解决传统的词袋模型（Bag of Words）等文本表示方法中无法有效捕捉词之间语义关系的问题。词袋模型将文本表示为词频的向量，这种表示忽略了词序和词之间的语义关系。Word2Vec通过学习词的向量表示，能够捕捉到更加细致的语义信息，并在诸如文本相似性、文本聚类、情感分析等任务中取得了显著的改进。

理论与公式

Word2Vec有两种架构：连续词袋（CBOW）和Skip-Gram。CBOW预测目标词基于上下文词，而Skip-Gram正好相反，它预测上下文词基于目标词。

Skip-Gram模型

Skip-Gram模型的目标是最大化给定目标词的条件下上下文词的出现概率，即最大化下面的目标函数：

$\frac{1}{T} \sum_{t=1}^{T} \sum_{-c \leq j \leq c, j \neq 0} \log p(w_{t+j}|w_t)$

其中， $T$ 是文本中的词汇总数， $c$ 是上下文大小， $w_t$ 是目标词， $w_{t+j}$ 是上下文词。 $p(w_{t+j}|w_t)$ 通常使用softmax函数来计算：

$p(wO∣wI)=exp⁡(vwO′TvwI)∑w=1Wexp⁡(vw′TvwI)p(w_O|w_I) = \frac{\exp({v'_{w_O}}^T v_{w_I})}{\sum_{w=1}^{W} \exp({v'_{w}}^T v_{w_I})}$

这里， $v_{w_I}$ 和 $v'_{w_O}$ 分别是输入词 $w_I$ 和输出词 $w_O$ 的向量表示， $W$ 是词汇表的大小。

代码实现

以下是使用Python中的gensim库来实现Word2Vec模型的示例代码。这个例子假设你已经有了一些预处理过的文本数据。

首先，确保安装了gensim：

pip install gensim

接下来，是Word2Vec模型的训练代码：

from gensim.models import Word2Vec
from gensim.utils import simple_preprocess

# 假设`documents`是已经预处理过的文本数据：一个文本列表
# 使用simple_preprocess进行简单的预处理。实际应用中可能需要更复杂的预处理
processed_docs = [simple_preprocess(doc) for doc in documents]

# 训练Word2Vec模型
model = Word2Vec(sentences=processed_docs, vector_size=100, window=5, min_count=2, workers=4)

# 保存模型以便未来使用
model.save("word2vec.model")

# 使用模型
vector = model.wv['example']  # 获取单词"example"的向量表示
similar_words = model.wv.most_similar('example')  # 找到与"example"最相似的词汇

这段代码首先对文本数据进行了预处理，然后使用Word2Vec函数训练模型。vector_size参数指定向量的维度，window指定了上下文窗口大小，min_count指定了最小词频阈值（即出现次数少于该值的单词会被忽略），workers参数用于设置训练时的线程数，以加快训练速度。

在训练完成后，可以使用model.save方法将模型保存到磁盘上，以便将来使用。你可以通过访问model.wv[word]来获取任何单词的向量表示。most_similar方法可以用来查找与给定单词最相似的词汇。

值得注意的是，gensim的Word2Vec实现提供了丰富的参数来调整模型，如sg（定义使用的训练算法，0表示CBOW，1表示Skip-Gram）、hs（如果为1，则使用层次softmax进行训练，如果为0且负采样negative不为0，则使用负采样）、negative（负采样中的噪声词的数量）等。这些参数可以根据具体的任务和数据集进行调整。

还可以利用预训练的Word2Vec模型，这在处理小数据集或希望利用在更大语料库上学到的语义信息时非常有用。gensim提供了一些预训练模型的接口，例如可以直接加载谷歌的预训练Word2Vec模型：

from gensim.models import KeyedVectors

# 加载预训练的Word2Vec模型
model = KeyedVectors.load_word2vec_format('GoogleNews-vectors-negative300.bin', binary=True)

# 现在可以使用这个模型来获取词向量或者计算相似度等

请确保你有足够的内存来加载预训练的模型，因为这些模型通常很大。例如，谷歌的预训练模型就有3GB大小。

通过使用Word2Vec，我们可以将词转换为向量，这些向量能够在多维空间中表达词之间的语义关系。这种技术已经被广泛应用于各种自然语言处理任务中，如文本分类、情感分析、机器翻译等，大大提高了这些任务的性能。