Ch06 神经网络基础

Embedding (嵌入 / 向量表示)

概述

Embedding(嵌入)是机器学习中将离散对象(词、句子、文档、图像、用户、商品等)映射到连续稠密向量空间的技术。与传统的 One-Hot 编码(高维稀疏)不同,Embedding 将对象表示为低维稠密向量(通常 128-4096 维),向量之间的距离和方向蕴含了语义信息。Embedding 是现代 NLP、推荐系统和多模态学习的基石,几乎所有深度学习模型都依赖 Embedding 来表示输入。

为什么需要 Embedding

One-Hot 编码的局限

方法维度稀疏性语义信息存储效率
One-Hot词汇量大小(如 50K)极高(99.99% 为 0)无(任意两个词的距离相同)极差
Embedding通常 128-4096稠密(非零值)丰富(相似词向量接近)好

Embedding 的优势

  1. 降维:从高维稀疏空间压缩到低维稠密空间
  2. 语义表示:相似的对象在向量空间中更接近
  3. 可迁移:预训练的 Embedding 可用于下游任务
  4. 可计算:向量间可进行加减乘除运算(如 “国王 - 男人 + 女人 = 女王”)

Embedding 的类型

类型嵌入对象向量维度典型模型应用
Word Embedding词/Token100-768Word2Vec, GloVe, FastText文本分类、NER
Sentence Embedding句子/段落768-4096SBERT, SimCSE, E5, BGE语义搜索、聚类
Document Embedding文档/长文本768-4096BGE, GTE, E5文档检索、RAG
Image Embedding图像512-2048ViT, ResNet, CLIP图像搜索、分类
Multimodal Embedding图文等512-4096CLIP, ImageBind跨模态检索
User/Item Embedding用户/商品32-256CF, Two-Tower推荐系统
Graph Embedding图中的节点64-512Node2Vec, GCN社交网络、知识图谱
Audio Embedding音频片段512-1024Wav2Vec, Whisper语音识别、音乐搜索

核心模型演进

Word Embedding 时代

模型年份核心思想特点
Word2Vec (CBOW/Skip-gram)2013用上下文预测词 / 用词预测上下文浅层网络,训练快
GloVe2014基于全局词共现矩阵分解结合全局统计信息
FastText2016子词(subword)级表示处理 OOV 词,字符 n-gram

Sentence Embedding 时代

模型年份方法特点
BERT (CLS)2018取 [CLS] 输出原生支持,但效果一般
SBERT2019孪生网络 + 池化句向量质量大幅提升
SimCSE2021对比学习 + Dropout 增强简单高效,无需标注
E52023大规模弱监督对比学习通用句向量,多任务
BGE / BGE-M32023-2024多阶段训练 + 多语言中文友好,1024 维
GTE2023大规模对比学习阿里出品,768 维

2026 年的通用 Embedding 模型

模型维度支持语言最大长度特点
BGE-M31024100+8192稠密 + 稀疏混合检索
GTE-Qwen2-7B3584多语言32768基于 LLM 的 Embedding
E5-Mistral-7B4096多语言32768基于 Mistral 的 Embedding
NV-Embed-v24096多语言32768NVIDIA 出品,MTEB 榜首
Jina-Embeddings-v31024多语言8192任务特定 LoRA 适配器

Embedding 的语义空间

在高质量 Embedding 空间中,向量运算具有语义意义:

  • 类比推理:vec(“国王”) - vec(“男人”) + vec(“女人”) ≈ vec(“女王”)
  • 语义相似:cos(出租车, 计程车) ≈ 1
  • 跨语言:vec_en(“cat”) ≈ vec_zh(“猫”)
  • 跨模态:vec_img(“猫的照片”) ≈ vec_text(“猫”)

评估指标

常用 MTEB (Massive Text Embedding Benchmark) 评估 Embedding 模型:

任务类别指标说明
分类Accuracy用 Embedding 做分类器输入
聚类V-Measure聚类质量
配对分类Average Precision判断两个文本是否语义相同
重排序Mean Average Precision对候选文档重排序
检索NDCG@10搜索排序质量
STSSpearman 相关系数语义文本相似度

2026 年趋势

  1. LLM 驱动的 Embedding:基于 7B+ 参数的 LLM 生成 Embedding,质量远超小模型
  2. 多向量 Embedding:ColBERT 式的 token-level 多向量表示,兼顾精度和速度
  3. 任务自适应 Embedding:通过 LoRA 或 prompt 调整 Embedding 以适应特定任务
  4. Matryoshka Embedding:单一模型支持多种维度(如 256/512/1024),灵活权衡精度和效率

相关笔记

来源

关联10

知识图谱

局部图谱 · 点节点跳转