Internalize the Temperature: On-Policy Self-Distillation as Policy Reheater for Reinforcement Learning
On the map: LLM reasoning → Reinforcement learning for reasoning
also in 2 more
Abstract (arXiv)
Reinforcement learning from verifiable rewards improves the reasoning ability of large language models, but often suffers from entropy collapse, in which increasingly concentrated policies reduce rollout diversity and useful learning signals. Existing remedies either constrain the RL objective (e.g., entropy regularization) or adjust sampling temperature during rollout collection, but these interventions remain external to the model parameters. We propose Temperature-Scaled On-Policy Self-Distillation (TS-OPSD), a lightweight policy reheating method that internalizes the exploratory effect of temperature into model parameters. Starting from an entropy-collapsed RL checkpoint, TS-OPSD constructs a self-teacher by applying high-temperature scaling to the model's own logits, then distills the resulting smoother distribution back into the student. This policy reheating requires no external teacher, privileged data, or additional inference cost. Experiments on Qwen3-4B-Base and Qwen3-8B-Base show that policy reheating yields a stronger initialization for continued RL than both standard continued RL and rollout-level temperature reheating. Further analyses show that TS-OPSD mainly reduces output sharpness while preserving intermediate representations, top candidate sets, and reasoning capability. These results suggest that entropy restoration can serve as a simple post-collapse intervention for extending reasoning-oriented RL.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 21 of its 28 references that are in the map. First posted on arXiv 30 May 2026.
Builds on
21 of 28 references.
Closest ancestors · 2
Same direction, newest first
- 2025Reasoning with Exploration: An Entropy Perspective17 Jun
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
Other references · 19
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2024SWIFT:A Scalable lightWeight Infrastructure for Fine-Tuningcited by 355
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 299
- 2026On-Policy Context Distillation for Language Modelscited by 186
- 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 145
- 2023f-Divergence Minimization for Sequence-Level Knowledge Distillationcited by 114
7 more
- 2021Annealing Knowledge Distillationcited by 61
- 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 58
- 2025BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clippingcited by 47
- 2024EDT: Improving Large Language Models' Generation by Entropy-based Dynamic Temperature Samplingcited by 45
- 2026Look Inward to Explore Outward: Learning Temperature Policy from LLM Internal States via Hierarchical RLcited by 27
- 2025On the Role of Temperature Sampling in Test-Time Scalingcited by 11
- 2026Temperature as a Meta-Policy: Adaptive Temperature in LLM Reinforcement Learningcited by 8
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026SIPO: Unifying Reinforcement Learning with On-Policy Self-Distillation
- 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models
- 2026Entropy-Preserving Reinforcement Learning
- 2025Exploring Multi-Temperature Strategies for Token- and Rollout-Level Control in RLVR
- 2026Train Ahead, Distill Back: Bootstrapping On-Policy Self-Distillation for Large Language Models
- 2026Scaling Properties of Same-Family On-Policy Distillation
- 2025Bottom-up Policy Optimization: Your Language Model Policy Secretly Contains Internal Policies