跳到正文
arXiv:cs.LG· Qinglin Su, Licheng Pan, Zhen Fang, Jiakang Wang, Qingnan Ren, Yunsheng Lu, Yan Gao, Zhixuan Chu·· 4 小时前

RM-OPD:用同策略蒸馏训练生成式奖励模型

RM-OPD: On-Policy Distillation for Generative Reward Models

AI 导读

RM-OPD 是一个两阶段框架,先将单域 GRPO 微调得到的专家教师模型,通过路由式同策略蒸馏、领域贡献重加权和可选的评判标准上下文蒸馏整合进一个学生模型。在多个 RM 基准和两个基座模型上,RM-OPD 持续优于原生 Mix-RL 及其他强基线,并出现"教师超越"效应。该工作旨在解决 Mix-RL 的奖励稀疏与跨域干扰问题,构建通用生成式奖励模型。

正文

View PDF HTML (experimental)

Abstract:Reinforcement Learning on Mixed-domain preference data (Mix-RL) has become the mainstream approach for training general-purpose Generative Reward Models (GRMs). However, Mix-RL suffers from two critical bottlenecks: the reward sparsity induced by outcome rewards, and the cross-domain interference arising from jointly optimizing heterogeneous capabilities, which together give rise to imbalanced capability acquisition and pervasive reward hacking. Inspired by the success of On-Policy Distillation (OPD) in multi-domain capability integration, we introduce RM-OPD (Reward Modeling via On-Policy Distillation), a two-stage framework that separates specialist acquisition from capability integration. First, we cultivate domain-specialized teacher models via single-domain GRPO fine-tuning, allowing each expert to reach its performance ceiling in isolation; then we consolidate these teachers into one student model through (i) routed on-policy distillation, which provides dense signed token-level supervision on student-generated critiques; (ii) domain contribution reweighting, which reweights token guidance toward target domain shares, preventing specific domains from dominating optimization; and (iii) optional criteria context distillation, which conditions teachers on domain-specific evaluation criteria to sharpen token-level supervision. Across multiple RM benchmarks and two base models, RM-OPD consistently outperforms vanilla Mix-RL and other strong baselines, achieving overall superior performance and exhibiting an emergent "teacher-surpassing" effect. These results establish RM-OPD as a promising paradigm for building generalist GRMs.
Comments: 34 pages, 4 figures; includes references and appendices
Subjects: Machine Learning (cs.LG); Computation and Language (cs.CL)
Cite as: arXiv:2610.20174 [cs.LG]
  (or arXiv:2610.20174v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2610.20174

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Qinglin Su [view email]
[v1] Wed, 14 Oct 2026 13:20:55 UTC (717 KB)

来源:arXiv:cs.LG · arxiv.org