Lightweight of Supervised Person Re-identification Via Knowledge Distillation

Xiaobin Wang,Jun Wang,Weifeng Liu,Baodi Liu
DOI: https://doi.org/10.1109/icsp56322.2022.9965351
2022-01-01
Abstract:The current state-of-the-art supervised person re-identification has achieved excellent performance. However, these methods all use large models such as ResNet50 as the backbone, and the amount of parameters and computation is enormous, which can not be deployed on small devices. However, the accuracy of lightweight models with few parameters does not meet the demand. To solve this problem, we propose a simple and effective ReID model lightweight method based on knowledge distillation, which introduces knowledge distillation into person re-identification and achieves lightweight by transferring the prior knowledge of the large model to the small model. Our method uses a high-precision model as a teacher model to guide the training of a student model with fewer parameters. It reduces the number of model parameters and computation by making the performance of the student model approach that of the teacher model. Our compression results on the Market-1501 and DukeMTMC-ID datasets show that the compressed model can approach or even outperform the original model with far fewer parameters than the original model, based on only adding knowledge distillation training.
What problem does this paper attempt to address?