用条件生成对抗网络合成类别专属图像,提升无数据知识蒸馏效果
Conditional Pseudo-Supervised Contrast for Data-Free Knowledge Distillation
- 设计条件GAN生成带类别区分的多样化假图像
- 通过教师-学生视图对比学习增强样本多样性
- 适合模型压缩与隐私保护场景下的知识蒸馏应用
无数据知识蒸馏(DFKD)在保护隐私的同时实现模型压缩与传输,近年受到广泛关注。现有方法多依赖生成器合成图像支持蒸馏,但仍存在若干问题:一是未充分利用监督学习优势,缺乏伪监督范式;二是生成图像无法区分不同类别分布,导致样本模糊,影响教师评估;三是未能优化类别内多样性,限制学生模型学习。为此,本文提出条件伪监督对比学习框架(CPSC-DFKD),创新包括:(1) 采用条件生成对抗网络合成类别特异性多样化图像以支持伪监督学习;(2) 改进生成器模块以区分不同类别分布;(3) 提出基于教师与学生视角的伪监督对比学习机制,增强样本多样性。在三个常用数据集上的实验表明,该方法显著提升了学生模型与生成器性能。代码已开源。
原文摘要 · Abstract (English)
Data-free knowledge distillation~(DFKD) is an effective manner to solve model compression and transmission restrictions while retaining privacy protection, which has attracted extensive attention in recent years. Currently, the majority of existing methods utilize a generator to synthesize images to support the distillation. Although the current methods have achieved great success, there are still many issues to be explored. Firstly, the outstanding performance of supervised learning in deep learning drives us to explore a pseudo-supervised paradigm on DFKD. Secondly, current synthesized methods cannot distinguish the distributions of different categories of samples, thus producing ambiguous samples that may lead to an incorrect evaluation by the teacher. Besides, current methods cannot optimize the category-wise diversity samples, which will hinder the student model learning from diverse samples and further achieving better performance. In this paper, to address the above limitations, we propose a novel learning paradigm, i.e., conditional pseudo-supervised contrast for data-free knowledge distillation~(CPSC-DFKD). The primary innovations of CPSC-DFKD are: (1) introducing a conditional generative adversarial network to synthesize category-specific diverse images for pseudo-supervised learning, (2) improving the modules of the generator to distinguish the distributions of different categories, and (3) proposing pseudo-supervised contrastive learning based on teacher and student views to enhance diversity. Comprehensive experiments on three commonly-used datasets validate the performance lift of both the student and generator brought by CPSC-DFKD. The code is available at https://github.com/RoryShao/CPSC-DFKD.git
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。