arrow
Return

Distillation embedded absorbable pruning for fast object re-identification

delete2024-08-01
delete3
PRE
AI
Y
Yi Xie
H
Hanxiao Wu
朱建清 (Jianqing Zhu) *
H
Huanqiang Zeng
DOI:10.1016/j.patcog.2024.110437delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Combining knowledge distillation (KD) and network pruning (NP) shows promise in learning a light network to accelerate object re -identification. However, KD requires an untrained student network to establish more critical connections in early epochs, but NP demands a well -trained student network to avoid destroying critical connections. This presents a dilemma, potentially leading to a collapse of the student network and harming object Re -ID performance. For that, we propose a distillation embedded absorbable pruning (DEAP) method. We design a pruner -convolution -pruner (PCP) unit to resolve the dilemma by loading NP's sparse regularization on extra untrained pruners. Additionally, we propose an asymmetric relation knowledge distillation method. It readily transfers feature representation knowledge and asymmetric pairwise similarity knowledge without using additional adaptation modules. Finally, we apply re -parameterization to absorb pruners of PCP units to simplify student networks. Experiments demonstrate the superiority of DEAP, such as on the VeRi-776 dataset, with ResNet-101 as a teacher, DEAP saves 73.24% of model parameters and 71.98% of floating-point operations without sacrificing accuracy.
Keywords:
Object re-identification
Knowledge distillation
Network pruning
Re-parameterization

Journal

Pattern Recognition cover
Pattern Recognition
IF:
7.6
Papers:
1.3W
Citations:
4.5W

Organization

H
huaqiao university
Scholars:
1.0W
Papers: 7.0K
Citations: 131