arXiv:2505.24782cs.IR2025-05EMNLP被引 13

让文档嵌入更懂上下文,提升检索准确率。

Context is Gold to find the Gold Passage: Evaluating and Training Contextual Document Embeddings

  • 引入上下文感知的对比训练方法,融合文档全局信息
  • 在新基准上显著提升检索效果,尤其在依赖上下文场景
  • 适合需要精准文档检索的研究者和工程师

当前文档检索嵌入方法通常独立编码同一文档的段落(chunk),忽略其余部分的重要上下文信息,影响单个段落表示质量。本文提出ConTEB(Context-aware Text Embedding Benchmark),用于评估模型利用文档级上下文的能力。实验表明,现有顶尖嵌入模型在需依赖上下文的检索任务中表现不佳。为此,我们提出InSeNT(In-sequence Negative Training)——一种新颖的对比后训练方法,结合后期分块池化,显著增强上下文表征学习,同时保持计算效率。该方法在ConTEB上大幅提升检索性能,且不损害基础模型表现。此外,使用该方法嵌入的段落对分块策略和更大检索语料库更具鲁棒性。所有代码与数据已开源:https://github.com/illuin-tech/contextual-embeddings。

原文摘要 · Abstract (English)

A limitation of modern document retrieval embedding methods is that they typically encode passages (chunks) from the same documents independently, often overlooking crucial contextual information from the rest of the document that could greatly improve individual chunk representations. In this work, we introduce ConTEB (Context-aware Text Embedding Benchmark), a benchmark designed to evaluate retrieval models on their ability to leverage document-wide context. Our results show that state-of-the-art embedding models struggle in retrieval scenarios where context is required. To address this limitation, we propose InSeNT (In-sequence Negative Training), a novel contrastive post-training approach which combined with late chunking pooling enhances contextual representation learning while preserving computational efficiency. Our method significantly improves retrieval quality on ConTEB without sacrificing base model performance. We further find chunks embedded with our method are more robust to suboptimal chunking strategies and larger retrieval corpus sizes. We open-source all artifacts at https://github.com/illuin-tech/contextual-embeddings.

文档检索上下文嵌入对比学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。