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

Vignesh Prabhakar, Jialing Pan, Anil Babu Ankisettipalli


Abstract
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.
Anthology ID:
2026.findings-emnlp.477
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
10001–10027
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.477/
DOI:
Bibkey:
Cite (ACL):
Vignesh Prabhakar, Jialing Pan, and Anil Babu Ankisettipalli. 2026. On-Policy Distillation Meets Off-Policy GRPO: Training Compact Instruction-Following Rerankers. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 10001–10027, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
On-Policy Distillation Meets Off-Policy GRPO: Training Compact Instruction-Following Rerankers (Prabhakar et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.477.pdf