arXiv:2607.18082cs.LGcs.AI2026-07被引 1

通过自蒸馏解决评分式强化学习中的探索不足与信号抑制问题

CriPO: Enhancing Rubric-based RL via Self-Distillation

论文配图:CriPO: Enhancing Rubric-based RL via Self-Distillation
图 1 · 摘自论文原文
  • 用自教师机制分别处理未满足和被压制的评分项
  • 实验显示性能更强,优化步骤减少约两倍
  • 适合关注大模型指令对齐与高效训练的研究者

评分式强化学习在提升大模型开放式任务表现方面展现出潜力。然而其主要局限在于探索不足:未被任何采样路径满足的评分项(未探索评分项,UC)无法获得优化信号。现有方法通过在采样阶段引入评分信息作为外部指导来缓解此问题,但导致训练-推理不一致——策略在有外部指导的采样下优化,而推理时无此指导,引发自回归解码中的误差累积。此外,这些方法忽略了另一种根本性失败模式:被压制评分项(SC)——某些采样路径虽满足特定评分项,但在标量奖励聚合过程中因整体优势非正而丢失学习信号。分析表明,SC极为普遍:训练中超过57%的样本存在该问题,平均每样本1.8个被压制评分项。为此,本文提出准则自蒸馏策略优化(CriPO),通过在线自蒸馏同时解决UC与SC问题,且不引入训练-推理不一致。针对UC,CriPO构建准则注入自教师并计算局部前向KL损失以注入缺失行为;针对SC,采用反事实自教师定位负优势采样中相关标记,并将其标记级优势翻转为正值,保留本应被压制的有效模式。在医学与科学基准上的实验表明,CriPO持续优于传统评分式强化学习,在约2倍更少的优化步数下实现更强最终性能。

原文摘要 · Abstract (English)

Rubric-based RL has recently shown promise in improving LLMs on open-ended tasks. A widely recognized limitation of rubric-based RL is limited exploration: criteria that no rollout manages to satisfy (Unexplored Criteria, UC) receive no optimization signal. Recent methods address this by incorporating rubric information as external guidance during rollout, yet they introduce a train-inference mismatch: the policy is optimized on rollouts produced under external guidance while this guidance is absent at inference time, causing error accumulation through autoregressive decoding. Moreover, these exploration-focused approaches overlook a fundamentally different failure mode that we term Suppressed Criteria (SC) -- criteria that are satisfied by some rollouts yet whose learning signals are lost during optimization because scalar reward aggregation assigns them non-positive aggregate advantages. Our analysis reveals that SC are remarkably prevalent: over 57% of samples exhibit this failure mode throughout training, with an average of 1.8 SC per sample. To simultaneously address both UC and SC without introducing training-inference mismatch, we propose Criterion-Distilled Policy Optimization (CriPO), which enhances rubric-based RL via on-policy self-distillation. For UC, CriPO constructs a criterion-injection self-teacher and computes a localized forward-KL loss to inject missing behaviors into the policy. For SC, CriPO employs a counterfactual self-teacher to locate criterion-relevant tokens in negative-advantage rollouts and flips their token-level advantages to positive values, preserving useful patterns that would otherwise be suppressed. Experiments on medicine and science benchmarks demonstrate that CriPO consistently outperforms rubric-based RL, achieving stronger final performance with approximately $2\times$ fewer optimization steps.

强化学习大模型对齐自蒸馏评分机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。