用扩散模型预训练,提升长文档检索的上下文捕捉能力。
Diffusion-Pretrained Dense and Contextual Embeddings
- 基于扩散预训练的双向注意力机制,增强段落上下文建模。
- 在多个基准上表现优异,长文档上下文任务创纪录。
- 适合需要高精度长文本检索的生产级搜索系统。
本文介绍pplx-embed系列多语言嵌入模型,采用扩散预训练语言模型作为主干,通过多阶段对比学习实现网络规模检索。借助扩散预训练带来的双向注意力机制,模型能充分捕获段落内双向上下文信息,支持均值池化与后期分块策略,有效保留长文档的全局上下文。发布两个版本:pplx-embed-v1用于标准检索,pplx-embed-context-v1用于融合全局文档上下文的上下文嵌入。pplx-embed-v1在MTEB(多语言版2.0)、MTEB(代码)、MIRACL、BERGEN和ToolRet等基准上表现竞争力;pplx-embed-context-v1在ConTEB基准上创下新纪录。此外,在包含10亿条生产网页构建的真实世界大规模搜索评估中,pplx-embed-v1也展现出强劲性能,验证了其在检索质量与效率至关重要的规模化应用场景中的有效性。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。