ArXiv

Distilled Reinforcement Learning for LLM Post-training

Authors
Chen Wang, Zhaochun Li, Jionghao Bai...
Categories
cs.LG, cs.AI
arXiv
https://arxiv.org/abs/2607.17247v1
PDF
https://arxiv.org/pdf/2607.17247v1

Brief

Distilled Reinforcement Learning (Distilled RL) addresses limitations of RL (coarse outcome supervision) and on-policy distillation (unconditional KL matching) by embedding teacher supervision into the RL objective. The method uses reverse importance sampling with clipping, negative sample reset, and sequence-level geometric normalization. A case study and broad experiments show substantial gains in pass@1 and pass@k over standard RL and OPD; code is public.

Why it matters

Distilled RL integrates teacher supervision into the RL objective to provide fine-grained guidance, enabling selective transfer of new knowledge and avoiding unconditional imitation; it introduces three mechanisms: reverse importance sampling with clipping, negative sample reset, and sequence-level geometric normalization.

Key details

  • In experiments across within-family and cross-family distillation settings, Distilled RL substantially outperforms standard RL and on-policy distillation (OPD) on pass@1 and pass@k metrics.
  • Paper by Chen Wang, Zhaochun Li, Jionghao Bai, et al., posted to arXiv (2607.17247v1) on 2026-07-19; code is available at https://github.com/597358816/Distilled-RL.
Source evidence

Abstract

Large language model (LLM) post-training is essential for improving reasoning, adaptation, and alignment. Existing methods mainly follow two paradigms: reinforcement learning (RL) and on-policy distillation (OPD). However, RL relies on coarse-grained outcome supervision, resulting in difficult credit assignment and limited capability to acquire new knowledge. OPD, meanwhile, unconditionally matches teacher logits through KL divergence, which creates a dilemma: similar teachers provide little new knowledge, while substantially different teachers often yield ineffective guidance, largely restricting OPD to within-family distillation. We propose Distilled Reinforcement Learning (Distilled RL), which integrates teacher supervision into the RL objective to provide fine-grained guidance, selectively transfer new knowledge and avoid unconditional imitation. Distilled RL contains three components: reverse importance sampling with clipping, negative sample reset, and sequence-level geometric normalization. Through a concise and interpretable case study, we demonstrate that Distilled RL can effectively transfer previously unavailable knowledge from a teacher model to a student model. Extensive experiments across both within-family and cross-family distillation settings show that Distilled RL substantially outperforms standard RL and OPD in terms of both pass@1 and pass@k. Our code is available at https://github.com/597358816/Distilled-RL.