arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.11151cs.LGcs.CLcs.IR

基于扩散预训练的密集和上下文嵌入

Diffusion-Pretrained Dense and Contextual Embeddings

Sedigheh Eslami, Maksim Gaiduk, Markus Krimmel, Louis Milliken, Bo Wang, Denis Bykov

AI总结:

本文提出pplx-embed模型,通过多阶段对比学习在扩散预训练语言模型基础上实现多语言嵌入,其中pplx-embed-v1在多个基准测试中表现优异,pplx-embed-context-v1在ConTEB基准中创纪录。

摘要:

在本报告中,我们介绍了pplx-embed,一个采用多阶段对比学习在扩散预训练语言模型基础上进行多语言嵌入模型的家族。通过利用基于扩散的预训练进行双向注意力,我们的模型能够捕捉段落中的全面双向上下文,从而使用均值池化和后期分块策略更好地保留长文档中的全局上下文。我们发布了两种模型类型:pplx-embed-v1用于标准检索,以及pplx-embed-context-v1用于上下文化嵌入,后者将全局文档上下文纳入段落表示中。pplx-embed-v1在MTEB(多语言,v2)、MTEB(代码)、MIRACL、BERGEN和ToolRet检索基准测试中表现出竞争力,而pplx-embed-context-v1在ConTEB基准测试中设定了新纪录。除了公开基准外,pplx-embed-v1在我们内部评估套件上表现强劲,专注于由10亿个生产网页构建的现实世界、大规模搜索场景。这些结果验证了模型在大规模检索质量与效率至关重要的生产环境中的有效性。

英文摘要:

In this report, we introduce pplx-embed, a family of multilingual embedding models that employ multi-stage contrastive learning on a diffusion-pretrained language model backbone for web-scale retrieval. By leveraging bidirectional attention through diffusion-based pretraining, our models capture comprehensive bidirectional context within passages, enabling the use of mean pooling and a late chunking strategy to better preserve global context across long documents. We release two model types: pplx-embed-v1 for standard retrieval, and pplx-embed-context-v1 for contextualized embeddings that incorporate global document context into passage representations. pplx-embed-v1 achieves competitive performance on the MTEB(Multilingual, v2), MTEB(Code), MIRACL, BERGEN, and ToolRet retrieval benchmarks, while pplx-embed-context-v1 sets new records on the ConTEB benchmark. Beyond public benchmarks, pplx-embed-v1 demonstrates strong performance on our internal evaluation suite, focusing on real-world, large-scale search scenarios constructed from 1B production web pages. These results validate the models' effectiveness in production environments where retrieval quality and efficiency are critical at scale.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑