通过多尺度解耦特征,提升知识蒸馏效率与效果。
Contrastive Representation Distillation via Multi-Scale Feature Decoupling
- 将全局特征解耦为多尺度局部特征,避免语义混淆。
- 单批次训练即可完成蒸馏,无需外部存储内存。
- 适用于异构模型,通用性强,性能优于现有方法。
知识蒸馏通过从强大教师网络向紧凑学生网络迁移知识,在不增加参数的情况下提升性能。在特征空间中,单个全局特征内的局部区域包含相互关联但语义各异的信息。以往基于特征的蒸馏方法主要关注全局特征对齐,忽略了单个全局特征内部局部区域的解耦,常导致语义混淆和性能下降。此外,传统对比表示蒸馏依赖大型内存缓冲区存储特征样本,效率低下。为此,本文提出MSDCRD——一种模型无关的蒸馏框架,系统地将全局特征解耦为多尺度局部特征,并利用生成的语义丰富特征样本,采用定制的样本级和特征级对比损失实现高效蒸馏。该设计仅需单个批次即可完成,彻底消除对外部内存的依赖。大量实验表明,MSDCRD不仅在同构师生架构下表现优异,更在特征差异更大的异构架构中展现出更强的泛化能力。
原文摘要 · Abstract (English)
Knowledge distillation enhances the performance of compact student networks by transferring knowledge from more powerful teacher networks without introducing additional parameters. In the feature space, local regions within an individual global feature encode distinct yet interdependent semantic information. Previous feature-based distillation methods mainly emphasize global feature alignment while neglecting the decoupling of local regions within an individual global feature, which often results in semantic confusion and suboptimal performance. Moreover, conventional contrastive representation distillation suffers from low efficiency due to its reliance on a large memory buffer to store feature samples. To address these limitations, this work proposes MSDCRD, a model-agnostic distillation framework that systematically decouples global features into multi-scale local features and leverages the resulting semantically rich feature samples with tailored sample-wise and feature-wise contrastive losses. This design enables efficient distillation using only a single batch, eliminating the dependence on external memory. Extensive experiments demonstrate that MSDCRD achieves superior performance not only in homogeneous teacher-student settings but also in heterogeneous architectures where feature discrepancies are more pronounced, highlighting its strong generalization capability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。