word2vec中CBOW和Skip-Gram训练模型的原理

本文深入解析word2vec的两种模型——CBOW与Skip-gram的工作原理。通过具体例子展示了如何利用上下文词预测中心词(CBOW)及利用中心词预测上下文词(Skip-gram),并介绍了one-hot向量如何转化为低维词向量。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

转载于CBOW模型Skip-Gram模型

前言

word2vec如何将corpus(语料库)的one-hot向量(模型的输入)转换成低维词向量(模型的中间产物,更具体来说是输入权重矩阵),真真切切感受到向量的变化,暂不涉及加速算法。

CBOW模型根据中心词W(t)周围的词来预测中心词:


Skip-gram模型则根据中心词W(t)来预测周围词:


CBOW模型的理解

假设我们现在的Corpus是这一个简单的只有四个单词的document:

{I drink coffee everyday}

我们选coffee作为中心词,window size设为2

也就是说,我们要根据单词"I","drink"和"everyday"来预测一个单词,并且我们希望这个单词是coffee。

输入层:上下文单词的onehot1*4维*3个词

输出层:1*4维的向量(概率表示







任何一个单词的one-hot表示乘以这个矩阵都将得到自己的word embedding。

SKip-Gram模型

首先,我们建立一个10000个词的词典,输入的单词就是一个的10000维one-hot向量,而网络的输出也是一样 10,000维的向量,代表每个词预测的概率



模型小trick

Negative Sampling


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值