arXiv:2606.23104cs.LGcs.AI2026-06

让大模型在自我学习时更关注错误推理,提升生成质量。

ReNIO: Reweighting Negative Trajectory Importance for LLM On-Policy Distillation

论文配图:ReNIO: Reweighting Negative Trajectory Importance for LLM On-Policy Distillation
图 1 · 摘自论文原文
  • 用学生-教师概率比识别导致错误的关键词元,动态加权负向轨迹。
  • 数学推理和代码生成任务上,最高提升10%性能,无需完整答案。
  • 适合做自训练的LLM研究者,尤其关注推理能力优化的团队。

基于策略蒸馏(OPD)能通过学生模型自生成输出进行训练来提升大模型推理能力,但传统方法对所有学生生成输出(SGO)同等对待,忽略其信息量差异。控制性实验发现:仅使用错误的SGO训练,效果优于仅用正确输出。分析表明,仅训练正确输出会导致推理链变短、反思能力弱化;而错误输出更能保留模型能力边界附近的探索性推理。为此,提出ReNIO,一种重加权负向轨迹重要性的方法:通过学生-教师概率比识别引发错误推理的关键词元,并将信息整合为归一化样本权重,无需判断最终答案正确性即可自动赋予可能错误轨迹更高权重。该方法仅依赖前缀条件概率,保持了前缀训练的优势,避免全回溯强化学习。在数学推理与代码生成任务中,ReNIO显著提升OPD与OPSD表现,对Qwen3-1.7B相对增益达8.90%,对R1-Distill-Qwen-7B达10.00%。代码仓库:https://github.com/BDML-lab/ReNIO。

原文摘要 · Abstract (English)

On-policy distillation (OPD) improves LLM reasoning by training a student model on its own generated outputs, but standard OPD treats all student-generated outputs (SGOs) equally regardless of their informativeness. We observe a consistent asymmetry in controlled filtering experiments: in both OPD and on-policy self distillation (OPSD), training only on incorrect SGOs outperforms training only on correct ones. Our further analysis suggests that models trained on correct-only SGOs tend to generate shorter reasoning traces and show weaker reflection behavior, while incorrect SGOs better preserve exploratory reasoning near the model's capability boundary. To exploit this signal without requiring full answer-containing rollouts, we introduce ReNIO, which Reweights Negative trajectory Importance for LLM On-policy distillation. By using the student-to-teacher probability ratio, ReNIO identifies pivotal tokens leading to wrong reasoning traces and aggregates their information into a normalized sample weight, inherently assigning larger weights to likely negative trajectories without observing the correctness of final-answer. Since Re-NIO only uses prefix-conditioned token probabilities, it preserves OPD's prefix training advantage over full-rollout reinforcement learning. Across both mathematical reasoning and code generation tasks, ReNIO improves both OPD and OPSD, with representative relative gains of up to 8.90% for Qwen3-1.7B and 10.00% for R1-Distill-Qwen-7B on mathematical reasoning benchmarks. Code repo: https://github.com/BDML-lab/ReNIO.

大模型自训练推理优化蒸馏

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