arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.02404cs.IRcs.AIcs.CLcs.LG
收藏

提升检索性能:一种面向难负例挖掘的集成方法

Enhancing Retrieval Performance: An Ensemble Approach For Hard Negative Mining

Hansa Meghwani

AI总结:

本研究针对信息检索中难负例筛选的难题,提出适用于企业领域数据集的鲁棒难负例挖掘技术,通过交叉编码器同时学习相似性与差异性提升检索性能,为优化RAG、ReAct等大模型系统提供新视角。

摘要:

排序始终是信息检索研究的核心方向之一。检索与排序模型是网页搜索、开放域问答、企业域问答、文本推荐系统等众多应用的基础。通常这类模型基于包含二元相关性标注的三元组训练,每个三元组由1个正例段落和1个负例段落组成。但在实际应用中,尤其是对大量潜在相关段落进行重排序时,需要对相关性有更精细的理解。尽管通过曝光、点击等用户反馈收集正例较为简单,但从数百万甚至数十亿文档的庞大语料库中筛选合适的负例对难度更大,而生成足量负例对是维持模型高质量的必要条件。现有研究已提出多种从大规模语料中选取合适负例对的方法。本研究聚焦于阐释难负例在交叉编码器(cross-encoder)模型训练过程中的关键作用, specifically旨在解释难负例采样相比随机采样带来的性能提升机制。我们针对具有领域特定上下文的企业数据集,提出了一种鲁棒的难负例挖掘技术,可实现交叉编码器重排序模型的高效训练。本研究为优化检索模型提供了全新视角,最终将影响检索增强生成(Retrieval-Augmented Generation, RAG)、推理与行动智能体(Reasoning and Action Agents, ReAct)等先进大语言模型系统的性能。实验结果表明,通过交叉编码器同时学习相似性与差异性,能够有效提升检索系统的性能。

英文摘要:

Ranking consistently emerges as a primary focus in information retrieval research. Retrieval and ranking models serve as the foundation for numerous applications, including web search, open domain QA, enterprise domain QA, and text-based recommender systems. Typically, these models undergo training on triplets consisting of binary relevance assignments, comprising one positive and one negative passage. However, their utilization involves a context where a significantly more nuanced understanding of relevance is necessary, especially when re-ranking a large pool of potentially relevant passages. Although collecting positive examples through user feedback like impressions or clicks is straightforward, identifying suitable negative pairs from a vast pool of possibly millions or even billions of documents possess a greater challenge. Generating a substantial number of negative pairs is often necessary to maintain the high quality of the model. Several approaches have been suggested in literature to tackle the issue of selecting suitable negative pairs from an extensive corpus. This study focuses on explaining the crucial role of hard negatives in the training process of cross-encoder models, specifically aiming to explain the performance gains observed with hard negative sampling compared to random sampling. We have developed a robust hard negative mining technique for efficient training of cross-encoder re-rank models on an enterprise dataset which has domain specific context. We provide a novel perspective to enhance retrieval models, ultimately influencing the performance of advanced LLM systems like Retrieval-Augmented Generation (RAG) and Reasoning and Action Agents (ReAct). The proposed approach demonstrates that learning both similarity and dissimilarity simultaneously with cross-encoders improves performance of retrieval systems.

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

↑