通过虚拟关系匹配提升知识蒸馏效果,显著改善学生模型性能。
VRM: Knowledge Distillation via Virtual Relation Matching
- 用虚拟视图构建关联图,传递样本间多维度关系知识
- 在ImageNet上实现ResNet50到MobileNetV2蒸馏74.0%准确率新纪录
- 适用于多种模型架构与任务,尤其适合轻量化部署场景
知识蒸馏旨在将能力强但复杂的教师模型知识迁移至轻量级学生模型。近年来,基于关系的蒸馏方法表现落后于实例匹配类方法。本文通过识别并解决关系类方法中的过拟合与虚假响应问题,重新激活了关系蒸馏。我们创新性地利用虚拟视图和关系构建紧凑的亲和图,封装丰富的样本间、类别间及视角间相关性,作为新型知识传递信号。学生模型因此获得更丰富的指导与更强正则化。为进一步缓解虚假响应影响,我们动态剔除冗余不可靠边。在CIFAR-100、ImageNet和MS-COCO上的大量实验表明,所提虚拟关系匹配(VRM)方法持续刷新多项基准记录。例如,在ImageNet上首次实现ResNet50→MobileNetV2蒸馏达到74.0%准确率,并使DeiT-T在CIFAR-100上相较使用ResNet56教师提升14.44%。
原文摘要 · Abstract (English)
Knowledge distillation (KD) aims to transfer the knowledge of a more capable yet cumbersome teacher model to a lightweight student model. In recent years, relation-based KD methods have fallen behind, as their instance-matching counterparts dominate in performance. In this paper, we revive relational KD by identifying and tackling several key issues in relation-based methods, including their susceptibility to overfitting and spurious responses. Specifically, we transfer novelly constructed affinity graphs that compactly encapsulate a wealth of beneficial inter-sample, inter-class, and inter-view correlations by exploiting virtual views and relations as a new kind of knowledge. As a result, the student has access to richer guidance signals and stronger regularisation throughout the distillation process. To further mitigate the adverse impact of spurious responses, we prune the affinity graphs by dynamically detaching redundant and unreliable edges. Extensive experiments on CIFAR-100, ImageNet, and MS-COCO datasets demonstrate the superior performance of the proposed virtual relation matching (VRM) method, where it consistently sets new state-of-the-art records over a range of models, architectures, tasks, and set-ups. For instance, VRM for the first time hits 74.0% accuracy for ResNet50-to-MobileNetV2 distillation on ImageNet, and improves DeiT-T by 14.44% on CIFAR-100 with a ResNet56 teacher.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。