AI开发-PyTorch-NLP-One-Hot/Word2Vec/GloVe/FastText

1 需求


2 接口


3 示例

  1. One - Hot Encoding(独热编码)
    • 原理
      • 独热编码是一种简单的词表示方法。对于一个包含个不同单词的词汇表,每个单词被表示为一个长度为的向量。在这个向量中,只有一个元素为 1,表示该单词在词汇表中的位置,其余元素都为 0。例如,词汇表是 "apple", "banana", "cherry","apple" 的独热编码是 1,0,0,"banana" 是 0,1,0,"cherry" 是 0,0,1
    • 优点
      • 简单直接,易于理解和实现。可以很方便地将文本中的单词转换为机器可处理的向量形式,用于模型输入。
    • 缺点
      • 高维度和稀疏性:当词汇表很大时,向量维度会很高且非常稀疏。例如,一个有 10000 个单词的词汇表,每个独热向量有 10000 个元素,只有一个 1,这会占用大量存储空间。
      • 缺乏语义信息:无法表示单词之间的语义关系。所有向量之间都是正交的,"apple" 和 "banana" 在向量空间中的距离与它们和一个随机噪声向量的距离相同。
    • 应用场景
      • 适用于简单的分类任务或在模型初期作为初步的单词表示,比如简单的文本情感分类,在数据预处理阶段将单词转换为独热向量输入简单神经网络。
  2. Word2Vec
    • 原理
      • 是一种基于神经网络的词向量生成模型,有两种主要架构:CBOW(Continuous Bag - Of - Words)和 Skip - Gram。
      • CBOW:以一个句子中的上下文单词作为输入,预测中心单词。例如在句子 "The cat sits on the mat" 中,如果中心单词是 "sits",上下文单词(如 "The"、"cat"、"on"、"the"、"mat")作为输入来预测 "sits"。
      • Skip - Gram:与 CBOW 相反,它以中心单词为输入来预测上下文单词。以同样句子为例,给定 "sits" 预测 "The"、"cat"、"on"、"the"、"mat"。
    • 优点
      • 能够生成低维度的密集词向量,通常在几十到几百维之间,相比独热编码大大减少了存储空间。
      • 词向量蕴含语义信息,语义相似的单词在向量空间中的距离相近。例如,"king" 和 "queen" 在向量空间中的距离会比 "king" 和 "apple" 近,因为前两者语义更相关。
    • 缺点
      • 训练过程比较复杂,需要在大规模语料上进行训练才能得到较好的效果。
      • 对低频词的表示可能不够准确,因为低频词在训练过程中出现次数少,模型学习到的信息有限。
    • 应用场景
      • 几乎应用于所有自然语言处理任务。在文本分类中作为词嵌入层输入神经网络,帮助模型理解语义;在机器翻译中用于词向量对齐,提高翻译质量;在文本生成任务中使生成的句子更符合语义逻辑。
  3. GloVe(Global Vectors for Word Representation)
    • 原理
      • GloVe 基于词的共现矩阵来学习词向量。它利用全局的词共现统计信息,通过对词共现概率矩阵进行分解等操作来得到词向量。例如,统计文本中 "apple" 和 "banana" 共同出现的频率,以及它们与其他单词的共现频率,基于这些统计信息构建模型。
    • 优点
      • 结合了全局的词汇统计信息,能够有效地利用大规模文本数据中的词共现关系。
      • 生成的词向量在语义任务上表现良好,对于词汇的语义关系捕捉能力较强,能够很好地反映单词之间的语义相似性和类比关系。
    • 缺点
      • 训练和优化过程可能比较复杂,需要处理大量的共现矩阵相关的数据和参数。
      • 对于动态变化的词汇或特定领域的新词汇,可能需要重新训练或者调整模型才能得到较好的表示。
    • 应用场景
      • 广泛应用于信息检索、文本分类、文本生成等任务。在信息检索中,可以根据词向量的相似性来检索相关的文档;在文本分类中,帮助分类器更好地理解文本语义,提高分类准确性。
  4. FastText
    • 原理
      • 是 Facebook 开发的一种用于文本分类和词向量生成的工具。它在 Word2Vec 的基础上进行了扩展,将单词看作是由字符级 n - grams 组成的。例如,对于单词 "apple",它会考虑 "ap"、"app"、"ple" 等字符 n - grams。这样即使对于未出现在训练语料中的单词,也可以通过其字符 n - grams 来生成词向量。
    • 优点
      • 能够处理低频词和未登录词(OOV)。由于考虑了字符级信息,对于罕见单词或者新单词可以通过字符组合来生成合理的向量表示。
      • 在文本分类任务中有出色的表现,训练速度相对较快,能够在大规模数据集上高效地进行训练。
    • 缺点
      • 对于长文本,由于考虑了字符 n - grams,可能会导致向量维度较高,计算成本增加。
      • 其词向量表示在语义细节上可能不如 Word2Vec 和 GloVe 精细,因为它更侧重于文本分类和处理未登录词。
    • 应用场景
      • 特别适用于文本分类任务,尤其是在处理有大量低频词和新词汇的文本数据集时非常有效。也可用于一些需要快速生成词向量并进行简单语义处理的场景,如短文本分类、话题分类等。

介绍一下Word2Vec的Skip-gram模型和CBOW模型

如何使用GloVe进行词向量训练?

对比FastText与Word2Vec的优缺点


4 参考资料

相关推荐
金伟API102413 小时前
如何从零打造一个极简的DeepSeek-R1大模型
人工智能·ai
江畔柳前堤13 小时前
大语言模型分布式训练:从并行策略到万卡工程的系统梳理
人工智能·分布式·深度学习·算法·目标检测·机器学习·语言模型
Shockang13 小时前
智能体环路工程实战
人工智能
美摄科技13 小时前
美摄美颜特效SDK Skill:以AI赋能智能视音频新视界
人工智能
Web3_Daisy13 小时前
Pump.fun 与 FOMO 竞争背后的 Meme 市场变局
大数据·人工智能·金融·web3·区块链
AI创界者13 小时前
MiniMax-H3 本地一键部署整合包:8G 显存玩转文图生视频、视频参考、角色替换与超分补帧全流程
人工智能·深度学习
江畔柳前堤14 小时前
HBM:大语言模型时代的「算力血液」——从内存墙到带宽革命的深度拆解
服务器·人工智能·windows·目标检测·语言模型·自然语言处理·软件工程
美摄科技14 小时前
视频一键成片SDK Skill:AI智能分析与语义理解
人工智能
fthux15 小时前
装闭 RenoPit 源码解析(05):FastAPI与Celery如何执行AI装修分析
人工智能·ai·开源·github·open source·renopit
格数致用15 小时前
数据库设计与表结构详解|信息化项目全流程管理系统源码逐行精讲(五)
人工智能·政务·数据库设计·外键约束