arXiv:2609.01947cs.LGcs.AI2026-09

用强化学习让小模型更好学大模型的排序能力,尤其在新场景下表现更优。

On-Policy Distillation Meets Off-Policy GRPO: Training Compact Instruction-Following Rerankers

论文配图:On-Policy Distillation Meets Off-Policy GRPO: Training Compact Instruction-Following Rerankers
图 1 · 摘自论文原文
  • 先用离线强化学习优化教师模型,再让学生在自己生成的排序上通过软奖励学习。
  • 在MAIR-11数据集上,小模型达0.7670 nDCG@6,比传统方法高4.6点。
  • 适合需要高效部署的指令跟随重排序任务,对模型大小敏感的场景最佳。

紧凑型指令跟随重排序器便于部署,但传统蒸馏方法通常在固定样本上离线模仿教师输出,监督受限于教师观察到的排序空间。本文从强化学习视角重新审视重排序蒸馏,提出两阶段框架:第一阶段,使用88K指令跟随样本和LLM判官反馈,通过离线GRPO强化4B教师模型;第二阶段,1B学生模型自主采样排序,并基于教师生成的软奖励获得指导,实现学生探索与知识迁移的耦合。在分布外情况下性能提升显著:在MAIR-11(11子集,869查询)上,学生模型nDCG@6达0.7670,较离线列表蒸馏提升4.6点。在更全面的MAIR-Full(126任务,9356查询)上,该方法在所有对比蒸馏变体中取得最高任务宏平均得分,nDCG@6为0.6808,MRR@6为0.7865。同时优于两个发布的7B RL训练重排序器。相同第二阶段训练流程可稳定提升三种不同架构的学生模型。在9861查询验证集上,1B模型达到0.7624 nDCG@6,兼具高质量与高效率。

原文摘要 · Abstract (English)

Compact instruction-following rerankers are attractive for deployment, but conventional distillation pipelines typically train students by offline imitation of teacher outputs on a fixed set of examples, constraining supervision to the teacher's observed ranking space. We revisit reranker distillation through the lens of reinforcement learning. We propose a two-stage framework combining off-policy teacher optimization with on-policy student distillation. In Stage 1, a 4B teacher reranker is strengthened with off-policy GRPO using LLM-judge feedback on 88K instruction-following examples. In Stage 2, a compact 1B student samples rankings from its own policy and receives soft teacher-derived rewards on those rankings, coupling student exploration with knowledge transfer. Our strongest gains appear under distribution shift. On MAIR-11, the original 11-subset, 869-query evaluation, the proposed student reaches 0.7670 nDCG@6, outperforming offline listwise KD by +4.6 points. Controlled comparisons against offline pairwise RankNet KD and on-policy GKD show that neither changing the offline distillation objective nor moving teacher-distribution matching on-policy reproduces the performance of reward-based on-policy distillation over student-sampled rankings. The advantage persists on MAIR-Full: across all 126 tasks and 9,356 queries, the proposed method obtains the highest task-macro point estimates among the evaluated distillation variants, reaching 0.6808 nDCG@6 and 0.7865 MRR@6. It also exceeds two released 7B RL-trained rerankers on the comparable MAIR-11 evaluation, while the same Stage 2 training procedure consistently improves three architecturally distinct alternative student backbones. On the 9,861-query validation benchmark, the resulting 1B reranker achieves 0.7624 nDCG@6 while providing a favorable quality-efficiency tradeoff relative to larger alternatives.

重排序强化学习模型蒸馏高效部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。