#335 · Primary category: Computer Vision

mdistiller

cifar coco computer-vision cvpr2022 deep-learning iccv2023 imagenet knowledge-distillation pytorch

The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679 and [ICCV2023] DOT: A Distillation-Oriented Trainer https://openaccess.thecvf.com/content/ICCV2023/papers/Zhao_DOT_A_Distillation-Oriented_Trainer_ICCV_2023_paper.pdf

Project last updated:11/05/23

GitHub Stars

897

Forks

131

Contributors

2

License

MIT

Why we included this project

Shrinking a heavy image-classification model for deployment usually costs accuracy, and mdistiller goes after that trade-off head-on. The repository bundles the code behind Decoupled Knowledge Distillation (DKD), published at CVPR 2022, and DOT, a distillation-oriented trainer from ICCV 2023, both by the repo's authors. These methods keep a compact student network close to its larger teacher in accuracy, and the package ships reproducible setups on standard benchmarks such as ImageNet and CIFAR-100, plus COCO-style object detection through a Detectron2-based pipeline. ML engineers who need a lighter classifier can run these configs to compare DKD and DOT against plain KD and older techniques like FitNet, AT, and ReviewKD instead of reimplementing the papers by hand, and a clean distiller interface lets you plug in your own method. Plan to configure and script things yourself, since this is a research toolkit rather than a turnkey service.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category