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

基于掩码图像建模的异构生成式知识蒸馏

Heterogeneous Generative Knowledge Distillation with Masked Image Modeling

Beihang University(北京航空航天大学)

Ziming Wang, Shumin Han, Xiaodi Wang, Jing Hao, Xianbin Cao, Baochang Zhang

AI总结:

本文提出首个基于掩码图像建模的异构生成式知识蒸馏方法H-GKD,通过稀疏卷积UNet学生网络桥接Transformer教师与CNN学生,在图像分类、检测和分割任务中取得最先进性能。

摘要:

基于CNN的小型模型通常需要在部署到计算资源受限的边缘设备之前,从大型模型中迁移知识。掩码图像建模(MIM)方法在各种视觉任务中取得了巨大成功,但在异构深度模型的知识蒸馏中仍基本未被探索。原因主要在于基于Transformer的大型模型与基于CNN的小型网络之间存在显著差异。在本文中,我们开发了首个基于MIM的异构生成式知识蒸馏(H-GKD)方法,该方法能够以生成式自监督的方式高效地将知识从大型Transformer模型迁移到小型CNN模型。我们的方法通过训练一个具有稀疏卷积的UNet风格学生网络,在基于Transformer的模型和CNN之间架起桥梁,该学生网络能够有效模仿教师在掩码建模上推断出的视觉表示。我们的方法是一种简单而有效的学习范式,用于从异构教师模型中学习数据的视觉表示和分布,这些教师模型可以使用先进的生成方法进行预训练。大量实验表明,该方法能很好地适应各种模型和规模,在图像分类、目标检测和语义分割任务中持续取得最先进的性能。例如,在ImageNet 1K数据集上,H-GKD将Resnet50(稀疏)的准确率从76.98%提高到80.01%。

英文摘要:

Small CNN-based models usually require transferring knowledge from a large model before they are deployed in computationally resource-limited edge devices. Masked image modeling (MIM) methods achieve great success in various visual tasks but remain largely unexplored in knowledge distillation for heterogeneous deep models. The reason is mainly due to the significant discrepancy between the Transformer-based large model and the CNN-based small network. In this paper, we develop the first Heterogeneous Generative Knowledge Distillation (H-GKD) based on MIM, which can efficiently transfer knowledge from large Transformer models to small CNN-based models in a generative self-supervised fashion. Our method builds a bridge between Transformer-based models and CNNs by training a UNet-style student with sparse convolution, which can effectively mimic the visual representation inferred by a teacher over masked modeling. Our method is a simple yet effective learning paradigm to learn the visual representation and distribution of data from heterogeneous teacher models, which can be pre-trained using advanced generative methods. Extensive experiments show that it adapts well to various models and sizes, consistently achieving state-of-the-art performance in image classification, object detection, and semantic segmentation tasks. For example, in the Imagenet 1K dataset, H-GKD improves the accuracy of Resnet50 (sparse) from 76.98% to 80.01%.

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

↑