mlsanityData to 7 Oct 2026

Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation

arXiv 2605.09253 ↗First posted 10 May 2026cs.CL cs.AIcited by 11

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

  1. 2025Qwen3 Technical Reportcited by 8,195
  2. 2023Let's Verify Step by Stepcited by 3,598
  3. 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
  4. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 877
  5. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  6. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
  7. 2024From Generation to Judgment: Opportunities and Challenges of LLM-as-a-judgecited by 480
  8. 2015Policy Distillationcited by 414
  9. 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 386
  10. 2025OpenThoughts: Data Recipes for Reasoning Modelscited by 287
  11. 2026Reinforcement Learning via Self-Distillationcited by 285
  12. 2025Reasoning with Exploration: An Entropy Perspectivecited by 277
38 more
  1. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
  2. 2026Self-Distillation Enables Continual Learningcited by 212
  3. 2024Large Language Models for Data Annotation and Synthesis: A Surveycited by 201
  4. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
  5. 2025Preference Leakage: A Contamination Problem in LLM-as-a-judgecited by 119
  6. 2026Beyond Benchmarks: MathArena as an Evaluation Platform for Mathematics with LLMscited by 115
  7. 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 78
  8. 2024Can LLMs Learn from Previous Mistakes? Investigating LLMs' Errors to Boost for Reasoningcited by 69
  9. 2025Frequency-Aligned Knowledge Distillation for Lightweight Spatiotemporal Forecastingcited by 53
  10. 2026MAGMA: A Multi-Graph based Agentic Memory Architecture for AI Agentscited by 49
  11. 2025FineCIR: Explicit Parsing of Fine-Grained Modification Semantics for Composed Image Retrievalcited by 41
  12. 2026ReTrack: Evidence-Driven Dual-Stream Directional Anchor Calibration Network for Composed Video Retrievalcited by 39
  13. 2025Chat-Driven Text Generation and Interaction for Person Retrievalcited by 34
  14. 2025SepPrune: Structured Pruning for Efficient Deep Speech Separationcited by 31
  15. 2025DRP: Distilled Reasoning Pruning with Skill-aware Step Decomposition for Efficient Large Reasoning Modelscited by 29
  16. 2024SGLP: A Similarity Guided Fast Layer Partition Pruning for Compressing Large Deep Modelscited by 24
  17. 2026HABIT: Chrono-Synergia Robust Progressive Learning Framework for Composed Image Retrievalcited by 24
  18. 2026Delving Deeper: Hierarchical Visual Perception for Robust Video-Text Retrievalcited by 23
  19. 2026SCRIBE: Structured Mid-Level Supervision for Tool-Using Language Modelscited by 20
  20. 2026Anatomy of Agentic Memory: Taxonomy and Empirical Analysis of Evaluation and System Limitationscited by 20
  21. 2026Enhancing Financial Report Question-Answering: A Retrieval-Augmented Generation System with Reranking Analysiscited by 20
  22. 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMscited by 20
  23. 2024Contextualization Distillation from Large Language Model for Knowledge Graph Completioncited by 18
  24. 2025AutoNeural: Co-Designing Vision-Language Models for NPU Inferencecited by 17
  25. 2026Air-Know: Arbiter-Calibrated Knowledge-Internalizing Robust Network for Composed Image Retrievalcited by 17
  26. 2024COMAE: COMprehensive Attribute Exploration for Zero-shot Hashingcited by 16
  27. 2025Learning How to Use Tools, Not Just When: Pattern-Aware Tool-Integrated Reasoningcited by 15
  28. 2025Find Your Optimal Teacher: Personalized Data Synthesis via Router-Guided Multi-Teacher Distillationcited by 15
  29. 2025How Do LLMs Use Their Depth?cited by 15
  30. 2026Task-Specific Efficiency Analysis: When Small Language Models Outperform Large Language Modelscited by 15
  31. 2026ConeSep: Cone-based Robust Noise-Unlearning Compositional Network for Composed Image Retrievalcited by 15
  32. 2025Token-Importance Guided Direct Preference Optimizationcited by 14
  33. 2026HVD: Human Vision-Driven Video Representation Learning for Text-Video Retrievalcited by 14
  34. 2026TEMA: Anchor the Image, Follow the Text for Multi-Modification Composed Image Retrievalcited by 13
  35. 2025Mitigating Forgetting in LLM Fine-Tuning via Low-Perplexity Token Learningcited by 12
  36. 2025MMT-ARD: Multimodal Multi-Teacher Adversarial Distillation for Robust Vision-Language Modelscited by 11
  37. 2026Distilling Time Series Foundation Models for Efficient Forecastingcited by 11
  38. 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

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 11.

11 papers.