mlsanityData to 14 Oct 2026

Dense Is Not Enough: Hierarchical Supervision Allocation for Long-Horizon On-Policy Distillation

arXiv 2609.33409 ↗First posted 27 Sep 2026cs.CLcited by 0

On the map: Knowledge distillation → On-policy and self-distillation for LLMs

Abstract (arXiv)

On-policy distillation (OPD) transfers the capabilities of a large language model to a smaller student by providing teacher supervision on the student's own rollouts. In long-horizon agentic tasks, however, uniform token-level matching can allocate supervision poorly: a large local discrepancy need not improve future behavior, while consequential guidance may be beyond the current student's reach or fail to persist without privileged input. We formulate long-horizon OPD as hierarchical supervision allocation and argue that productive guidance lies at the intersection of future utility and current learnability. Crucially, this intersection evolves as the student learns. Based on this principle, we propose LENS-OPD, a coarse-to-fine framework that organizes supervision through Locate, Validate, and Refine. Locate adapts trajectory exposure to the student's evolving competence and proposes a candidate decision for intervention. Validate tests whether teacher guidance at that decision improves the same student's subsequent behavior. Refine internalizes the beneficial guided behavior into the deployable policy and concentrates token-level supervision on decisive teacher-student conflicts within the validated turn. These stages are nested: each finer allocation is conditioned on the coarser decision, rather than being optimized as an independent importance score. Experiments across multiple long-horizon agent benchmarks and student-teacher configurations show that LENS-OPD consistently improves task performance over vanilla OPD and strong curriculum- and selection-based baselines. Our results suggest that effective long-horizon distillation requires teaching at the right depth, the right decision, and the right token.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 32 of its 35 references that are in the map. First posted on arXiv 27 Sep 2026.

Builds on

32 of 35 references.

Closest ancestors · 17

Same direction, newest first

  1. 2026Beyond Imitation: Filtering On-Policy Distillation by Reasoning Progress19 Aug
  2. 2026Look Ahead Before You Distill: Future Trajectory Validation of Teacher Guidance for Agentic On-Policy Distillation3 Aug
  3. 2026DASH-OPD: Discrepancy-Aware Switching with Hysteresis for On-Policy Distillation31 Jul
  4. 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Training7 Jul
  5. 2026Finding the Evidence: Discovering Decision-Supporting Tokens for On-Policy Reasoning Distillation22 Jun
  6. 2026On the Position Bias of On-Policy Distillation21 Jun
  7. 2026SAGE-OPD: Selective Agent-Guided Intervention for Multi-Turn On-Policy Distillation17 Jun
  8. 2026On-Policy Distillation with Curriculum Turn-level Guidance for Multi-turn Agents14 Jun
  9. 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation1 Jun
  10. 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation26 May
  11. 2026Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Level7 May
  12. 2026TCOD: Exploring Temporal Curriculum in On-Policy Distillation for Multi-turn Autonomous Agents27 Apr
5 more
  1. 2026TIP: Token Importance in On-Policy Distillation15 Apr
  2. 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
  3. 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
  4. 2026Self-Distillation Enables Continual Learning27 Jan
  5. 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models26 Jan

Other references · 15

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  3. 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,463
  4. 2022WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agentscited by 1,251
  5. 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,205
  6. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
  7. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
  8. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
  9. 2016Sequence-Level Knowledge Distillationcited by 801
  10. 2019Patient Knowledge Distillation for BERT Model Compressioncited by 508
  11. 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 482
  12. 2022ScienceWorld: Is your Agent Smarter than a 5th Grader?cited by 327
3 more
  1. 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 245
  2. 2023MiniLLM: On-Policy Distillation of Large Language Modelscited by 139
  3. 2025SelecTKD: Selective Token-Weighted Knowledge Distillation for LLMscited by 22

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Reward-Aligned Reweighting for On-Policy Distillation
  2. 2026Teacher-Student Gaps Are Not Enough: Outcome-Guided On-Policy Distillation for Multi-Turn Autonomous Agents
  3. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
  4. 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillation
  5. 2026When Teachers Mislead: Spurious-Signal-Aware On-Policy Distillation
  6. 2026Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations
  7. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
  8. 2026Flux-OPD: On-Policy Distillation with Evolving Contexts