mlsanityData to 14 Oct 2026

Adaptive Weighting in Knowledge Distillation: An Axiomatic Framework for Multi-Scale Teacher Ensemble Optimization

arXiv 2601.17910 ↗First posted 25 Jan 2026cs.LGcited by 1

On the map: Knowledge distillation → General knowledge distillation methods

Abstract (arXiv)

Knowledge distillation with multiple teachers is increasingly used to improve robustness, efficiency, and safety, yet existing approaches rely largely on heuristic or implementation-specific weighting schemes. This paper develops an operator-agnostic axiomatic framework for adaptive weighting in multi-teacher knowledge distillation across three complementary scales: token, task, and context. We formalize structural conditions under which adaptive weighting operators are well-defined, admit multiple non-equivalent implementations, and can be hierarchically composed via product-structure normalization. Within this framework, we establish existence and non-uniqueness of conforming operators, characterize convergence of gradient-based optimization under standard assumptions, analyze stability and perturbation robustness, and provide an abstract formulation of safety-constrained distillation. The results decouple theoretical guarantees from specific weighting formulas, enabling principled analysis of adaptive distillation methods under heterogeneity, distribution shift, and safety constraints.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in General knowledge distillation methods, one of 12 directions in Knowledge distillation. Builds on 10 of its 19 references that are in the map. First posted on arXiv 25 Jan 2026.

Builds on

10 of 19 references.

Closest ancestors · 6

Same direction, newest first

  1. 2026Recursive Meta-Distillation: An Axiomatic Framework for Iterative Knowledge Refinement19 Jan
  2. 2026Multi-Teacher Ensemble Distillation: A Mathematical Framework for Probability-Domain Knowledge Aggregation14 Jan
  3. 2026Sparse Knowledge Distillation: A Mathematical Framework for Probability-Domain Temperature Scaling and Multi-Stage Compression6 Jan
  4. 2025A Comprehensive Survey on Knowledge Distillation15 Mar
  5. 2025Multi-Teacher Knowledge Distillation with Reinforcement Learning for Visual Recognition22 Feb
  6. 2021Adaptive Multi-Teacher Multi-level Knowledge Distillation6 Mar

Other references · 4

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,358
  3. 2018Multi-Task Learning as Multi-Objective Optimizationcited by 821
  4. 2026Hallucinations Live in Variancecited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2023Adaptive Multi-Teacher Knowledge Distillation with Meta-Learning
  2. 2026Integrating Knowledge Distillation Methods: A Sequential Multi-Stage Framework
  3. 2026Uncertainty-Aware Knowledge Distillation for Multimodal Large Language Models
  4. 2022Knowledge Distillation via Weighted Ensemble of Teaching Assistants
  5. 2024AdaKD: Dynamic Knowledge Distillation of ASR models using Adaptive Loss Weighting
  6. 2025Knowledge Distillation with Adapted Weight
  7. 2021Confidence-Aware Multi-Teacher Knowledge Distillation
  8. 2026On the Generalization of Knowledge Distillation: An Information-Theoretic View