论文

DecisionRank对齐候选排序与教师反馈,训练单次前向重排模型

DecisionRank: On-Policy Distillation for Decision-Model Rerankers

模型优化模型推理判别式推理与决策蒸馏Jev

摘要

决策模型通过直接预测选择或分数来定位结构化任务,避免自回归解码延迟。然而,将生成式 LLM 的排名知识转移到决策模型仍然具有挑战性:教师通过文档 ID 序列的排名列表表达偏好,而学生则使用第一个 token 位置的分数来确定排名。为了弥补这一差距,我们引入了 DecisionRank,这是一种通过条目空间同策略蒸馏 (OPD) 训练的决策模型。条目空间 OPD 通过将文档分数排序为排名序列来构建学生的 rollout。生成式教师将生成的排名列表作为输入,并提供条件反馈来直接监督这些分数。与离线排名监督相结合,DecisionRank 在 TREC DL、BEIR 和 BRIGHT 等一般检索和推理密集型检索基准上均实现了最先进的性能。在四个基准中,条目空间 OPD 始终优于其他训练目标。即使在 1.7B 参数下,DecisionRank 在 TREC DL19--23 上也超过了其 27B 生成式教师模型。在 8B 时,它的性能优于对应的生成式列表排序模型,同时在 TREC DL 上运行快 24.5 倍,在 BRIGHT 上运行快 16.7 倍。

DecisionRank以候选空间在策略蒸馏训练单次前向重排决策:论文配图
图1:DecisionRank通过蒸馏实现有效、高效的单token重排。所有方法均对TREC DL19—23上同一组BM25前100名候选重排。(a)蒸馏显著改善全部三种学生规模的排序准确率,每种学生都超过27B生成式教师(62.91)。箭头为相对零样本评分的nDCG@10×100绝对增益。(b)DecisionRank学生相对其他重排器取得较好的准确率—时延权衡。灰色三角为Jev,采用服务器报告时延的中位数。Qwen3-RR指Qwen3-Reranker。