arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.03375cs.LGcs.CV
收藏

面向无数据知识蒸馏的条件伪监督对比学习

Conditional Pseudo-Supervised Contrast for Data-Free Knowledge Distillation

organization= School of Computer Science

Renrong Shao, Wei Zhang, Jun wang

AI总结:

针对现有无数据知识蒸馏方法存在的样本类别分布模糊、多样性不足等问题,提出CPSC-DFKD范式,通过条件生成对抗网络与伪监督对比学习提升学生模型及生成器性能。

摘要:

无数据知识蒸馏(DFKD)是一种在保护隐私的同时解决模型压缩与传输限制问题的有效方法,近年来受到广泛关注。目前,大多数现有方法利用生成器合成图像以支撑蒸馏过程。尽管现有方法已取得显著成效,但仍有诸多问题有待探索。首先,监督学习在深度学习中的优异表现促使我们在DFKD领域探索伪监督范式。其次,当前的合成方法无法区分不同类别样本的分布,因此会生成模糊样本,可能导致教师模型做出错误评估。此外,现有方法无法优化类别层面的样本多样性,这会阻碍学生模型从多样化样本中学习,进而难以取得更优性能。本文针对上述局限提出了一种全新的学习范式,即面向无数据知识蒸馏的条件伪监督对比学习(CPSC-DFKD)。CPSC-DFKD的主要创新点包括:(1)引入条件生成对抗网络合成类别特异性的多样化图像,用于伪监督学习;(2)改进生成器模块以区分不同类别的分布;(3)提出基于教师与学生视图的伪监督对比学习以增强多样性。在三个常用数据集上开展的综合实验验证了CPSC-DFKD能够同时提升学生模型和生成器的性能。代码已开源至https://github.com/RoryShao/CPSC-DFKD.git。

英文摘要:

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

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

↑