【Word2Vec】传统词嵌入矩阵训练方法

目录

  • [1. Word2Vec 简介](#1. Word2Vec 简介)
  • [2. Word2Vec 的训练方法](#2. Word2Vec 的训练方法)
    • [2.1 Skip-Gram模型](#2.1 Skip-Gram模型)
    • [2.2 CBOW(Continuous Bag of Words)模型](#2.2 CBOW(Continuous Bag of Words)模型)
  • [3. Word2Vec 中的词嵌入表示](#3. Word2Vec 中的词嵌入表示)
  • [4. 训练过程中是否使用独热编码?](#4. 训练过程中是否使用独热编码?)

1. Word2Vec 简介

Word2Vec 是一种词嵌入模型,主要通过无监督学习来训练词汇的稠密向量表示。它通过分析大量的语料库,学习到每个词(token)在该语料库中的上下文关系。Word2Vec 使用神经网络模型来生成每个词的嵌入向量,嵌入向量的维度通常远小于词汇表的大小。

2. Word2Vec 的训练方法

Word2Vec 主要有两种训练方式:

  • 连续词袋模型 (Continuous Bag of Words, CBOW
  • 跳字模型(Skip-gram)

无论是 CBOW 还是 Skip-gram ,Word2Vec 都是通过上下文关系来训练词向量的,而不是像 独热编码 那样直接表示每个词。

2.1 Skip-Gram模型

Skip-Gram 模型的目标是,给定一个中心词(target word),预测它的上下文词(context words)。这种方法适用于小数据集,能够更好地捕捉到词汇的稀有用法。

  • 输入:中心词(target word)
  • 输出:上下文词(context words)
  • 例子
    • 如果上下文窗口为 2,给定句子 "The cat sat on the mat" ,在 "sat" 这个词的中心下,模型会尝试预测 "The"、"cat"、"on" 和 "the" 这四个词作为上下文。

训练时,Skip-Gram 模型通过最大化目标词和上下文词的条件概率,更新模型中的参数。

2.2 CBOW(Continuous Bag of Words)模型

CBOW 模型的目标是,给定上下文词(context words),预测目标词(center word)。CBOW 适合较大数据集,计算上通常比 Skip-Gram 更高效。

  • 输入:上下文词(context words)
  • 输出:中心词(target word)
  • 例子
    • 给定上下文窗口为 2,句子 "The cat sat on the mat" ,在上下文词 "The""cat""on""the" 下,CBOW 模型会尝试预测 "sat" 作为中心词。

在训练时,CBOW 模型通过最大化目标词的条件概率来更新参数。

联想成物理中的向量分量合并与分解的操作即可

3. Word2Vec 中的词嵌入表示

在 Word2Vec 中,输入词汇并不直接使用 独热编码 ,而是通过 嵌入矩阵(embedding matrix)将每个词映射到一个稠密向量空间。这些向量是在训练过程中学习到的。

  • 独热编码 :独热编码(One-Hot Encoding)是将每个词表示为一个维度为词汇表大小的稀疏向量,其中只有对应词的位置是 1,其他位置是 0。例如,如果词汇表大小是 5,"dog" 在词汇表中的位置是 3,那么它的独热编码表示为 [0, 0, 1, 0, 0]。这种表示方式通常会导致非常稀疏的高维向量,维度非常大,并且无法有效地捕捉词与词之间的关系。

  • 词嵌入 :而在 Word2Vec 中,每个词会被映射到一个固定维度的 密集向量(通常是 50 到 300 维)。这些向量表示了词与词之间的语义关系,例如,"king" 和 "queen" 会有相似的向量表示。

Word2Vec 的训练方式实际上是通过 嵌入矩阵 来进行的。在训练过程中,模型会根据上下文词来调整每个词的向量,使得在语义空间中,相似的词会有相似的向量。

4. 训练过程中是否使用独热编码?

是的,在 Word2Vec 中,一般会先将词转换成独热编码(one-hot encoding)向量,再通过嵌入矩阵转换成稠密的低维嵌入向量。

这里是一个简化的流程:

  1. 独热编码 :假设词汇表大小为 V,每个词用一个 V 维的独热编码向量表示。例如,如果"喜欢"是词汇表中的第 i 个词,则它的独热编码在第 i 个位置为 1,其他位置为 0。

  2. 嵌入矩阵 :设嵌入矩阵 W 的维度是 V × d,其中 d 是嵌入的维度。嵌入矩阵 W 存储了每个词的嵌入向量(每行对应词汇表中的一个词)。

  3. 嵌入向量 :将独热编码向量 o 乘以嵌入矩阵 W,得到词的嵌入向量 e。计算过程为 e = o × W,结果就是一个 d 维的嵌入向量。

这样,每个词的独热编码就被映射到一个稠密、低维的嵌入空间中,用以捕获词与词之间的语义关系。

相关推荐
葫三生1 小时前
如何评价《论三生原理》在科技界的地位?
人工智能·算法·机器学习·数学建模·量子计算
m0_751336392 小时前
突破性进展:超短等离子体脉冲实现单电子量子干涉,为飞行量子比特奠定基础
人工智能·深度学习·量子计算·材料科学·光子器件·光子学·无线电电子
美狐美颜sdk5 小时前
跨平台直播美颜SDK集成实录:Android/iOS如何适配贴纸功能
android·人工智能·ios·架构·音视频·美颜sdk·第三方美颜sdk
DeepSeek-大模型系统教程5 小时前
推荐 7 个本周 yyds 的 GitHub 项目。
人工智能·ai·语言模型·大模型·github·ai大模型·大模型学习
郭庆汝5 小时前
pytorch、torchvision与python版本对应关系
人工智能·pytorch·python
小雷FansUnion7 小时前
深入理解MCP架构:智能服务编排、上下文管理与动态路由实战
人工智能·架构·大模型·mcp
资讯分享周7 小时前
扣子空间PPT生产力升级:AI智能生成与多模态创作新时代
人工智能·powerpoint
叶子爱分享9 小时前
计算机视觉与图像处理的关系
图像处理·人工智能·计算机视觉
鱼摆摆拜拜9 小时前
第 3 章:神经网络如何学习
人工智能·神经网络·学习
一只鹿鹿鹿9 小时前
信息化项目验收,软件工程评审和检查表单
大数据·人工智能·后端·智慧城市·软件工程