Reranking is fundamental to information retrieval and retrieval-augmented generation, with recent Large Language Models (LLMs) significantly advancing reranking quality. Most current works rely on large-scale LLMs (>7B parameters), presenting high computational costs. Small Language Models (SLMs) offer a promising alternative because of computational efficiency. However, our preliminary quantitative analysis reveals key limitations of SLMs: their representation space is narrow, leading to reduced expressiveness, and they struggle with understanding task prompts without fine-tuning. To address these issues, we introduce a novel two-stage training approach, ProRank, for SLM-based document reranking. We propose using reinforcement learning to improve the understanding of task prompts. Additionally, we introduce fine-grained score learning to enhance representation expressiveness and further improve document reranking quality. Extensive experiments suggest that ProRank consistently outperforms both the most advanced open-source and proprietary reranking models. Notably, our 0.5B ProRank even surpasses powerful LLM reranking models on the BEIR benchmark, establishing that properly trained SLMs can achieve superior document reranking performance while maintaining computational efficiency.
翻译:重排序是信息检索与检索增强生成中的基础环节,近年来大语言模型(LLMs)显著提升了重排序质量。当前大多数工作依赖大规模LLM(参数超过7B),导致计算成本高昂。小语言模型(SLMs)凭借计算高效性成为有潜力的替代方案。然而,我们的定量预分析揭示了SLMs的关键局限性:其表示空间狭窄导致表达能力不足,且未经微调时难以理解任务提示。针对这些问题,我们提出了一种名为ProRank的两阶段训练方法,用于基于SLM的文档重排序。该方法通过强化学习提升对任务提示的理解能力,并引入细粒度分数学习增强表示表达能力,进一步优化文档重排序质量。大量实验表明,ProRank在性能上始终优于最先进的开源和商业重排序模型。值得注意的是,我们的0.5B参数ProRank甚至在BEIR基准测试中超越了强大的LLM重排序模型,证明了经过适当训练的SLM能在保持计算效率的同时实现卓越的文档重排序性能。