用大模型知识蒸馏提升边缘模型性能,解决大小模型差异难题。
CustomKD: Customizing Large Vision Foundation for Edge Model Improvement via Knowledge Distillation
- 将大视觉模型特征定制适配到小模型,减少架构差异。
- 在无标签数据场景下显著提升边缘模型效果,达新基准。
- 适合资源受限设备上部署高性能视觉模型的开发者。
我们提出一种新型知识蒸馏方法 CustomKD,有效利用大视觉基础模型(LVFMs)如 DINOv2、CLIP 来提升边缘模型(如 MobileNetV3)性能。尽管 LVFMs 近期进展显著,其在知识蒸馏中用于增强边缘模型的潜力尚未充分探索。由于教师模型与学生模型间存在容量与架构异质性,导致知识迁移效果受限。实验表明,虽使用更大骨干网络(如 ViT-S 到 ViT-L)可提升教师模型下游任务表现,但知识蒸馏对学生的增益远低于教师自身。CustomKD 通过将大模型中的通用特征定制适配至目标学生模型,降低模型差异。该方法不仅传递原始知识,还对齐师生特征,使学生更易理解并克服结构鸿沟。在无监督域适应(OfficeHome、DomainNet)和半监督学习(CIFAR-100 400 标注样本、ImageNet 1% 标注样本)等场景下,显著提升边缘模型性能,达到新的最先进水平。
原文摘要 · Abstract (English)
We propose a novel knowledge distillation approach, CustomKD, that effectively leverages large vision foundation models (LVFMs) to enhance the performance of edge models (e.g., MobileNetV3). Despite recent advancements in LVFMs, such as DINOv2 and CLIP, their potential in knowledge distillation for enhancing edge models remains underexplored. While knowledge distillation is a promising approach for improving the performance of edge models, the discrepancy in model capacities and heterogeneous architectures between LVFMs and edge models poses a significant challenge. Our observation indicates that although utilizing larger backbones (e.g., ViT-S to ViT-L) in teacher models improves their downstream task performances, the knowledge distillation from the large teacher models fails to bring as much performance gain for student models as for teacher models due to the large model discrepancy. Our simple yet effective CustomKD customizes the well-generalized features inherent in LVFMs to a given student model in order to reduce model discrepancies. Specifically, beyond providing well-generalized original knowledge from teachers, CustomKD aligns the features of teachers to those of students, making it easy for students to understand and overcome the large model discrepancy overall. CustomKD significantly improves the performances of edge models in scenarios with unlabeled data such as unsupervised domain adaptation (e.g., OfficeHome and DomainNet) and semi-supervised learning (e.g., CIFAR-100 with 400 labeled samples and ImageNet with 1% labeled samples), achieving the new state-of-the-art performances.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。