谈一谈向量嵌入（Vector Embeddings）

发布日期：2024-09-13 06:42:02 浏览次数： 2971

作者：黄家润研究院

微信搜一搜，关注“黄家润研究院”

对于只能处理数字的计算机或者机器来说，区分苹果和梨并不是一件简单的事情。

这时候就需要用到向量嵌入（Vector Embeddings）。它可以将文字、图像等数据转化为计算机可以理解和操作的数值表示形式。

在有关于 Transformer 工作原理的文章中，我也有提到过向量嵌入的概念，我对它做过一些较为简单的解释。感兴趣可以点击阅读：你知道 GPT 代表什么意思吗？Transformer 架构深度解析

什么是向量嵌入

可以把向量嵌入看作是单词或者其他数据片段的数字指纹。就像是有序的数值列表一样，向量嵌入通过一种叫做向量的特定结构来排列数字。

可以把每个向量想象成多维空间里的一个点，这个点的位置就携带了关于所代表的单词或者数据的重要信息。

我们上学的时候就在数学课上学到过有关于向量的概念，它们有方向和大小。同理，向量嵌入也共享这一个基本概念，但是，向量嵌入所运行的空间的维度更多。

比如在上一篇文章RAG 入门指南：从零开始构建一个 RAG 系统（2）中提到的 all-MiniLM-L6-v2 这个 Sentence Transformer 模型，它可以将句子和段落映射到一个 384 维的稠密向量空间。

极高的维度对于捕捉自然语言中的复杂和细微的差别来说非常重要，这里面包含了诸如语气、语境和语法特征等等。也就是说，一个向量不仅能够区分一段文字的字面含义，还能够捕捉到文字的情感。

通过将高维和分类数据转换为低维、连续的表示形式，向量嵌入能够将复杂的数据转换为适合机器学习算法处理的格式。这样的处理不仅能够提高模型的性能和计算效率，还能够保留数据的内在模式。

下面是一张展示了 8 个示例维度的数值范围的表格。

表格中提到的 One-hot 编码是一种将分类数据转换为二进制表示的方法，广泛用于机器学习和深度学习中。

其核心思想是：将一个具有 nnn 个类别的分类变量表示为一个长度为 nnn 的向量，其中只有一个位置的值为 1，其余位置的值为 0。

举个例子，假设我们有三个类别：猫、狗和鸟，我们可以用以下 one-hot 编码表示它们：

每个类别都有自己唯一的向量表示，1 的位置指示当前类别，其余位置为 0。这种编码方式不会赋予类别任何顺序或大小，适用于不具备数值意义的分类数据。

比如按照以上的表格，单词 cat 可能会是以下的向量表示：

[0.9, 0.2, 0.7, 0.3, 1, 0, 0, 0, 0.4, 0.8, 0.9]

而单词 freedom 可能是：

[0.1, 0.8, 0.6, 0.7, 1, 0, 0, 0, 0.7, 0.3, 0.2]

每一个向量就像是独特的标识符，不仅包含了单词的含义，还反映了当前单词与其他单词之间的关系。

含义相似的单词在数值空间中的向量就像是地图上的相邻点一样也会靠的越近。

下面这张散点图很形象的展示了单词的向量嵌入概念。空间中的每一个点都代表一个单词，它的位置由单词的向量嵌入决定。

聚集在一起的蓝色点表示的是与动物相关的单词：cat dog pet animal，而红色的点表示的是与车辆相关的单词：car vehicle。

Word2Vec 是一种可以用于为单词分配向量的 NLP 技术。

它自身是一种机器学习模型，可以基于单词在大规模的语料库中的上下文来学习单词之间的关联。这些学习到的关系会被编码为数值向量，随后就可以应用于各种 NLP 任务。

Word2Vec 通过两种主要架构来捕捉单词关系：CBOW 和 Skip-gram，下面表格展示了它们之间的区别。

嵌入技术的多样性不仅限于单个单词的领域，句子嵌入能够捕捉整个句子的整体含义。将句子表示为密集向量，我们可以衡量不同文本之间的语义相似性。

与单词嵌入是高维空间中的点一样，句子嵌入也是向量。只是说它们通常具有更高的维度，这样才能适应句子级信息的复杂性。

我们可以对这些结果向量进行数学运算，以衡量语义相似性，从而可以执行更复杂的任务，比如信息检索、文本分类和情感分析。

一种简单的方法是对句子中所有单词的嵌入取平均值。虽然这种方法比较基础，但一般能够提供一个不错的基线结果。

为了捕捉更复杂的语义和句法信息，更高级的技术如循环神经网络（RNNs）和基于 Transformer 的模型被广泛应用。

下面这张表格展示了它们的区别。

尽管 RNN 曾在很长一段时间内占据主导地位，但在许多自然语言处理任务中，包括生成句子嵌入，Transformer 模型已经在性能和效率上超越了 RNN。不过，RNN 仍在某些需要顺序处理的特定应用中发挥作用。