English

REDistill: Robust Estimator Distillation for Balancing Robustness and Efficiency

Machine Learning 2026-02-05 v1 Computer Vision and Pattern Recognition

Abstract

Knowledge Distillation (KD) transfers knowledge from a large teacher model to a smaller student by aligning their predictive distributions. However, conventional KD formulations - typically based on Kullback-Leibler divergence - assume that the teacher provides reliable soft targets. In practice, teacher predictions are often noisy or overconfident, and existing correction-based approaches rely on ad-hoc heuristics and extensive hyper-parameter tuning, which hinders generalization. We introduce REDistill (Robust Estimator Distillation), a simple yet principled framework grounded in robust statistics. REDistill replaces the standard KD objective with a power divergence loss, a generalization of KL divergence that adaptively downweights unreliable teacher output while preserving informative logit relationships. This formulation provides a unified and interpretable treatment of teacher noise, requires only logits, integrates seamlessly into existing KD pipelines, and incurs negligible computational overhead. Extensive experiments on CIFAR-100 and ImageNet-1k demonstrate that REDistill consistently improves student accuracy in diverse teacher-student architectures. Remarkably, it achieves these gains without model-specific hyper-parameter tuning, underscoring its robustness and strong generalization to unseen teacher-student pairs.

Keywords

Cite

@article{arxiv.2602.04677,
  title  = {REDistill: Robust Estimator Distillation for Balancing Robustness and Efficiency},
  author = {Ondrej Tybl and Lukas Neumann},
  journal= {arXiv preprint arXiv:2602.04677},
  year   = {2026}
}
R2 v1 2026-07-01T09:36:07.421Z