chrisliu298/awesome-on-policy-distillation

A curated collection of papers, technical reports, frameworks, and tools for on-policy distillation (OPD) of large language models

843

212 commits

updated Aug 30, 2026

See the code

README

⚗️ Awesome On-Policy Distillation

Awesome On-Policy Distillation

Entries GitHub Stars GitHub Forks Last Commit

A curated collection of papers, technical reports, frameworks, and tools for on-policy distillation (OPD) of large language models.

On-policy distillation trains a student on samples from its own evolving policy, while a teacher (external, privileged, or self-conditioned) provides dense supervision on those same samples.

On-policy distillation (OPD) trains a student on trajectories sampled from its own policy while a teacher scores the student-visited prefixes with dense token-level guidance. This on-policy data collection reduces the train-inference distribution gap that affects off-policy KD/SFT on fixed traces. Depending on the estimator, OPD looks like GKD on student rollouts or policy-gradient/RL with teacher-defined per-token KL/log-prob rewards, making the natural contrast sparse outcome-reward RL rather than RL as a whole. As of 2026, OPD is a standard post-training primitive at Alibaba (Qwen3), DeepSeek (V4), Xiaomi (MiMo), Zhipu (GLM-5), NVIDIA (Nemotron-Cascade 2), and others.

Shipping today? Jump to Frameworks and Implementations. New to OPD? Read Start Here.

Contents

Start Here

A fast path through the field:

  1. Survey. OPD Survey — taxonomy, methods, and open problems in one place.
  2. Foundations. MiniLLM, GKD, and ExOPD — the core student-rollout plus teacher-supervision loop, including its dense KL-constrained RL framing.
  3. Practical intuition. Thinking Machines blog — the clearest end-to-end explanation of why and when OPD applies.
  4. When OPD works and when it breaks. Revisiting OPD, Entropy-Aware OPD, and Rethinking OPD — failure modes (instability, diversity collapse, tokenizer mismatch) and success conditions (compatible thinking patterns, novel teacher capability).
  5. No teacher logits. Black-Box OPD — discriminator-based reward when the teacher is API-only.
  6. No teacher at all. OPSD and SDFT — same model as student and self-teacher.
  7. Context and experience. OPCD and OEL — distill prompts and deployment traces into weights.
  8. Industrial recipes. Qwen3, DeepSeek-V4, MiMo-V2-Flash, GLM-5 — how labs ship OPD in production.

Key decision: access to teacher logits? Yes → white-box (GKD, Veto, Entropy-Aware OPD). No → black-box (GAD, OVD) or self-distillation (OPSD, SDFT).

Surveys and Essays

Surveys and Position Papers

Essays, Blog Posts, and Walkthroughs

Core OPD Papers

The papers that define on-policy distillation for LLMs.

Scope rule: A paper belongs here if its primary contribution is a new component of the OPD training loop itself — an objective, divergence formulation, stability fix, teacher access-mode variant, self-distillation variant, context-internalization mechanism, or systems/efficiency/privacy constraint applied to that loop — with student rollouts central to the learning signal, evaluated on LLM text generation or reasoning. Operational test: if removing the OPD-loop component leaves a working contribution (a working RL recipe, preference method, or KD baseline), the OPD piece is auxiliary → Adjacent. Papers that enable OPD (cross-tokenizer alignment, calibration), compose with OPD as one component of a larger RL/preference structure, or apply OPD to non-text-reasoning substrates live in Adjacent and Enabling Work or Domain Extensions.

Foundations

Gap-Bridging

Stability and Objective Design

Self-Distillation

Context and Experience Internalization

Efficiency, Systems, and Privacy Variants

Taxonomy

Cross-cutting views over the Core papers — representative examples per category, illustrative rather than exhaustive. For the full set, see the Core OPD Papers subsections. Many entries span multiple categories.

By Teacher Type

By Primary Goal

GoalPapers
Compression / strong-to-weak transferMiniLLM, GKD, Prefix OPD, Rethinking OPD, Lightning OPD, TrOPD, TRD, SWITCH, KAT, PowerOPD, AsyncOPD, TOP-D, ShortOPD, CADENCE, Relay-OPD, Adaptive FastOPD, Dual Nature, STAR-OPD (+23 more)
Stability / objective designDistiLLM, DistiLLM-2, Veto, Entropy-Aware OPD, PACED, SCOPE, Uni-OPD, AOPD, vOPD, OPD+, AD-OPSD, OPD2, Demystifying OPD, Distilled RL, Outcome-Confounded OPD, β-OPSD, Lightning OPD 2.0, SAF-OPD, Counterfactual Recoverability, SPOT, DASH, RP-OPSD, RSTG, DAPD, SA-OPD, Privileged Likelihood, USD, TIDE, SR-OPSD, WDL-OPD, ReOrder-OPD, Simple-OPD, PAST, Adaptive Anchoring, REOPD, CROP, SOPD, R2-OPD, SuRe, AED, TA-OPD, GC-OPD, SMOPD, SPEAR, Rethinking PI, DualOPSD, MoE Routing (+ more)
Post-RL consolidation / skill integrationExOPD, OPCritD, TS-OPSD, Multi-Teacher MOPD, W2S-OPD, TU-OPD, RoCo-ACE, Open-MOPD (+3 more)
Continual learningSDFT, SPoT, OPCD, OEL, MixSD, CaMOPD, CCOPD, SD-Recovery, PromptSD, Sleep, Agent Experience, RESD, SPEE, Latent OPSD, HPSE, EDGE (+4 more)
RL replacement / augmentationSDPO, REOPOLD, RLSD, SDZero, OGLS-SD, PBSD, CREDIT, DA-SD, PGPO, Constitutional OPD, PHF, Direct-OPD, CANON, CriPO, H²SD, Coach, Flux-OPD, CSCR, Privileged but Biased, U-OPSD, Woodpecker, PS-OPSD, RuPI, CausalOPD, CoDA, OP2SD, SKALD, I-SDPO, OPDVR (+39 more)
Reasoning compressionOPSDC, MPD, TRSD, BIRD, Masked Distillation
Black-box distillationGAD, OVD, ROPD, OmniOPD
Safety / alignmentSafeSteer, Constitutional OPD, Multilingual Safety Alignment, DUET, SecOPD

Adjacent and Enabling Work

Papers that are not canonical OPD but matter for understanding or deploying it.

Cross-Tokenizer and Model-Family Enablers

Mismatch Mitigation and Student Quality

Preference, Reward-Guided, and Hybrid RL+KD

Self-Play and Iterative Bootstrapping

Precursors

Domain Extensions

OPD applied to non-text-reasoning settings — agents, multimodal models, diffusion, audio, robotics — and to inference acceleration via speculative decoding. These pass the inclusion criterion (student rollouts central to the learning signal) but on substrates beyond LLM text reasoning.

Agents and Tool-Use

Multimodal and Vision-Language

Speech and Audio

Diffusion, Flow, and Generative Media

Embodied, Robotics, and Control

Speculative Decoding (Draft-Model Training)

Draft-model training for speculative decoding shares OPD's core loop: the draft (student) generates, the target (teacher) verifies, and the draft is updated to match. Included for breadth even though the goal is inference acceleration rather than student capability.

Technical Reports and Industrial Recipes

Production training pipelines that use OPD as a post-training stage.

YearSystemOPD UsageLink
2024Gemma 2KD as alternative to next-token prediction for 2B and 9B studentsarXiv
2025Qwen3Strong-to-weak; off-policy then on-policy distillationarXiv
2025Qwen3-OmniOff-policy then on-policy distillation before GSPOarXiv
2025GLM-4.5 / 4.6Multi-stage post-training with expert model iteration and RLarXiv
2025HY-MT1.5Multi-stage translation: SFT + OPD + RLarXiv
2026MiMo-V2-FlashMulti-Teacher OPD (MOPD) as post-training stagearXiv
2026GLM-5On-policy cross-stage distillation to recover earlier skillsarXiv
2026Typhoon-SMinimal sovereign recipe: SFT + OPD + small-scale RFTarXiv
2026Nemotron-Cascade 2Cascade RL + multi-domain on-policy distillationarXiv
2026Baichuan-M3Task RL → offline policy distillation → multi-teacher OPDarXiv
2026MobileLLM-R1.5Final-stage on-policy KD as primary improvement over R1model card
2026Nanbeige4-3B-ThinkingOPD preferred over off-policy for math reasoningmodel card
2026DeepSeek-V4Domain-expert SFT+GRPO → unified model consolidation via OPDreport
2026Qwen3.5-OmniSpecialist distillation → privileged-input self-distillation aligning audio to textarXiv
2026HY-Embodied-0.532B → 2B on-policy distillation; student rollouts, teacher token-level supervisionarXiv
2026KAT-Coder-V2Specialize-then-Unify: 5 domain-expert agents → unified via OPD on student trajectoriesarXiv
2026KAT-Coder-V2.5Multi-Teacher OPD (MOPD): reverse-KL on student trajectories fuses 5 experts, stabilized by drift-aware truncationarXiv
2026Cursor Composer 2.5Hint-conditioned self-teacher OPD KL added to RL for targeted behaviors (tool calls, style); built on Kimi K2.5blog
2026MAI-Thinking-1Self-distillation on own RL rollouts to resume climbs after collapses or base-policy refreshesreport
2026Nemotron 3 UltraIterative MOPD: 10+ domain-specialized teachers consolidated into the student, refreshed from it each cyclereport
2026InternVideo3Final-stage reverse-KL distillation on student-sampled video rollouts from a stronger teacherarXiv
2026Kwai Keye-VL-2.0Cross-modal multi-teacher OPD: dense token-level teacher feedback on student multimodal rolloutsarXiv
2026NebulaExp-8BSingle- and multi-teacher OPD (MOPD), finding teacher capability outweighs scalearXiv
2026Mach-Mind-4-FlashMulti-Teacher OPD (MOPD): routed reverse-KL on student rollouts fuses 10+ specialists, ending mixed-reward see-sawarXiv
2026Agents-A1Final-stage multi-teacher OPD: routed teachers supervise student rollouts via salient vocabulary alignmentarXiv
2026AudexCascade RL + multi-domain OPD (Nemotron-Cascade lineage), preserving text intelligence in unified audio-text modelarXiv
2026OvisOCR20.8B OCR student distilled from a 4B RL teacher via top-k reverse-KL on its own page-output rolloutsarXiv
2026Gryphon-v2Generate-and-rank recommender replacing a cascade through teacher-ranked student candidate rolloutsarXiv

Frameworks and Implementations

Training Frameworks

FrameworkDescriptionLink
TRLGKD, GOLD, and MiniLLM trainers; most accessible starting pointdocs
NeMo-RLMulti-teacher and cross-tokenizer OPD at scaledocs, repo
veRLAsync on-policy KD trading strict on-policy guarantees for throughputdocs
MS-SwiftGKD and OPSD sections in the ModelScope ecosystemdocs
EasyDistillComprehensive KD toolkit for black-box and white-box LLM distillationarXiv
KDFlowOff-policy, on-policy, and cross-tokenizer distillation via decoupled backendsarXiv, repo
slimeUnified RL stack supporting on-policy distillation and hindsight hintsrepo
OpenClaw-RLAgentic RL stack with hindsight-guided OPDarXiv
NexRLDedicated on-policy distillation recipesrepo
SkyRLOPD examples and blog resourcesrepo
ATLASContinual-learning framework using GKD/GRPO from runtime tracesdocs
AReaLOPD and KDRL over student-sampled trajectories with teacher log-prob guidancedocs
rLLMAgent RL framework (UC Berkeley Sky) with first-class OPD: examples/math_distill/ (DeepMath OPSD + train_deepmath_distill_tinker.{py,sh}) and rllm/trainer/distill/ modules over verl or tinker backendsdocs, repo
SpecForgeSpeculative draft training with EAGLE-3 support and hybrid parallelismarXiv, repo
TorchSpecTorch-native speculative draft training with disaggregated inference/training; streams target hidden states via Mooncake store; Kimi-K2.5/MiniMax-M2.5/Qwen3-Coder-Next examplesblog, repo
Tinker CookbookThinking Machines' Tinker SDK recipes for off-policy KD, single/multi-teacher OPD, multi-turn tool userecipes, repo
ROLLAlibaba's scalable RL library for LLMs/VLMs with an OPD pipelinerepo
EasyOPDUnified verl-based OPD with method-local supervision hooks for cross-tokenizer, self-distillation, and step-wise settingsarXiv, repo
EasyDeLJAX/Flax framework whose GKDTrainer distills onto student-generated completions via a generalized Jensen-Shannon objectiverepo
SpecRollExact speculative rollout serving adapts lightweight proposers at two timescales without changing the target policy distributionarXiv
Mixed Rollout SchedulingMix-aware session admission allocates cache by workload demand while preserving trainer-defined rollout mixtures across OPD plus reinforcement learningarXiv

Implementations

  • OPSD — Official code for Self-Distilled Reasoner / OPSD.
  • SCOPE — Dual-path OPD: student-PPL-weighted MLE for correct rollouts, teacher-PPL-weighted KL for incorrect.
  • CaOPD — K student rollouts → empirical success rate → confidence target replacement → reverse-KL OPD.
  • OPSD-OnPolicyDistillation — verl-based OPD with separate teacher, agent-loop rollouts, and memory-efficient execution.
  • nano-opd — Hackable OPD library decoupling vLLM rollout, FSDP training, and teacher forwards across independent GPU groups.
  • Rethinking OPD — Official code for Rethinking OPD, with verl-based scripts and top-k teacher–student overlap diagnostics merged upstream.
  • DiffusionOPD — Official implementation of round-robin multi-task diffusion OPD distilling task-specialized teachers into one student along its own rollout trajectories.

Acknowledgments

This list draws on the parallel curation effort at thinkwee/AwesomeOPD, which provided pointers to several papers (notably speculative-decoding draft training, BoN distillation, self-play, multilingual and crosslingual self-distillation, clinical and multimodal agentic OPD, additional industrial reports, and several training frameworks). The two lists organize differently — thinkwee/AwesomeOPD groups by feedback signal and access mode; this list groups by methodological role — and are best read together.

Contributing

Contributions welcome. See CONTRIBUTING.md for criteria, section placement, and formatting.

  • Inclusion criteria: the work should involve student rollouts as central to the learning signal, or directly enable OPD deployment (cross-tokenizer, frameworks, etc.).
  • Entry format: [Title](url) *(Year)* — One-line description. See CONTRIBUTING.md for full examples.

Citation

@software{awesome-on-policy-distillation,
  title = {{Awesome On-Policy Distillation}},
  author = {Liu, Chris Yuhao and others},
  year = {2026},
  doi = {10.5281/zenodo.19411493},
  url = {https://github.com/chrisliu298/awesome-on-policy-distillation},
  version = {v1.0.0}
}

awesome
awesome-list
distillation
gkd
knowledge-distillation
llm
llm-distillation
llm-training
minillm
on-policy-distillation
opd
post-training
reinforcement-learning
rl
rlhf
self-distillation
speculative-decoding

Contributors

chrisliu298

210 commits

Linwenye

1 commits

songmzhang

1 commits

chrisliu298/awesome-on-policy-distillation

A curated collection of papers, technical reports, frameworks, and tools for on-policy distillation (OPD) of large language models

843

212 commits

updated Aug 30, 2026

See the code

README

⚗️ Awesome On-Policy Distillation

Awesome On-Policy Distillation

Entries GitHub Stars GitHub Forks Last Commit

A curated collection of papers, technical reports, frameworks, and tools for on-policy distillation (OPD) of large language models.

On-policy distillation trains a student on samples from its own evolving policy, while a teacher (external, privileged, or self-conditioned) provides dense supervision on those same samples.

On-policy distillation (OPD) trains a student on trajectories sampled from its own policy while a teacher scores the student-visited prefixes with dense token-level guidance. This on-policy data collection reduces the train-inference distribution gap that affects off-policy KD/SFT on fixed traces. Depending on the estimator, OPD looks like GKD on student rollouts or policy-gradient/RL with teacher-defined per-token KL/log-prob rewards, making the natural contrast sparse outcome-reward RL rather than RL as a whole. As of 2026, OPD is a standard post-training primitive at Alibaba (Qwen3), DeepSeek (V4), Xiaomi (MiMo), Zhipu (GLM-5), NVIDIA (Nemotron-Cascade 2), and others.

Shipping today? Jump to Frameworks and Implementations. New to OPD? Read Start Here.

Contents

Start Here

A fast path through the field:

  1. Survey. OPD Survey — taxonomy, methods, and open problems in one place.
  2. Foundations. MiniLLM, GKD, and ExOPD — the core student-rollout plus teacher-supervision loop, including its dense KL-constrained RL framing.
  3. Practical intuition. Thinking Machines blog — the clearest end-to-end explanation of why and when OPD applies.
  4. When OPD works and when it breaks. Revisiting OPD, Entropy-Aware OPD, and Rethinking OPD — failure modes (instability, diversity collapse, tokenizer mismatch) and success conditions (compatible thinking patterns, novel teacher capability).
  5. No teacher logits. Black-Box OPD — discriminator-based reward when the teacher is API-only.
  6. No teacher at all. OPSD and SDFT — same model as student and self-teacher.
  7. Context and experience. OPCD and OEL — distill prompts and deployment traces into weights.
  8. Industrial recipes. Qwen3, DeepSeek-V4, MiMo-V2-Flash, GLM-5 — how labs ship OPD in production.

Key decision: access to teacher logits? Yes → white-box (GKD, Veto, Entropy-Aware OPD). No → black-box (GAD, OVD) or self-distillation (OPSD, SDFT).

Surveys and Essays

Surveys and Position Papers

Essays, Blog Posts, and Walkthroughs

Core OPD Papers

The papers that define on-policy distillation for LLMs.

Scope rule: A paper belongs here if its primary contribution is a new component of the OPD training loop itself — an objective, divergence formulation, stability fix, teacher access-mode variant, self-distillation variant, context-internalization mechanism, or systems/efficiency/privacy constraint applied to that loop — with student rollouts central to the learning signal, evaluated on LLM text generation or reasoning. Operational test: if removing the OPD-loop component leaves a working contribution (a working RL recipe, preference method, or KD baseline), the OPD piece is auxiliary → Adjacent. Papers that enable OPD (cross-tokenizer alignment, calibration), compose with OPD as one component of a larger RL/preference structure, or apply OPD to non-text-reasoning substrates live in Adjacent and Enabling Work or Domain Extensions.

Foundations

Gap-Bridging

Stability and Objective Design

Self-Distillation

Context and Experience Internalization

Efficiency, Systems, and Privacy Variants

Taxonomy

Cross-cutting views over the Core papers — representative examples per category, illustrative rather than exhaustive. For the full set, see the Core OPD Papers subsections. Many entries span multiple categories.

By Teacher Type

By Primary Goal

GoalPapers
Compression / strong-to-weak transferMiniLLM, GKD, Prefix OPD, Rethinking OPD, Lightning OPD, TrOPD, TRD, SWITCH, KAT, PowerOPD, AsyncOPD, TOP-D, ShortOPD, CADENCE, Relay-OPD, Adaptive FastOPD, Dual Nature, STAR-OPD (+23 more)
Stability / objective designDistiLLM, DistiLLM-2, Veto, Entropy-Aware OPD, PACED, SCOPE, Uni-OPD, AOPD, vOPD, OPD+, AD-OPSD, OPD2, Demystifying OPD, Distilled RL, Outcome-Confounded OPD, β-OPSD, Lightning OPD 2.0, SAF-OPD, Counterfactual Recoverability, SPOT, DASH, RP-OPSD, RSTG, DAPD, SA-OPD, Privileged Likelihood, USD, TIDE, SR-OPSD, WDL-OPD, ReOrder-OPD, Simple-OPD, PAST, Adaptive Anchoring, REOPD, CROP, SOPD, R2-OPD, SuRe, AED, TA-OPD, GC-OPD, SMOPD, SPEAR, Rethinking PI, DualOPSD, MoE Routing (+ more)
Post-RL consolidation / skill integrationExOPD, OPCritD, TS-OPSD, Multi-Teacher MOPD, W2S-OPD, TU-OPD, RoCo-ACE, Open-MOPD (+3 more)
Continual learningSDFT, SPoT, OPCD, OEL, MixSD, CaMOPD, CCOPD, SD-Recovery, PromptSD, Sleep, Agent Experience, RESD, SPEE, Latent OPSD, HPSE, EDGE (+4 more)
RL replacement / augmentationSDPO, REOPOLD, RLSD, SDZero, OGLS-SD, PBSD, CREDIT, DA-SD, PGPO, Constitutional OPD, PHF, Direct-OPD, CANON, CriPO, H²SD, Coach, Flux-OPD, CSCR, Privileged but Biased, U-OPSD, Woodpecker, PS-OPSD, RuPI, CausalOPD, CoDA, OP2SD, SKALD, I-SDPO, OPDVR (+39 more)
Reasoning compressionOPSDC, MPD, TRSD, BIRD, Masked Distillation
Black-box distillationGAD, OVD, ROPD, OmniOPD
Safety / alignmentSafeSteer, Constitutional OPD, Multilingual Safety Alignment, DUET, SecOPD

Adjacent and Enabling Work

Papers that are not canonical OPD but matter for understanding or deploying it.

Cross-Tokenizer and Model-Family Enablers

Mismatch Mitigation and Student Quality

Preference, Reward-Guided, and Hybrid RL+KD

Self-Play and Iterative Bootstrapping

Precursors

Domain Extensions

OPD applied to non-text-reasoning settings — agents, multimodal models, diffusion, audio, robotics — and to inference acceleration via speculative decoding. These pass the inclusion criterion (student rollouts central to the learning signal) but on substrates beyond LLM text reasoning.

Agents and Tool-Use

Multimodal and Vision-Language

Speech and Audio

Diffusion, Flow, and Generative Media

Embodied, Robotics, and Control

Speculative Decoding (Draft-Model Training)

Draft-model training for speculative decoding shares OPD's core loop: the draft (student) generates, the target (teacher) verifies, and the draft is updated to match. Included for breadth even though the goal is inference acceleration rather than student capability.

Technical Reports and Industrial Recipes

Production training pipelines that use OPD as a post-training stage.

YearSystemOPD UsageLink
2024Gemma 2KD as alternative to next-token prediction for 2B and 9B studentsarXiv
2025Qwen3Strong-to-weak; off-policy then on-policy distillationarXiv
2025Qwen3-OmniOff-policy then on-policy distillation before GSPOarXiv
2025GLM-4.5 / 4.6Multi-stage post-training with expert model iteration and RLarXiv
2025HY-MT1.5Multi-stage translation: SFT + OPD + RLarXiv
2026MiMo-V2-FlashMulti-Teacher OPD (MOPD) as post-training stagearXiv
2026GLM-5On-policy cross-stage distillation to recover earlier skillsarXiv
2026Typhoon-SMinimal sovereign recipe: SFT + OPD + small-scale RFTarXiv
2026Nemotron-Cascade 2Cascade RL + multi-domain on-policy distillationarXiv
2026Baichuan-M3Task RL → offline policy distillation → multi-teacher OPDarXiv
2026MobileLLM-R1.5Final-stage on-policy KD as primary improvement over R1model card
2026Nanbeige4-3B-ThinkingOPD preferred over off-policy for math reasoningmodel card
2026DeepSeek-V4Domain-expert SFT+GRPO → unified model consolidation via OPDreport
2026Qwen3.5-OmniSpecialist distillation → privileged-input self-distillation aligning audio to textarXiv
2026HY-Embodied-0.532B → 2B on-policy distillation; student rollouts, teacher token-level supervisionarXiv
2026KAT-Coder-V2Specialize-then-Unify: 5 domain-expert agents → unified via OPD on student trajectoriesarXiv
2026KAT-Coder-V2.5Multi-Teacher OPD (MOPD): reverse-KL on student trajectories fuses 5 experts, stabilized by drift-aware truncationarXiv
2026Cursor Composer 2.5Hint-conditioned self-teacher OPD KL added to RL for targeted behaviors (tool calls, style); built on Kimi K2.5blog
2026MAI-Thinking-1Self-distillation on own RL rollouts to resume climbs after collapses or base-policy refreshesreport
2026Nemotron 3 UltraIterative MOPD: 10+ domain-specialized teachers consolidated into the student, refreshed from it each cyclereport
2026InternVideo3Final-stage reverse-KL distillation on student-sampled video rollouts from a stronger teacherarXiv
2026Kwai Keye-VL-2.0Cross-modal multi-teacher OPD: dense token-level teacher feedback on student multimodal rolloutsarXiv
2026NebulaExp-8BSingle- and multi-teacher OPD (MOPD), finding teacher capability outweighs scalearXiv
2026Mach-Mind-4-FlashMulti-Teacher OPD (MOPD): routed reverse-KL on student rollouts fuses 10+ specialists, ending mixed-reward see-sawarXiv
2026Agents-A1Final-stage multi-teacher OPD: routed teachers supervise student rollouts via salient vocabulary alignmentarXiv
2026AudexCascade RL + multi-domain OPD (Nemotron-Cascade lineage), preserving text intelligence in unified audio-text modelarXiv
2026OvisOCR20.8B OCR student distilled from a 4B RL teacher via top-k reverse-KL on its own page-output rolloutsarXiv
2026Gryphon-v2Generate-and-rank recommender replacing a cascade through teacher-ranked student candidate rolloutsarXiv

Frameworks and Implementations

Training Frameworks

FrameworkDescriptionLink
TRLGKD, GOLD, and MiniLLM trainers; most accessible starting pointdocs
NeMo-RLMulti-teacher and cross-tokenizer OPD at scaledocs, repo
veRLAsync on-policy KD trading strict on-policy guarantees for throughputdocs
MS-SwiftGKD and OPSD sections in the ModelScope ecosystemdocs
EasyDistillComprehensive KD toolkit for black-box and white-box LLM distillationarXiv
KDFlowOff-policy, on-policy, and cross-tokenizer distillation via decoupled backendsarXiv, repo
slimeUnified RL stack supporting on-policy distillation and hindsight hintsrepo
OpenClaw-RLAgentic RL stack with hindsight-guided OPDarXiv
NexRLDedicated on-policy distillation recipesrepo
SkyRLOPD examples and blog resourcesrepo
ATLASContinual-learning framework using GKD/GRPO from runtime tracesdocs
AReaLOPD and KDRL over student-sampled trajectories with teacher log-prob guidancedocs
rLLMAgent RL framework (UC Berkeley Sky) with first-class OPD: examples/math_distill/ (DeepMath OPSD + train_deepmath_distill_tinker.{py,sh}) and rllm/trainer/distill/ modules over verl or tinker backendsdocs, repo
SpecForgeSpeculative draft training with EAGLE-3 support and hybrid parallelismarXiv, repo
TorchSpecTorch-native speculative draft training with disaggregated inference/training; streams target hidden states via Mooncake store; Kimi-K2.5/MiniMax-M2.5/Qwen3-Coder-Next examplesblog, repo
Tinker CookbookThinking Machines' Tinker SDK recipes for off-policy KD, single/multi-teacher OPD, multi-turn tool userecipes, repo
ROLLAlibaba's scalable RL library for LLMs/VLMs with an OPD pipelinerepo
EasyOPDUnified verl-based OPD with method-local supervision hooks for cross-tokenizer, self-distillation, and step-wise settingsarXiv, repo
EasyDeLJAX/Flax framework whose GKDTrainer distills onto student-generated completions via a generalized Jensen-Shannon objectiverepo
SpecRollExact speculative rollout serving adapts lightweight proposers at two timescales without changing the target policy distributionarXiv
Mixed Rollout SchedulingMix-aware session admission allocates cache by workload demand while preserving trainer-defined rollout mixtures across OPD plus reinforcement learningarXiv

Implementations

  • OPSD — Official code for Self-Distilled Reasoner / OPSD.
  • SCOPE — Dual-path OPD: student-PPL-weighted MLE for correct rollouts, teacher-PPL-weighted KL for incorrect.
  • CaOPD — K student rollouts → empirical success rate → confidence target replacement → reverse-KL OPD.
  • OPSD-OnPolicyDistillation — verl-based OPD with separate teacher, agent-loop rollouts, and memory-efficient execution.
  • nano-opd — Hackable OPD library decoupling vLLM rollout, FSDP training, and teacher forwards across independent GPU groups.
  • Rethinking OPD — Official code for Rethinking OPD, with verl-based scripts and top-k teacher–student overlap diagnostics merged upstream.
  • DiffusionOPD — Official implementation of round-robin multi-task diffusion OPD distilling task-specialized teachers into one student along its own rollout trajectories.

Acknowledgments

This list draws on the parallel curation effort at thinkwee/AwesomeOPD, which provided pointers to several papers (notably speculative-decoding draft training, BoN distillation, self-play, multilingual and crosslingual self-distillation, clinical and multimodal agentic OPD, additional industrial reports, and several training frameworks). The two lists organize differently — thinkwee/AwesomeOPD groups by feedback signal and access mode; this list groups by methodological role — and are best read together.

Contributing

Contributions welcome. See CONTRIBUTING.md for criteria, section placement, and formatting.

  • Inclusion criteria: the work should involve student rollouts as central to the learning signal, or directly enable OPD deployment (cross-tokenizer, frameworks, etc.).
  • Entry format: [Title](url) *(Year)* — One-line description. See CONTRIBUTING.md for full examples.

Citation

@software{awesome-on-policy-distillation,
  title = {{Awesome On-Policy Distillation}},
  author = {Liu, Chris Yuhao and others},
  year = {2026},
  doi = {10.5281/zenodo.19411493},
  url = {https://github.com/chrisliu298/awesome-on-policy-distillation},
  version = {v1.0.0}
}

awesome
awesome-list
distillation
gkd
knowledge-distillation
llm
llm-distillation
llm-training
minillm
on-policy-distillation
opd
post-training
reinforcement-learning
rl
rlhf
self-distillation
speculative-decoding

Contributors

chrisliu298

210 commits

Linwenye

1 commits

songmzhang

1 commits