新基准评估文本嵌入模型在事实性、安全性和推理上的短板。
ATEB: Evaluating and Improving Advanced NLP Tasks for Text Embedding Models
- 将事实性、安全性等任务转化为检索问题,利用检索模型优势提升理解能力。
- 单任务微调下,事实性分类提升8%,安全性分类提升13%。
- 适合关注模型深层理解与实际应用的开发者和研究者。
传统文本嵌入基准主要评估语义相似性,但更复杂的NLP任务需要对文本有更深的理解,如事实性、安全性、指令遵循、推理和文档级理解。我们提出一个新基准,用于评估现有信息检索数据混合训练的嵌入模型在这些高级能力上的表现,并揭示其不足。该基准涵盖多样化的任务,模拟真实场景中这些能力至关重要的情况。此外,我们提出一种新方法,将多种任务重新构造成检索任务。通过将安全性或事实性分类问题转化为检索问题,利用检索模型捕捉语义关系的优势,同时推动其深入理解上下文与内容。采用此方法并结合单任务微调,我们在事实性分类上取得8%的性能提升,在安全性分类上提升13%。代码与数据将公开共享。
原文摘要 · Abstract (English)
Traditional text embedding benchmarks primarily evaluate embedding models' capabilities to capture semantic similarity. However, more advanced NLP tasks require a deeper understanding of text, such as safety and factuality. These tasks demand an ability to comprehend and process complex information, often involving the handling of sensitive content, or the verification of factual statements against reliable sources. We introduce a new benchmark designed to assess and highlight the limitations of embedding models trained on existing information retrieval data mixtures on advanced capabilities, which include factuality, safety, instruction following, reasoning and document-level understanding. This benchmark includes a diverse set of tasks that simulate real-world scenarios where these capabilities are critical and leads to identification of the gaps of the currently advanced embedding models. Furthermore, we propose a novel method that reformulates these various tasks as retrieval tasks. By framing tasks like safety or factuality classification as retrieval problems, we leverage the strengths of retrieval models in capturing semantic relationships while also pushing them to develop a deeper understanding of context and content. Using this approach with single-task fine-tuning, we achieved performance gains of 8\% on factuality classification and 13\% on safety classification. Our code and data will be publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。