Data-free Knowledge Distillation for Fine-grained Visual Categorization

Renrong Shao, Wei Zhang, Jianhua Yin, Jun Wang; Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), 2023, pp. 1515-1525

Abstract


Data-free knowledge distillation (DFKD) is a promising approach for addressing issues related to model compression, security privacy, and transmission restrictions. Although the existing methods exploiting DFKD have achieved inspiring achievements in coarse-grained classification, in practical applications involving fine-grained classification tasks that require more detailed distinctions between similar categories, sub-optimal results are obtained. To address this issue, we propose an approach called DFKD-FGVC that extends DFKD to fine-grained vision categorization (FGVC) tasks. Our approach utilizes an adversarial distillation framework with attention generator, mixed high-order attention distillation, and semantic feature contrast learning. Specifically, we introduce a spatial-wise attention mechanism to the generator to synthesize fine-grained images with more details of discriminative parts. We also utilize the mixed high-order attention mechanism to capture complex interactions among parts and the subtle differences among discriminative features of the fine-grained categories, paying attention to both local features and semantic context relationships. Moreover, we leverage the teacher and student models of the distillation framework to contrast high-level semantic feature maps in the hyperspace, comparing variances of different categories. We evaluate our approach on three widely-used FGVC benchmarks (Aircraft, Cars196, and CUB200) and demonstrate its superior performance.

Related Material


[pdf] [supp]
[bibtex]
@InProceedings{Shao_2023_ICCV, author = {Shao, Renrong and Zhang, Wei and Yin, Jianhua and Wang, Jun}, title = {Data-free Knowledge Distillation for Fine-grained Visual Categorization}, booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)}, month = {October}, year = {2023}, pages = {1515-1525} }