mlsanityData to 14 Oct 2026

Rethinking Reverse KL as Adaptive Entropy Distillation

arXiv 2608.14685 ↗First posted 5 Aug 2026cs.LG stat.MLcited by 0

On the map: LLM reasoning → Prompting and chain-of-thought reasoning

also in 1 more
Abstract (arXiv)

Knowledge distillation (KD) is widely used to transfer the capabilities of large language models (LLMs) to smaller students, but existing objectives often struggle to balance faithful imitation and robust generation. In particular, existing methods mainly combine FKL and RKL, overlooking that RKL itself provides a mechanism for adjusting the student's imitation strength. Motivated by this, we revisit on-policy Reverse Kullback-Leibler (RKL) distillation and decompose its objective into a teacher-fitting term and a student-entropy term, without introducing an explicit FKL branch. We show theoretically that the token-level optimal student distribution corresponds to a tempered variant of the teacher distribution, where the adaptive weight controls the trade-off between mode-seeking and uncertainty preservation. Guided by this insight, we propose Adaptive Entropy Distillation (AED), which uses the teacher's entropy to dynamically calibrate token-level imitation strength. Experiments on instruction-following and mathematical reasoning benchmarks demonstrate that AED achieves superior overall performance and generally improves teacher--student distributional and entropy alignment.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 24 of its 28 references that are in the map. First posted on arXiv 5 Aug 2026.

Builds on

24 of 28 references.

Closest ancestors · 1

Same direction, newest first

  1. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr

Other references · 23

  1. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  3. 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
  4. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  5. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  6. 2022Self-Instruct: Aligning Language Models with Self-Generated Instructionscited by 2,724
  7. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
  8. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
  9. 2024Length-Controlled AlpacaEval: A Simple Way to Debias Automatic Evaluatorscited by 897
  10. 2022Super-NaturalInstructions: Generalization via Declarative Instructions on 1600+ NLP Taskscited by 868
  11. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
  12. 2016Sequence-Level Knowledge Distillationcited by 801
11 more
  1. 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 737
  2. 2024TinyLlama: An Open-Source Small Language Modelcited by 618
  3. 2015Sequence Level Training with Recurrent Neural Networkscited by 576
  4. 2018Born Again Neural Networkscited by 546
  5. 2022Unnatural Instructions: Tuning Language Models with (Almost) No Human Laborcited by 397
  6. 2023Specializing Smaller Language Models towards Multi-Step Reasoningcited by 251
  7. 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
  8. 2023f-Divergence Minimization for Sequence-Level Knowledge Distillationcited by 114
  9. 2024Knowledge Distillation Based on Transformed Teacher Matchingcited by 37
  10. 2025ToDi: Token-wise Distillation via Fine-Grained Divergence Controlcited by 27
  11. 2022BD-KD: Balancing the Divergences for Online Knowledge Distillationcited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Learning What to Imitate: Entropy-Aware Distribution Mixing
  2. 2026Distilling What Matters: Confidence-Aware Selective Distillation for Large Language Models
  3. 2026EGAD: Entropy-Guided Adaptive Distillation for Token-Level Knowledge Transfer
  4. 2026ReTaCo: Residual-Target Control for On-Policy Distillation
  5. 2026Do We Really Need KL Divergence for On-Policy Distillation of Large Language Models?
  6. 2024Exploring and Enhancing the Transfer of Distribution in Knowledge Distillation for Autoregressive Language Models
  7. 2026Beyond Imitation: Auditing the Recoverability of Reasoning in Distilled Models
  8. 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models