用强化学习让大模型更会推理,小数据也能做出好排名。
Rank-R1: Enhancing Reasoning in LLM-based Document Rerankers via Reinforcement Learning
- 用强化学习训练大模型,让它先推理再排序
- 仅用18%数据达到微调方法的水平,复杂查询更优
- 提升结果可解释性,适合需要透明度的搜索场景
本文提出Rank-R1,一种基于大语言模型的文档重排方法,通过在排序前对用户查询和候选文档进行推理来提升相关性判断。现有方法多依赖提示或微调,而Rank-R1仅需少量相关性标签(无推理监督),结合强化学习增强模型推理能力。实验表明,在TREC DL和BRIGHT数据集上,Rank-R1在域内数据表现与监督微调相当,但仅使用其18%的训练数据;在域外复杂查询任务中,显著优于零样本和监督微调方法,尤其当使用140亿参数模型时。定性分析显示,其推理过程提升了排序结果的可解释性,为搜索引擎呈现与使用带来新可能。
原文摘要 · Abstract (English)
In this paper, we introduce Rank-R1, a novel LLM-based reranker that performs reasoning over both the user query and candidate documents before performing the ranking task. Existing document reranking methods based on large language models (LLMs) typically rely on prompting or fine-tuning LLMs to order or label candidate documents according to their relevance to a query. For Rank-R1, we use a reinforcement learning algorithm along with only a small set of relevance labels (without any reasoning supervision) to enhance the reasoning ability of LLM-based rerankers. Our hypothesis is that adding reasoning capabilities to the rerankers can improve their relevance assessement and ranking capabilities. Our experiments on the TREC DL and BRIGHT datasets show that Rank-R1 is highly effective, especially for complex queries. In particular, we find that Rank-R1 achieves effectiveness on in-domain datasets at par with that of supervised fine-tuning methods, but utilizing only 18\% of the training data used by the fine-tuning methods. We also find that the model largely outperforms zero-shot and supervised fine-tuning when applied to out-of-domain datasets featuring complex queries, especially when a 14B-size model is used. Finally, we qualitatively observe that Rank-R1's reasoning process improves the explainability of the ranking results, opening new opportunities for search engine results presentation and fruition.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。