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

通过多尺度特征解耦的对比表示蒸馏

Contrastive Representation Distillation via Multi-Scale Feature Decoupling

Key Laboratory of Information Science of Electromagnetic Waves, Fudan University(电磁波信息科学重点实验室,复旦大学)

Cuipeng Wang, Haipeng Wang

AI总结:

本文提出与模型无关的蒸馏框架MSDCRD,通过将全局特征解耦为多尺度局部特征并结合定制对比损失,实现高效的单批次蒸馏,显著提升了同构及异构师生架构的性能。

摘要:

知识蒸馏通过从更强大的教师网络转移知识,在不引入额外参数的情况下提升了紧凑学生网络的性能。在特征空间中,单个全局特征内的局部区域编码了不同但相互依赖的语义信息。以往基于特征的蒸馏方法主要强调全局特征对齐,而忽略了单个全局特征内局部区域的解耦,这通常导致语义混淆和次优性能。此外,传统的对比表示蒸馏由于依赖大型内存缓冲区来存储特征样本,导致效率低下。为了解决这些限制,本文提出了MSDCRD,这是一个与模型无关的蒸馏框架,它系统地将全局特征解耦为多尺度局部特征,并利用由此产生的语义丰富的特征样本,配合定制的样本级和特征级对比损失。这种设计使得仅需单个批次即可实现高效蒸馏,消除了对外部内存的依赖。大量实验表明,MSDCRD不仅在同构师生设置中取得了卓越的性能,而且在特征差异更显著的异构架构中也表现出色,突显了其强大的泛化能力。

英文摘要:

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.

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

↑