arXiv:2411.17046cs.CV2024-11被引 5

不依赖真实数据,在ImageNet上用多分辨率生成图像实现高效知识蒸馏。

Large-Scale Data-Free Knowledge Distillation for ImageNet via Multi-Resolution Data Generation

  • 先生成低分辨率图像,再用类激活图保留关键特征。
  • 在ImageNet上实现接近两倍的性能提升,优于现有方法。
  • 适合追求高精度且无原始数据的模型压缩场景。

数据无关知识蒸馏(DFKD)是一种无需原始训练数据即可将教师模型知识迁移至学生模型的技术。尽管该技术在CIFAR10和CIFAR100等小数据集上表现良好,但在ImageNet等大尺寸高分辨率数据集上仍面临挑战。以往方法直接生成224×224的高分辨率合成图像,缺乏真实图像信息,导致生成图像噪声大、类别特征缺失。此外,大规模数据生成带来的计算开销也难以承受。本文提出多分辨率数据无关(MUSE)方法,通过在低分辨率下生成图像,并利用类激活图(CAMs)确保关键类别特征保留。同时引入多分辨率生成与嵌入多样性机制,增强潜在空间表示能力,显著提升模型多样性。实验表明,MUSE在各类图像数据集上均达到领先性能,尤其在ImageNet及子集上提升幅度接近两位数。代码已开源:https://github.com/tmtuan1307/muse。

原文摘要 · Abstract (English)

Data-Free Knowledge Distillation (DFKD) is an advanced technique that enables knowledge transfer from a teacher model to a student model without relying on original training data. While DFKD methods have achieved success on smaller datasets like CIFAR10 and CIFAR100, they encounter challenges on larger, high-resolution datasets such as ImageNet. A primary issue with previous approaches is their generation of synthetic images at high resolutions (e.g., $224 \times 224$) without leveraging information from real images, often resulting in noisy images that lack essential class-specific features in large datasets. Additionally, the computational cost of generating the extensive data needed for effective knowledge transfer can be prohibitive. In this paper, we introduce MUlti-reSolution data-freE (MUSE) to address these limitations. MUSE generates images at lower resolutions while using Class Activation Maps (CAMs) to ensure that the generated images retain critical, class-specific features. To further enhance model diversity, we propose multi-resolution generation and embedding diversity techniques that strengthen latent space representations, leading to significant performance improvements. Experimental results demonstrate that MUSE achieves state-of-the-art performance across both small- and large-scale datasets, with notable performance gains of up to two digits in nearly all ImageNet and subset experiments. Code is available at https://github.com/tmtuan1307/muse.

知识蒸馏图像生成模型压缩无数据训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。