Knowledge distillation (KD) is a promising yet challenging model compression technique that transfers rich learning representations from a well-performing but cumbersome teacher model to a compact student model. Previous methods for image super-resolution (SR) mostly compare the feature maps directly or after standardizing the dimensions with basic algebraic operations (e.g. average, dot-product). However, the intrinsic semantic differences among feature maps are overlooked, which are caused by the disparate expressive capacity between the networks. This work presents MiPKD, a multi-granularity mixture of prior KD framework, to facilitate efficient SR model through the feature mixture in a unified latent space and stochastic network block mixture. Extensive experiments demonstrate the effectiveness of the proposed MiPKD method.
@article{arxiv.2404.02573,
title = {Knowledge Distillation with Multi-granularity Mixture of Priors for Image Super-Resolution},
author = {Simiao Li and Yun Zhang and Wei Li and Hanting Chen and Wenjia Wang and Bingyi Jing and Shaohui Lin and Jie Hu},
journal= {arXiv preprint arXiv:2404.02573},
year = {2024}
}