DecisionRank:面向决策模型重排序器的同策略蒸馏

Hugging Face Daily Papers 论文

摘要

DecisionRank 使用条目空间的同策略蒸馏,将生成式教师模型的排序知识迁移到高效决策模型,在 TREC DL、BEIR、BRIGHT 等检索基准上达到最先进性能,同时显著提升推理速度(最高可达 24.5 倍),仅用 17 亿参数即可超越 270 亿参数的生成式教师模型。

决策模型通过直接预测选项或分数来处理结构化任务,从而避免自回归解码带来的延迟。然而,将生成式大语言模型的排序知识迁移到决策模型仍然具有挑战性:教师模型通过由文档 ID 序列构成的排序列表表达偏好,而学生模型则使用首个 token 位置的分数来确定排序。为弥合这一差距,我们提出了 DecisionRank,这是一种通过项目空间同策略蒸馏(OPD)训练的决策模型。项目空间 OPD 将学生模型的文档分数按序排列,构成排序序列,以此构建学生模型的 rollout。生成式教师模型将得到的排序列表作为输入,并提供条件反馈,直接监督这些分数。结合离线排序监督,DecisionRank 在 TREC DL、BEIR 和 BRIGHT 等通用及推理密集型检索基准上均达到了最先进的性能。在四个基准上,项目空间 OPD 始终优于其他训练目标。即使仅有 17 亿参数,DecisionRank 在 TREC DL19--23 上也超过了其 270 亿参数的生成式教师模型。在 80 亿参数规模下,它的性能优于对应的生成式列表级排序模型,同时在 TREC DL 和 BRIGHT 上的运行速度分别达到后者的 24.5 倍和 16.7 倍。
查看原文
查看缓存全文

缓存时间: 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:面向决策模型重排序器的同策略蒸馏

arXiv cs.CL

Meta 的 DecisionRank 论文提出了项目空间中的同策略蒸馏方法,将生成式大语言模型教师的排序知识迁移到单令牌决策模型重排序器中,在 TREC DL、BEIR 和 BRIGHT 上取得了最先进的结果,同时运行速度最高可达对应生成式模型的 24.5 倍,并以仅 17 亿参数超越了其 270 亿参数的教师模型。

OPRD:在策略表示蒸馏

Hugging Face Daily Papers

OPRD提出了一种新的知识蒸馏方法,该方法在策略部署期间跨层对齐学生和教师的隐藏状态,消除了来自词空间KL估计的采样方差。实验表明,OPRD在数学推理基准(AIME 2024/2025、AIMO)上优于输出空间基线,同时速度快1.44倍,内存使用减少54%。

超越模仿:基于推理进展过滤策略蒸馏

arXiv cs.AI

论文提出了Reasoning-Progress-Aware Reward Filtering for On-Policy Distillation (R2-OPD) 来解决教师派生奖励与真实推理进展之间的不匹配问题,提高语言模型训练中的推理性能。