摘要:2024 年,美国麻省理工学院团队曾提出“柏拉图表征假说”(Platonic Representation Hypothesis),该假说推测现代神经网络的表征空间正在趋于收敛。即不同的 AI 模型正在趋向于一个统一的现实表征。就像在柏拉图洞穴寓言中囚犯们看到
2024 年,美国麻省理工学院团队曾提出“柏拉图表征假说”(Platonic Representation Hypothesis),该假说推测现代神经网络的表征空间正在趋于收敛。即不同的 AI 模型正在趋向于一个统一的现实表征。就像在柏拉图洞穴寓言中囚犯们看到的影子是现实的投影一样,正在不断迭代的 AI 模型也开始理解投影背后更高维度的现实。当时,相关论文还曾获得前 OpenAI 首席科学家伊利亚·苏茨克维(Ilya Sutskever)的点赞。
2025 年 5 月,在上述基础之上,美国康奈尔大学博士生张瑞杰和所在研究团队提出“强柏拉图表征假说”(Strong Platonic Representation ypothesis),即潜在的通用表征是可以被学习并加以利用的,也能在无需任何编码器或成对数据的情况下实现表征空间之间的转换。
图 | 张瑞杰(来源:https://collinzrj.github.io/)
研究中,他们还提出一种名为 vec2vec 的新方法,这是一种能将文本嵌入从一个向量空间转换到另一个向量空间的方法。其表示这也是第一种无需任何配对数据、编码器或预定义匹配集即可实现上述能力的方法。
也就是说,作为一种无监督方法,vec2vec 能将任意嵌入与“柏拉图表征假说”推测的通用语义结构进行双向转换。
具体来说,vec2vec 能够转换由未知编码器生成的未知文档嵌入,并且对于分布外的输入具有鲁棒性。哪怕模型架构、参数规模和训练数据各不相同,vec2vec 在模型对之间仍能实现高度的余弦相似度。在保留未知嵌入几何结构的同时,该方法能够将其转换到不同空间。
换言之,vec2vec 能够学习“与领域无关”的转换,这一能力主要基于不同嵌入空间中表示相同语义时所通用的几何结构关系。同时,研究团队还证明 vec2vec 转换能够保留足够的输入语义,从而支持属性推理。
此外,他们从一些患者记录和企业邮件中提取了一些敏感疾病信息和其他相关内容,不过他们仅仅访问了文档嵌入,并未接触生成这些嵌入的编码器。
通过本次研究他们发现,更好的转换方法将能实现更高保真度的信息提取,这再次印证了一个事实:嵌入所揭示的信息几乎与其输入内容一样多。
因此,此次发现能为基于文本的模型的“强柏拉图表征假说”提供令人信服的证据。
基于 OpenAI 几年前推出的“对比语言 - 图像预训练”(CLIP,Contrastive Language - Image Pretraining)模型,本次研究的初步实验结果表明,通用几何结构也可用于其他模态。
需要说明的是,本次成果仅仅是表征间转换的一个下限。随着更好、更稳定的学习算法的面世,以及相关架构的改进,预计本次成果将能扩展到更多数据、更多模型家族和更多模态之中。
余弦相似度高达 0.92
据了解,文本嵌入是现代自然语言处理(NLP,Natural Language Processing)的核心,它能为检索、检索增强生成(RAG,Retrieval-Augmented Generation)、分类和聚类等任务提供支持。
对于许多嵌入模型来说,它们是在不同数据集、不同数据打乱方式和不同初始化条件下训练而来的。
文本的嵌入编码了其语义信息:一个优秀的模型会将语义相近的文本,映射到嵌入空间中彼此接近的向量上。由于语义是文本的属性,同一文本的不同嵌入应该编码相同的语义。
然而,在实践中,不同的模型会将文本编码到完全不同且不兼容的向量空间中。
此前,由麻省理工学院团队提出的“柏拉图表征假说”推测:所有足够大的图像模型都具有相同的潜在表征。针对文本模型,本次研究团队提出了该假说的一个更强的建设性版本:文本表征的通用潜在结构是可以被学习的,并能进一步地在无需任何配对数据或编码器的情况下,利用该结构将表征从一个空间转换到另一个空间。
在这项工作中,研究团队证明强柏拉图表征假说在实践中是成立的。通过给定来自两个具有不同架构和训练数据的模型的未配对嵌入示例,他们使用 vec2vec 学习了一个潜在表征,其中这些嵌入几乎完全相同。
研究中,他们从跨语言词嵌入对齐研究和无监督图像翻译研究中汲取灵感。据介绍,vec2vec 使用对抗性损失和循环一致性,来学习将嵌入编码到共享潜在空间中,并能以最小的损失进行解码,这使得无监督转换成为了可能。
同时,研究团队采用了一种对抗性方法,并结合向量空间保持技术,来学习如何将未知嵌入分布映射到已知分布。
通过此,他们发现 vec2vec 转换在目标嵌入空间中与真实向量的余弦相似度高达 0.92,并且在 8000 多个随机排列的 vec2vec 嵌入上实现了完美匹配,而且无需预先访问匹配集合。为了证明上述转换同时保留了“嵌入的相对几何结构”和“底层输入的语义”,他们在完全不了解生成原始嵌入模型的情况下,使用零样本的属性开展推断和反演,并从这些向量中成功提取到了信息。
无监督嵌入转换
据了解,关于嵌入向量集之间的匹配问题或对应问题,已经有大量的研究。但是,这些方法都不适用于本次研究的设置,因为此前研究假设存在由不同编码器从相同输入产生的两组或更多组的嵌入向量。
换句话说,对于每个未知向量来说,必须已经存在另一组不同嵌入空间中的候选向量,其中有一个是正确匹配项。
但是,在实际应用中,很难获得这样的数据库。研究团队表示,他们之所以认为无监督嵌入转换是可行的,是因为它完全取决于更强版本的柏拉图表征假说。
为此,他们提出了如下猜想:当使用相同的目标和模态,但是使用不同数据以及由不同模型架构训练的神经网络,将会收敛到一个通用的潜在空间,从而在无需任何成对对应关系的情况下,即可学习各自表征之间的转换。
在计算机视觉领域,结合了循环一致性和对抗正则化的无监督转换已经取得成功。因此,研究团队在 vec2vec 的设计上,也从这些方法中获得了一些启发。
与此同时,他们希望实现具有循环一致性和不可区分性的嵌入空间转换。与图像不同的是,嵌入向量不具有任何空间偏差。
研究中,研究团队并没有使用卷积神经网络(CNN,Convolutional Neural Network),而是采用了具有残差连接、层归一化和 SiLU 非线性激活函数的多层感知机(MLP,Multilayer Perceptron)。在判别器上则采用了与生成器类似的结构,但是省略了残差连接,以便让对抗学习过程得到简化。
无需任何配对数据,就能学习转换嵌入向量
在数据集上,研究团队使用了由真实用户查询的自然问题(NQ,Natural Questions)数据集,并使用了由维基百科答案训练的数据集。
为了针对信息提取进行评估:
首先,他们使用了 TweetTopic,这是一个由 19 个主题组成的、有着多标签标记的推文数据集。
其次,他们使用了伪重新识别的 MIMIC-III(MIMIC)的随机 8192 个记录子集,并使用了由 2673 个 MedCAT 疾病描述多重标记的患者记录的 MIMIC 数据集的伪重新识别版本。
再次,他们使用了已经倒闭的能源公司安然(Enron)的电子邮件语料库的 50 封随机电子邮件子集,这也是一个未标记的公共数据集。
在模型上,研究团队使用了代表三种规模类别、四种 Transformer 主干架构和两种输出维度的嵌入模型。其中,Granite 是多语言模型,CLIP 是多模态模型。
实验中,他们证明 vec2vec 能够学习一个通用的潜在空间,并证明这个空间保留了所有嵌入的几何结构。这让他们可以将其用作一种文本编码器的通用语言,并且无需任何配对数据就能转换其表征。
实验结果显示,vec2vec 转换能够反映目标空间的几何结构。如下图所示,vec2vec 在模型对之间生成了近乎最优分配的嵌入,实现了高达 0.92 的余弦相似性分数、高达 100% 的 top-1 准确率,且矩阵秩(rank)低至 1。
在相同骨干网络的配对组合中,vec2vec 甚至能够接近于借助先知(oracle)的最优分配方案的性能。vec2vec 生成的嵌入向量,比 naïve 基线更加接近真实值。
在跨主干配对中,vec2vec 在所有指标上都远胜一筹,而基线方法的表现则与随机猜测相差无几。如下图所示,这种性能甚至可以扩展到分布外数据。
如前所述,vec2vec 转换器是在 NQ 数据集上训练的,但是在 X 推文和医疗记录上进行评估时,它仍然表现出较高的余弦相似性、较高的准确率以及较低的矩阵秩。
研究团队指出,vec2vec 始终优于最优任务基线。这些结果表明,本次方法在适应新模态方面具有潜力,特别是 CLIP 的嵌入空间已经成功与其他模态比如热图、音频和深度图建立了连接。
使用 vec2vec 转换来提取信息
研究中,研究团队证明 vec2vec 转换不仅保留了嵌入的几何结构,还保留了足够的语义以便能够支持属性推理。
如下图所示,对 vec2vec 转换进行的属性推理始终优于 naïve 基线,并且往往比理想的零样本基线表现更好。需要说明的是,这一理想基线旨在针对同一空间中的真实文档嵌入和属性嵌入进行推理。由于在本次研究场景中无法获得这些嵌入,因此它是一个假设性基线。
研究团队表示,vec2vec 转换甚至适用于医疗记录的嵌入向量。在同主干配对中,针对转换后的嵌入进行属性推理的表现与 naïve 基线相当,而在跨主干配对中则大幅优于简单基线。
比如,vec2vec 能够保留像“牙槽骨骨膜炎”这类概念的语义,而这类概念从未出现在训练数据中,这证明 vec2vec 的潜在空间确实是一种通用表示。
反演,即重建文本输入。相比属性推断,反演更加具有挑战性。而 vec2vec 转换能够保留足够的语义信息,以至于就算使用那些“原本为标准编码器生成的嵌入”而开发的现成零样本反演方法,也能仅凭转换后的嵌入,来从一些模型对中重建多达 80% 的文档内容。
当然,这些反演并不完美。研究团队表示,他们将在未来针对转换后嵌入开发专门的反演器。总的来说,本次研究证明所有语言模型都会收敛于相同的“通用意义几何”,从而将给向量数据库的发展带来一定影响。
参考资料:
来源:DeepTech深科技一点号