DecisionRank:面向决策模型重排序器的同策略蒸馏
摘要
DecisionRank 使用条目空间的同策略蒸馏,将生成式教师模型的排序知识迁移到高效决策模型,在 TREC DL、BEIR、BRIGHT 等检索基准上达到最先进性能,同时显著提升推理速度(最高可达 24.5 倍),仅用 17 亿参数即可超越 270 亿参数的生成式教师模型。
查看缓存全文
缓存时间: 2026/11/03 03:33
论文页面 - DecisionRank:面向决策模型重排序器的同策略蒸馏
来源:https://huggingface.co/papers/2610.42128 作者:
,
,
,
,
,
,
,
,
,
,
,
摘要
DecisionRank 使用条目空间同策略蒸馏,将排序知识从生成式教师模型迁移到高效的决策模型,在实现最先进检索性能的同时显著加快推理速度\。
决策模型 (https://huggingface.co/papers?q=Decision%20models)通过直接预测选择或分数来处理结构化任务,避免自回归解码带来的延迟\。然而,将生成式大语言模型 (https://huggingface.co/papers?q=generative%20LLM)的排序知识 (https://huggingface.co/papers?q=ranking%20knowledge)迁移到决策模型仍然面临挑战:教师模型通过由文档 ID 序列构成的排序列表表达偏好,而学生模型则使用首个词元位置的分数确定排序\。为弥合这一差距,我们提出了 DecisionRank,这是一种通过条目空间同策略蒸馏 (https://huggingface.co/papers?q=on-policy%20distillation)(OPD)训练的决策模型\。条目空间 OPD (https://huggingface.co/papers?q=Item-space%20OPD)通过对学生模型的文档分数 (https://huggingface.co/papers?q=document%20scores)进行排序,形成有序序列,从而构建学生模型的 rollout\。生成式教师模型以得到的排序列表作为输入,并提供条件反馈 (https://huggingface.co/papers?q=conditional%20feedback),直接监督这些分数\。结合离线排序监督,DecisionRank 在 TREC DL、BEIR 和 BRIGHT 等通用及推理密集型检索基准 (https://huggingface.co/papers?q=retrieval%20benchmarks)上均实现了最先进的性能\。在四个基准上,条目空间 OPD (https://huggingface.co/papers?q=item-space%20OPD)始终优于其他训练目标\。即使参数量仅为 1.7B,DecisionRank 在 TREC DL19--23 上也超越了其 27B 参数的生成式教师模型\。在参数量为 8B 时,它的性能优于对应的生成式列表排序模型,同时在 TREC DL 上的运行速度快 24.5 倍,在 BRIGHT 上快 16.7 倍\。
查看 arXiv 页面 (https://arxiv.org/abs/2610.42128)查看 PDF (https://arxiv.org/pdf/2610.42128)添加到合集 (https://huggingface.co/login?next=%2Fpapers%2F2610.42128)
在你的智能体中获取这篇论文:
hf papers read 2610\.42128
还没有最新版 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
引用这篇论文的模型0
没有关联这篇论文的模型
在模型的 README.md 中引用 arxiv.org/abs/2610.42128,即可从此页面关联该模型\。
引用这篇论文的数据集0
没有关联这篇论文的数据集
在数据集的 README.md 中引用 arxiv.org/abs/2610.42128,即可从此页面关联该数据集\。
引用这篇论文的 Space0
没有关联这篇论文的 Space
在 Space 的 README.md 中引用 arxiv.org/abs/2610.42128,即可从此页面关联该 Space\。
包含这篇论文的合集0
没有包含这篇论文的合集
将这篇论文添加到合集 (https://huggingface.co/new-collection),即可从此页面关联该合集\。
相似文章
DecisionRank:面向决策模型重排序器的同策略蒸馏
Meta 的 DecisionRank 论文提出了项目空间中的同策略蒸馏方法,将生成式大语言模型教师的排序知识迁移到单令牌决策模型重排序器中,在 TREC DL、BEIR 和 BRIGHT 上取得了最先进的结果,同时运行速度最高可达对应生成式模型的 24.5 倍,并以仅 17 亿参数超越了其 270 亿参数的教师模型。
基于策略的蒸馏与离策略GRPO结合:训练紧凑指令遵循重排序器
本文提出了一种基于强化学习的蒸馏框架,用于训练紧凑的指令遵循重排序器。该框架使用离策略GRPO增强教师模型,并通过基于策略的蒸馏训练学生模型,在分布偏移下展现出优越性能。
提示级蒸馏:一种高效推理的非参数化模型微调替代方案
提示级蒸馏(PLD)从教师模型中提取推理模式,转化为结构化指令用于学生模型的系统提示,在不增加微调开销的情况下提升推理任务性能。
OPRD:在策略表示蒸馏
OPRD提出了一种新的知识蒸馏方法,该方法在策略部署期间跨层对齐学生和教师的隐藏状态,消除了来自词空间KL估计的采样方差。实验表明,OPRD在数学推理基准(AIME 2024/2025、AIMO)上优于输出空间基线,同时速度快1.44倍,内存使用减少54%。
超越模仿:基于推理进展过滤策略蒸馏
论文提出了Reasoning-Progress-Aware Reward Filtering for On-Policy Distillation (R2-OPD) 来解决教师派生奖励与真实推理进展之间的不匹配问题,提高语言模型训练中的推理性能。