Finding the Evidence: Discovering Decision-Supporting Tokens for On-Policy Reasoning Distillation
On the map: LLM reasoning → Prompting and chain-of-thought reasoning
also in 1 more
Abstract (arXiv)
On-policy distillation transfers reasoning ability through dense token-level supervision, yet the nature of the transferable signal remains unclear. We discover that reasoning chains contain two types of knowledge that require different discovery mechanisms: decisions (where to branch), which surface through student uncertainty, and evidence (intermediate steps that justify decisions), which hides in positions where the student is confident yet wrong. Current methods capture only decisions; the substantive knowledge in evidence tokens remains untransferred. We propose DEAR(Decision-Evidence Aware Reasoning Distillation), which first identifies decisions via student entropy, then discovers their supporting evidence through hidden-state cosine similarity to decision anchors, boosted by teacher-student divergence to prioritize the largest knowledge gaps. Across three student-teacher configurations on math and code benchmarks, DEAR consistently outperforms standard OPD, with up to +2.5pp on competition math and +5.7pp on code generation.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 19 of its 23 references that are in the map. First posted on arXiv 22 Jun 2026.
Builds on
19 of 23 references.
Closest ancestors · 3
Same direction, newest first
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
- 2024Keypoint-based Progressive Chain-of-Thought Distillation for LLMs25 May
Other references · 16
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,059
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2021Measuring Coding Challenge Competence With APPScited by 902
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
- 2024Omni-MATH: A Universal Olympiad Level Mathematic Benchmark For Large Language Modelscited by 284
- 2025DeepMath-103K: A Large-Scale, Challenging, Decontaminated, and Verifiable Mathematical Dataset for Advancing Reasoningcited by 280
- 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolationcited by 184
4 more
- 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
- 2026TIP: Token Importance in On-Policy Distillationcited by 104
- 2026Fast and Effective On-policy Distillation from Reasoning Prefixescited by 53
- 2026Reasoning Can Be Restored by Correcting a Few Decision Tokenscited by 2
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026Matching Supervision to the Student's Learning Capacity: A Unified Framework for On-Policy Self-Distillationcited by 3
- 2026TTPO: Test-Time Policy Optimizationcited by 2
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillationcited by 2
- 2026Distillation as Probability Transport: Routed On-Policy Distillationcited by 0
- 2026Dense Is Not Enough: Hierarchical Supervision Allocation for Long-Horizon On-Policy Distillationcited by 0
- 2026A Privileged Teacher That Only Scores Cannot Search: Critique-Conditioned On-Policy Self-Distillation for Reasoningcited by 0
- 2026Reward-Aligned Distillationcited by 0
- 2026Token Utility for On-Policy Distillation via Local Trust-Region Optimizationcited by 0
- 2026Reward-Aligned Distillation10 Oct 2026
- 2026A Privileged Teacher That Only Scores Cannot Search: Critique-Conditioned On-Policy Self-Distillation for Reasoning9 Oct 2026
- 2026Token Utility for On-Policy Distillation via Local Trust-Region Optimization1 Oct 2026
- 2026Dense Is Not Enough: Hierarchical Supervision Allocation for Long-Horizon On-Policy Distillation27 Sep 2026
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation21 Sep 2026
- 2026Distillation as Probability Transport: Routed On-Policy Distillation8 Sep 2026
- 2026TTPO: Test-Time Policy Optimization27 Aug 2026
- 2026Matching Supervision to the Student's Learning Capacity: A Unified Framework for On-Policy Self-Distillation8 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026DIAL-OPD: Learning More from Fewer Tokens in On-Policy Distillation
- 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation
- 2026Learning from a Thoughtful Teacher: Adaptive On-Policy Self-Distillation for Mathematical Reasoning
- 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation
- 2026On-Policy Distillation Teaches New Skills but Not New Knowledge
- 2026E2-OPSD: Taming Entropy Overshoot in On-Policy Self-Distillation
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models
- 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why