用大模型生成领域定制的文档排序数据,大幅减少人工标注成本。
QBD-RankedDataGen: Generating Custom Ranked Datasets for Improving Query-By-Document Search Using LLM-Reranking with Reduced Human Effort
- 利用大模型结合专家输入自动生成文档评分与排名
- 在TREC数据集上优化BM25模型参数,提升检索效果
- 适合需要高效构建领域检索数据的研究者与工程师
查询文档(Query-By-Document, QBD)是一种信息检索任务,其中查询为文档,候选结果也为文档,常用于专利匹配、法律案例检索及学术文献回顾等场景。现有方法如关键词搜索和文档嵌入可通过领域特定数据集优化,但构建这些数据集成本高、耗时长。本文提出QBD-RankedDataGen流程,比较多种基于大语言模型(LLM)的方法,通过引入领域专家反馈生成文档得分与排名,并提供可解释性结果以供人工审核。该方法显著降低领域数据构建的人工投入,同时保留足够专家知识用于检索模型调优。我们在Text Retrieval Conference(TREC)的QBD数据集上评估,使用生成数据微调工业级搜索引擎常用算法BM25的参数,验证了其有效性。
原文摘要 · Abstract (English)
The Query-By-Document (QBD) problem is an information retrieval problem where the query is a document, and the retrieved candidates are documents that match the query document, often in a domain or query specific manner. This can be crucial for tasks such as patent matching, legal or compliance case retrieval, and academic literature review. Existing retrieval methods, including keyword search and document embeddings, can be optimized with domain-specific datasets to improve QBD search performance. However, creating these domain-specific datasets is often costly and time-consuming. Our work introduces a process to generate custom QBD-search datasets and compares a set of methods to use in this problem, which we refer to as QBD-RankedDatagen. We provide a comparative analysis of our proposed methods in terms of cost, speed, and the human interface with the domain experts. The methods we compare leverage Large Language Models (LLMs) which can incorporate domain expert input to produce document scores and rankings, as well as explanations for human review. The process and methods for it that we present can significantly reduce human effort in dataset creation for custom domains while still obtaining sufficient expert knowledge for tuning retrieval models. We evaluate our methods on QBD datasets from the Text Retrieval Conference (TREC) and finetune the parameters of the BM25 model -- which is used in many industrial-strength search engines like OpenSearch -- using the generated data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。