接下来我们来了解嵌入模型。从宏观应用场景来看机器学习模型可以划分为几大类别计算机视觉、自然语言处理NLP、音频处理等等每一类下都有多种可用模型。知名平台 Hugging Face 上就能看到各类模型、数据集与应用该平台也是模型资源的常用查找渠道。 我们本次聚焦自然语言处理中的句子相似度任务。在该领域Sentence Transformers 系列下的 all-MiniLM-L6-v2 是主流模型之一。该模型能够将句子、段落映射至 384 维稠密向量空间可用于聚类与语义检索参数量仅 2200 万。 上面这些参数、维度数字是什么意思 参数量可以类比成模型的 “算力 / 认知能力”。你可以把参数理解为存储在 AI 记忆里的习得知识每一个参数都是模型在训练过程中学到的数值用于理解各类语言规律。 2200 万参数代表该模型存有 2200 万个训练习得数值依靠这些数值模型能理解词汇间的关联、句子背后的语义分辨不同概念之间的异同。 我们拿大家熟悉的 GPT 系列模型做对比 我们所用的 GPT-3.5、GPT-4 参数量规模远高于这个 MiniLM 模型。all-MiniLM 仅有 2200 万参数而 GPT-3.5 拥有 1750 亿参数GPT-4 参数量高达 1.8 万亿。 模型文件体积也和参数量成正比MiniLM 模型文件仅 90MB可直接在本地笔记本电脑运行而 GPT-3.5、GPT-4 模型体积分别达到 350GB、3.6TB。 二者适用场景因此完全不同all-MiniLM 非常适合作为嵌入模型满足我们语义检索的需求而 GPT 系列模型主要用于文本生成、逻辑推理类任务。一、先拆解文中三个核心数字384 维向量、2200 万参数、模型文件体积1. 384 维稠密向量嵌入输出维度嵌入模型的核心工作把人类文字句子 / 段落转换成一串固定长度的数字数组这串数字就叫向量Embedding。384 维 转换后固定输出384 个浮点数比如[0.12, -0.45, 0.78, ...]一共 384 个数字作用数字组合代表这句话完整语义语义相近的句子两组数字整体距离会很近语义无关则距离很远稠密向量每个维度都携带语义信息没有大量 0适合做余弦相似度计算、向量库检索、文本聚类维度高低取舍维度越高语义表达越精细但占用内存、检索速度越慢384 维是平衡效果与速度的通用优选。2. 2200 万参数量模型的 “记忆知识库”通俗类比参数 模型训练后学到的所有规则、语言知识全部存储为可计算的数字权重是模型的认知储备。all-MiniLM-L6-v22200 万个独立权重数值训练时通过海量文本自动调整这些数字学会同义词、上下文、语义逻辑参数量越大存储的语言知识、复杂逻辑越多2200 万轻量化只学习语义匹配、短句理解擅长判断两句文字像不像GPT-3.51750 亿、GPT-41.8 万亿海量参数能记忆海量知识、完成长文本创作、复杂数学推理、多轮逻辑对话关键区分参数≠输出向量维度两者完全无关参数模型内部神经网络的权重总数代表模型学习能力上限向量维度模型最终输出给用户的数字数组长度用于下游检索、聚类。3. 模型文件体积权重文件大小模型文件本质就是把所有参数权重打包存储在精度固定时参数量和文件大小近似成正比all-MiniLM-L6-v22200 万参数 → 仅 90MB普通笔记本 CPU 就能本地加载、实时计算嵌入向量GPT-3.51750 亿参数 → 350GB普通电脑硬盘、内存完全放不下只能云端服务器部署GPT-41.8 万亿参数 → 3.6TB需要多卡高性能集群才能运行二、轻量嵌入模型 vs 大生成模型 核心差异对照表表格指标all-MiniLM-L6-v2嵌入模型GPT-3.5/GPT-4大生成模型参数量2200 万1750 亿 / 1.8 万亿模型大小90MB350GB / 3.6TB核心任务语义向量化、相似度计算、检索、聚类文本续写、问答、推理、创作、多轮对话输出结果固定 384 维数字向量无自然文字自由变长自然语言文本运行硬件笔记本 CPU / 轻薄 GPU本地离线跑高端多卡服务器无法单机本地部署能力边界只会判断语义相似不会写文章、做复杂推理具备通识推理、长文本生成、逻辑计算但不擅长快速批量向量检索三、为什么两种模型分工完全不同嵌入模型只做 “语义编码” 单一任务不需要理解复杂逻辑、不需要生成通顺文字仅需捕捉句子浅层语义小参数量完全够用追求低资源、高速度适合百万级文档批量向量化、本地知识库检索。GPT 类生成模型要完成完整语言闭环不仅要理解语义还要组织文字、推理逻辑、调用海量世界知识必须靠万亿级参数存储海量信息但计算成本极高批量给上万条文本生成向量会很慢不适合检索场景。四、补充通俗例子帮你理解向量维度两句话 A小猫在草地上玩耍 B猫咪在草坪上嬉戏 C汽车在公路行驶 模型输出 384 组数字A 和 B 的向量数值整体非常接近A 和 C 差距极大。我们通过数学公式余弦相似度算出分值就能实现输入一句提问在知识库中匹配语义最接近的文档也就是 RAG 检索的底层核心。