Distillation and Refinement of Reasoning in Small Language Models for Document Re-ranking
On the map: LLM reasoning → Knowledge- and tool-augmented reasoning
also in 1 more
Abstract (arXiv)
We present a novel approach for training small language models for reasoning-intensive document ranking that combines knowledge distillation with reinforcement learning optimization. While existing methods often rely on expensive human annotations or large black-box language models, our methodology leverages web data and a teacher LLM to automatically generate high-quality training examples with relevance explanations. By framing document ranking as a reinforcement learning problem and incentivizing explicit reasoning capabilities, we train a compact 3B parameter language model that achieves state-of-the-art performance on the BRIGHT benchmark. Our model ranks third on the leaderboard while using substantially fewer parameters than other approaches, outperforming models that are over 20 times larger. Through extensive experiments, we demonstrate that generating explanations during inference, rather than directly predicting relevance scores, enables more effective reasoning with smaller language models. The self-supervised nature of our method offers a scalable and interpretable solution for modern information retrieval systems.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Knowledge- and tool-augmented reasoning, one of 17 directions in LLM reasoning. Builds on 21 of its 33 references that are in the map. First posted on arXiv 4 Apr 2025.
Builds on
21 of 33 references.
Closest ancestors · 2
Same direction, newest first
- 2024JudgeRank: Leveraging Large Language Models for Reasoning-Intensive Reranking31 Oct
- 2024ReasoningRank: Teaching Student Models to Rank through Reasoning-Based Knowledge Distillation7 Oct
Other references · 19
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2023Qwen Technical Reportcited by 3,344
- 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,605
- 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,253
- 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 723
- 2019Passage Re-ranking with BERTcited by 645
- 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 554
- 2024RewardBench: Evaluating Reward Models for Language Modelingcited by 499
- 2023Is ChatGPT Good at Search? Investigating Large Language Models as Re-Ranking Agentscited by 360
7 more
- 2024Skywork-Reward: Bag of Tricks for Reward Modeling in LLMscited by 327
- 2020Overview of the TREC 2019 deep learning trackcited by 286
- 2024Generative Representational Instruction Tuningcited by 233
- 2019Multi-Stage Document Ranking with BERTcited by 201
- 2023RankZephyr: Effective and Robust Zero-Shot Listwise Reranking is a Breeze!cited by 155
- 2023RankVicuna: Zero-Shot Listwise Document Reranking with Open-Source Large Language Modelscited by 70
- 2025Hypencoder: Hypernetworks for Information Retrievalcited by 7
Built on by
Citing papers per year: 2025: 4; 2026: 2.
6 papers.
- 2025DIVER: A Multi-Stage Approach for Reasoning-intensive Information Retrievalcited by 37
- 2025Rank-K: Test-Time Reasoning for Listwise Rerankingcited by 29
- 2025RaDeR: Reasoning-aware Dense Retrieval Modelscited by 27
- 2026Unlocking Multimodal Document Intelligence: From Current Triumphs to Future Frontiers of Visual Document Retrievalcited by 8
- 2025The Evolution of Reranking Models in Information Retrieval: From Heuristic Methods to Large Language Modelscited by 3
- 2026Fine-tuning Small Language Models as Efficient Enterprise Search Relevance Labelerscited by 0
- 2026Unlocking Multimodal Document Intelligence: From Current Triumphs to Future Frontiers of Visual Document Retrieval23 Feb 2026
- 2026Fine-tuning Small Language Models as Efficient Enterprise Search Relevance Labelers6 Jan 2026
- 2025The Evolution of Reranking Models in Information Retrieval: From Heuristic Methods to Large Language Models18 Dec 2025
- 2025DIVER: A Multi-Stage Approach for Reasoning-intensive Information Retrieval11 Aug 2025
- 2025RaDeR: Reasoning-aware Dense Retrieval Models23 May 2025
- 2025Rank-K: Test-Time Reasoning for Listwise Reranking20 May 2025
Similar, not cited
Close in wording, no citation either way
- 2025Rank-R1: Enhancing Reasoning in LLM-based Document Rerankers via Reinforcement Learning
- 2025DeAR: Dual-Stage Document Reranking with Reasoning Agents via LLM Distillation
- 2025ProRank: Prompt Warmup via Reinforcement Learning for Small Language Models Reranking
- 2025Rethinking Reasoning in Document Ranking: Why Chain-of-Thought Falls Short
- 2025MM-R5: MultiModal Reasoning-Enhanced ReRanker via Reinforcement Learning for Document Retrieval
- 2025DynamicRAG: Leveraging Outputs of Large Language Model as Feedback for Dynamic Reranking in Retrieval-Augmented Generation
- 2025REARANK: Reasoning Re-ranking Agent via Reinforcement Learning
- 2025Optimizing Generative Ranking Relevance via Reinforcement Learning in Xiaohongshu Search