通过对比学习提升学生模型类别表征能力,动态调整难样本学习权重。
Preview-based Category Contrastive Learning for Knowledge Distillation
- 基于对比学习同时优化实例特征与类别中心关系
- 在CIFAR-100和ImageNet上超越现有最优方法
- 适合需要高效压缩且保持分类性能的场景
知识蒸馏通过将大模型(教师)的知识迁移至小模型(学生)来实现模型压缩。现有方法多关注实例级特征或预测的一致性,忽视类别级信息及样本难度差异,导致性能受限。为此,我们提出一种基于预览的类别对比学习知识蒸馏方法(PCKD)。该方法以对比学习方式蒸馏实例级特征对应关系及实例特征与类别中心之间的关联,显式优化类别表征,增强类别中心的区分性,提升分类效果。此外,引入新颖的预览策略,根据样本难度动态调整学生模型的学习权重:对困难样本赋予较小权重作为预览,引导训练过程。相比传统等权重处理或仅过滤难样本的课程学习,该方法更有效。在多个挑战性数据集(包括CIFAR-100和ImageNet)上的大量实验表明,PCKD优于当前最优方法。
原文摘要 · Abstract (English)
Knowledge distillation is a mainstream algorithm in model compression by transferring knowledge from the larger model (teacher) to the smaller model (student) to improve the performance of student. Despite many efforts, existing methods mainly investigate the consistency between instance-level feature representation or prediction, which neglects the category-level information and the difficulty of each sample, leading to undesirable performance. To address these issues, we propose a novel preview-based category contrastive learning method for knowledge distillation (PCKD). It first distills the structural knowledge of both instance-level feature correspondence and the relation between instance features and category centers in a contrastive learning fashion, which can explicitly optimize the category representation and explore the distinct correlation between representations of instances and categories, contributing to discriminative category centers and better classification results. Besides, we introduce a novel preview strategy to dynamically determine how much the student should learn from each sample according to their difficulty. Different from existing methods that treat all samples equally and curriculum learning that simply filters out hard samples, our method assigns a small weight for hard instances as a preview to better guide the student training. Extensive experiments on several challenging datasets, including CIFAR-100 and ImageNet, demonstrate the superiority over state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。