mlsanityData to 7 Oct 2026

The Bridge-Garden Dilemma in LLM Distillation: Why Mixing Hard and Soft Labels Works

arXiv 2605.26246 ↗First posted 25 May 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Knowledge distillation (KD) transfers knowledge from a large teacher model to a smaller student. In language modeling, the student is trained either on tokens sampled from the teacher (hard labels) or the teacher's full next-token distribution (soft labels). Despite soft labels appear strictly richer, we find that mixing hard and soft labels consistently yields better results. Crucially, we show that this gain cannot be explained by closer teacher matching during training. Instead, it comes from reduced exposure bias, the mismatch between training and inference distributions. To explain this phenomenon, we introduce the Bridge-Garden Decomposition theory, which categorizes generation steps into two types: Bridges, where the next token must be exact, and Gardens, where it can be flexible. We show that hard-only KD excels in Bridges by avoiding risky deviations, while soft-only KD preserves diversity in Gardens. A hybrid strategy handles both cases and, as a result, reduces exposure bias across the sequence. Guided by this theory, we develop a family of Bridge-Garden hybrid supervision methods that adaptively balance hard and soft labels. Across a primary suite of seven teacher-student pairs (including Qwen, Llama, Gemma, and DeepSeek) and benchmarks in reasoning and coding, our approach outperforms divergence-based and on-policy KD baselines while reducing training cost by 9.7x, enabling efficient model compression. Code is available at https://github.com/ghwang-s/bridge_garden_hybrid_kd_release.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 63 of its 78 references that are in the map. First posted on arXiv 25 May 2026.

Builds on

63 of 78 references.

References · 63

  1. 2023GPT-4 Technical Reportcited by 18,132
  2. 2024The Llama 3 Herd of Modelscited by 14,192
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  4. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  5. 2025Qwen3 Technical Reportcited by 8,195
  6. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  7. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  8. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  9. 2024Qwen2.5 Technical Reportcited by 4,713
  10. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  11. 2021Program Synthesis with Large Language Modelscited by 3,509
  12. 2022Training Compute-Optimal Large Language Modelscited by 2,927
38 more
  1. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
  2. 2022Self-Instruct: Aligning Language Models with Self-Generated Instructionscited by 2,674
  3. 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
  4. 2025Gemma 3 Technical Reportcited by 1,686
  5. 2024DeepSeek-Coder: When the Large Language Model Meets Programming -- The Rise of Code Intelligencecited by 1,154
  6. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
  7. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  8. 2016Sequence-Level Knowledge Distillationcited by 761
  9. 2023Instruction Tuning with GPT-4cited by 676
  10. 2019On the Efficacy of Knowledge Distillationcited by 377
  11. 2021Knowledge distillation: A good teacher is patient and consistentcited by 255
  12. 2024OpenCodeInterpreter: Integrating Code Generation with Execution and Refinementcited by 242
  13. 2022Knowledge Distillation from A Stronger Teachercited by 204
  14. 2023TheoremQA: A Theorem-driven Question Answering datasetcited by 189
  15. 2024Direct Nash Optimization: Teaching Language Models to Self-Improve with General Preferencescited by 176
  16. 2024Stealing Part of a Production Language Modelcited by 152
  17. 2024DistiLLM: Towards Streamlined Distillation for Large Language Modelscited by 145
  18. 2022Knowledge Distillation with the Reused Teacher Classifiercited by 140
  19. 2021Rethinking Soft Labels for Knowledge Distillation: A Bias-Variance Tradeoff Perspectivecited by 116
  20. 2023f-Divergence Minimization for Sequence-Level Knowledge Distillationcited by 112
  21. 2022Curriculum Temperature for Knowledge Distillationcited by 110
  22. 2024Logit Standardization in Knowledge Distillationcited by 101
  23. 2024Speculative Knowledge Distillation: Bridging the Teacher-Student Gap Through Interleaved Samplingcited by 85
  24. 2024MARIO: MAth Reasoning with code Interpreter Output -- A Reproducible Pipelinecited by 81
  25. 2023One-for-All: Bridge the Gap Between Heterogeneous Architectures in Knowledge Distillationcited by 77
  26. 2025DistiLLM-2: A Contrastive Approach Boosts the Distillation of LLMscited by 73
  27. 2021Annealing Knowledge Distillationcited by 61
  28. 2024Rethinking Kullback-Leibler Divergence in Knowledge Distillation for Large Language Modelscited by 60
  29. 2025Distillation Scaling Lawscited by 59
  30. 2025Infinity Instruct: Scaling Instruction Selection and Synthesis to Enhance Language Modelscited by 49
  31. 2024Revisiting Knowledge Distillation for Autoregressive Language Modelscited by 41
  32. 2023Knowledge Distillation on Graphs: A Surveycited by 38
  33. 2024Knowledge Distillation Based on Transformed Teacher Matchingcited by 37
  34. 2021AlphaNet: Improved Training of Supernets with Alpha-Divergencecited by 36
  35. 2021Preventing Catastrophic Forgetting and Distribution Mismatch in Knowledge Distillation via Synthetic Datacited by 32
  36. 2024MiniPLM: Knowledge Distillation for Pre-Training Language Modelscited by 27
  37. 2024Pre-training Distillation for Large Language Models: A Design Space Explorationcited by 25
  38. 2023A Unified Perspective for Loss-Oriented Imbalanced Learning via Localizationcited by 24

50 most cited of 63 All 63 (JSON)

Built on by

Not cited yet.