arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.12037cs.CV

无数据知识蒸馏用于细粒度视觉分类

Data-free Knowledge Distillation for Fine-grained Visual Categorization

School of Computer Science and Technology, East China Normal University(华东师范大学计算机科学与技术学院);School of Computer Science and Technology, Shandong University(山东大学计算机科学与技术学院)

Renrong Shao, Wei Zhang, Jianhua Yin, Jun Wang

AI总结:

提出DFKD-FGVC方法,通过对抗蒸馏框架、注意力生成器、混合高阶注意力蒸馏和语义特征对比学习,将无数据知识蒸馏扩展到细粒度视觉分类,在三个基准上取得优越性能。

摘要:

无数据知识蒸馏(DFKD)是解决模型压缩、安全隐私和传输限制相关问题的一种有前景的方法。尽管现有的利用DFKD的方法在粗粒度分类中取得了令人鼓舞的成就,但在需要更细致区分相似类别的细粒度分类任务的实际应用中,获得的结果是次优的。为了解决这个问题,我们提出了一种名为DFKD-FGVC的方法,将DFKD扩展到细粒度视觉分类(FGVC)任务。我们的方法利用了一个带有注意力生成器的对抗性蒸馏框架、混合高阶注意力蒸馏和语义特征对比学习。具体来说,我们向生成器引入了空间注意力机制,以合成具有更多判别性部位细节的细粒度图像。我们还利用混合高阶注意力机制来捕获部位之间的复杂交互以及细粒度类别判别性特征之间的细微差异,同时关注局部特征和语义上下文关系。此外,我们利用蒸馏框架的教师和学生模型在高维空间中对比高层语义特征图,比较不同类别的方差。我们在三个广泛使用的FGVC基准(Aircraft、Cars196和CUB200)上评估了我们的方法,并展示了其优越的性能。

英文摘要:

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 visual 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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑