提出简单有效框架,提升跨架构知识蒸馏效果。
Heterogeneous Complementary Distillation
- 融合师生互补特征,通过共享logits对齐表示
- 在CIFAR-100、ImageNet-1K等数据集上超越主流方法
- 适合需要高效跨模型知识迁移的场景
知识蒸馏(KD)将复杂教师模型的暗知识迁移到紧凑学生模型中。然而,如视觉变压器(ViT)到ResNet18这类异构架构蒸馏面临空间特征表示差异的挑战。传统方法多针对同构架构设计,难以有效应对差异。尽管近期出现了一些异构KD方法,但往往计算开销大、结构复杂,或过度依赖logit对齐,限制了互补特征的利用。为此,本文提出异构互补蒸馏(HCD),一种简单而有效的框架,通过整合互补的师生特征来对齐共享logits。这些logits被分解并施加约束,以促进多样化的知识传递。具体而言,HCD通过卷积投影器和自适应池化处理学生中间特征,将其与教师最后一层特征拼接,并经由包含全连接层的互补特征映射模块(CFM)生成共享logits。此外,引入子logit解耦蒸馏(SDD),将共享logits划分为n个子logits,与教师logits融合以修正分类结果。为确保子logits多样性并减少冗余知识传递,提出正交性损失(OL)。HCD在保留学生特有优势的同时,有效利用教师知识,提升了学生的鲁棒性和泛化能力。在CIFAR-100、细粒度数据集(如CUB200)以及ImageNet-1K上的大量实验表明,HCD优于当前最优的蒸馏方法,证明其是异构蒸馏的有效解决方案。
原文摘要 · Abstract (English)
Knowledge distillation (KD)transfers the dark knowledge from a complex teacher to a compact student. However, heterogeneous architecture distillation, such as Vision Transformer (ViT) to ResNet18, faces challenges due to differences in spatial feature representations.Traditional KD methods are mostly designed for homogeneous architectures and hence struggle to effectively address the disparity. Although heterogeneous KD approaches have been developed recently to solve these issues, they often incur high computational costs and complex designs, or overly rely on logit alignment, which limits their ability to leverage the complementary features. To overcome these limitations, we propose Heterogeneous Complementary Distillation (HCD),a simple yet effective framework that integrates complementary teacher and student features to align representations in shared logits.These logits are decomposed and constrained to facilitate diverse knowledge transfer to the student. Specifically, HCD processes the student's intermediate features through convolutional projector and adaptive pooling, concatenates them with teacher's feature from the penultimate layer and then maps them via the Complementary Feature Mapper (CFM) module, comprising fully connected layer,to produce shared logits.We further introduce Sub-logit Decoupled Distillation (SDD) that partitions the shared logits into n sub-logits, which are fused with teacher's logits to rectify classification.To ensure sub-logit diversity and reduce redundant knowledge transfer, we propose an Orthogonality Loss (OL).By preserving student-specific strengths and leveraging teacher knowledge,HCD enhances robustness and generalization in students.Extensive experiments on the CIFAR-100, Fine-grained (e.g., CUB200)and ImageNet-1K datasets demonstrate that HCD outperforms state-of-the-art KD methods,establishing it as an effective solution for heterogeneous KD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。