Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation
Not on the map. Why?
Abstract (arXiv)
While recent work in Reinforcement Learning with Verifiable Rewards (RLVR) has shown that a small subset of critical tokens disproportionately drives reasoning gains, an analogous token-level understanding of On-Policy Distillation (OPD) remains largely unexplored. In this work, we investigate high-loss tokens, a token type that--as the most direct signal of student-teacher mismatch under OPD's per-token KL objective--should progressively diminish as training converges according to existing studies; however, our empirical analysis shows otherwise. Even after OPD training reaches apparent saturation, a substantial subset of tokens continues to exhibit persistently high loss; these tokens, which we term Rock Tokens, can account for up to 18% of the tokens in generated outputs. Our investigation reveals two startling paradoxes. First, despite their high occurrence frequency providing a disproportionately large share of total gradient norms, Rock Tokens themselves remain stagnant throughout training, resisting teacher-driven corrections. Second, through causal intervention, we find that these tokens provide negligible functional contribution to the model's actual reasoning performance. These findings suggest that a vast amount of optimization bandwidth is spent on structural and discourse residuals that the student model cannot or need not internalize. By deconstructing these dynamics, we demonstrate that strategically bypassing these ``stumbling blocks'' can significantly streamline the alignment process, challenging the necessity of uniform token weighting and offering a more efficient paradigm for large-scale model distillation.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 68 of its 87 references that are in the map. First posted on arXiv 10 May 2026.
Builds on
68 of 87 references.
References · 68
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Let's Verify Step by Stepcited by 3,598
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 877
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
- 2024From Generation to Judgment: Opportunities and Challenges of LLM-as-a-judgecited by 480
- 2015Policy Distillationcited by 414
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 386
- 2025OpenThoughts: Data Recipes for Reasoning Modelscited by 287
- 2026Reinforcement Learning via Self-Distillationcited by 285
- 2025Reasoning with Exploration: An Entropy Perspectivecited by 277
38 more
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2026Self-Distillation Enables Continual Learningcited by 212
- 2024Large Language Models for Data Annotation and Synthesis: A Surveycited by 201
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
- 2025Preference Leakage: A Contamination Problem in LLM-as-a-judgecited by 119
- 2026Beyond Benchmarks: MathArena as an Evaluation Platform for Mathematics with LLMscited by 115
- 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 78
- 2024Can LLMs Learn from Previous Mistakes? Investigating LLMs' Errors to Boost for Reasoningcited by 69
- 2025Frequency-Aligned Knowledge Distillation for Lightweight Spatiotemporal Forecastingcited by 53
- 2026MAGMA: A Multi-Graph based Agentic Memory Architecture for AI Agentscited by 49
- 2025FineCIR: Explicit Parsing of Fine-Grained Modification Semantics for Composed Image Retrievalcited by 41
- 2026ReTrack: Evidence-Driven Dual-Stream Directional Anchor Calibration Network for Composed Video Retrievalcited by 39
- 2025Chat-Driven Text Generation and Interaction for Person Retrievalcited by 34
- 2025SepPrune: Structured Pruning for Efficient Deep Speech Separationcited by 31
- 2025DRP: Distilled Reasoning Pruning with Skill-aware Step Decomposition for Efficient Large Reasoning Modelscited by 29
- 2024SGLP: A Similarity Guided Fast Layer Partition Pruning for Compressing Large Deep Modelscited by 24
- 2026HABIT: Chrono-Synergia Robust Progressive Learning Framework for Composed Image Retrievalcited by 24
- 2026Delving Deeper: Hierarchical Visual Perception for Robust Video-Text Retrievalcited by 23
- 2026SCRIBE: Structured Mid-Level Supervision for Tool-Using Language Modelscited by 20
- 2026Anatomy of Agentic Memory: Taxonomy and Empirical Analysis of Evaluation and System Limitationscited by 20
- 2026Enhancing Financial Report Question-Answering: A Retrieval-Augmented Generation System with Reranking Analysiscited by 20
- 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMscited by 20
- 2024Contextualization Distillation from Large Language Model for Knowledge Graph Completioncited by 18
- 2025AutoNeural: Co-Designing Vision-Language Models for NPU Inferencecited by 17
- 2026Air-Know: Arbiter-Calibrated Knowledge-Internalizing Robust Network for Composed Image Retrievalcited by 17
- 2024COMAE: COMprehensive Attribute Exploration for Zero-shot Hashingcited by 16
- 2025Learning How to Use Tools, Not Just When: Pattern-Aware Tool-Integrated Reasoningcited by 15
- 2025Find Your Optimal Teacher: Personalized Data Synthesis via Router-Guided Multi-Teacher Distillationcited by 15
- 2025How Do LLMs Use Their Depth?cited by 15
- 2026Task-Specific Efficiency Analysis: When Small Language Models Outperform Large Language Modelscited by 15
- 2026ConeSep: Cone-based Robust Noise-Unlearning Compositional Network for Composed Image Retrievalcited by 15
- 2025Token-Importance Guided Direct Preference Optimizationcited by 14
- 2026HVD: Human Vision-Driven Video Representation Learning for Text-Video Retrievalcited by 14
- 2026TEMA: Anchor the Image, Follow the Text for Multi-Modification Composed Image Retrievalcited by 13
- 2025Mitigating Forgetting in LLM Fine-Tuning via Low-Perplexity Token Learningcited by 12
- 2025MMT-ARD: Multimodal Multi-Teacher Adversarial Distillation for Robust Vision-Language Modelscited by 11
- 2026Distilling Time Series Foundation Models for Efficient Forecastingcited by 11
- 2025Hy-Facial: Hybrid Feature Extraction by Dimensionality Reduction Methods for Enhanced Facial Expression Classificationcited by 10
50 most cited of 68 All 68 (JSON)
Built on by
Citing papers per year: 2026: 11.
11 papers.
- 2026When Safe Skills Collide: Measuring Compositional Risk in Agent Skill Ecosystemscited by 14
- 2026Bridging Reasoning Trajectories in On-Policy Distillation via Near-Future Guidancecited by 12
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 5
- 2026CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillationcited by 5
- 2026Right Knowledge, Wrong Answer: Characterizing Parametric Temporal Conflict in Open-Weight Language Modelscited by 2
- 2026IMAGINE: Adaptive Schema-Imagery Enhanced Composition for Composed Video Retrievalcited by 1
- 2026OracleZoom: On-Policy Self-Distillation Inspired Reference-Constrained Recursive Image Super Resolutioncited by 1
- 2026BEFORE THE FLIP: Measuring Hidden Score Shifts In Quantized Vision Language Models Before The Answer Changes for Visual Question Answeringcited by 1
- 2026SCOPE-OPSD: Fisher-Conditioned Privileged Subspaces for On-Policy Self-Distillationcited by 1
- 2026A Shared Learning Rate Is Not a Neutral Control in Selective On-Policy Distillationcited by 0
- 2026Why On-Policy Distillation Sometimes Fails: Vanishing Learning Signalscited by 0
- 2026Why On-Policy Distillation Sometimes Fails: Vanishing Learning Signals8 Oct 2026
- 2026SCOPE-OPSD: Fisher-Conditioned Privileged Subspaces for On-Policy Self-Distillation11 Sep 2026
- 2026BEFORE THE FLIP: Measuring Hidden Score Shifts In Quantized Vision Language Models Before The Answer Changes for Visual Question Answering7 Sep 2026
- 2026OracleZoom: On-Policy Self-Distillation Inspired Reference-Constrained Recursive Image Super Resolution6 Sep 2026
- 2026A Shared Learning Rate Is Not a Neutral Control in Selective On-Policy Distillation17 Aug 2026
- 2026CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillation13 Aug 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
- 2026Right Knowledge, Wrong Answer: Characterizing Parametric Temporal Conflict in Open-Weight Language Models18 Jun 2026
- 2026IMAGINE: Adaptive Schema-Imagery Enhanced Composition for Composed Video Retrieval6 Jun 2026
- 2026When Safe Skills Collide: Measuring Compositional Risk in Agent Skill Ecosystems30 May 2026
- 2026Bridging Reasoning Trajectories in On-Policy Distillation via Near-Future Guidance29 May 2026