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

面向异构架构的多级解耦关系蒸馏

Multi-Level Decoupled Relational Distillation for Heterogeneous Architectures

School of Information Science and Technology, Fudan University(复旦大学信息科学与技术学院)

Yaoxin Yang, Peng Ye, Weihao Lin, Kangcong Li, Yan Wen, Jia Hao, Tao Chen

AI总结:

针对异构蒸馏未充分利用教师暗知识的问题,提出MLDR-KD框架,通过logit和特征层面的解耦细粒度关系对齐及多尺度动态融合,在四类架构和两个数据集上实现性能提升,最高达4.86%和2.78%。

摘要:

异构蒸馏是将知识从跨架构教师模型迁移到学生模型的有效方法。然而,现有异构蒸馏方法未充分利用教师输出中隐藏的暗知识,限制了性能。为此,我们提出一种名为多级解耦关系知识蒸馏(MLDR-KD)的新框架,以释放异构蒸馏中关系蒸馏的潜力。具体而言,我们首先在logit和特征层面引入解耦细粒度关系对齐(DFRA),以平衡蒸馏暗知识与异构教师模型对正确类别的置信度之间的权衡。然后,应用多尺度动态融合(MSDF)模块,动态融合学生模型不同阶段多尺度特征的投影logit,进一步提升特征层面的方法性能。我们在四种架构(CNN、Transformer、MLP和Mamba)、两个数据集(CIFAR-100和Tiny-ImageNet)上验证了方法。与现有最佳方法相比,MLDR-KD在CIFAR-100上提升学生模型性能最高达4.86%,在Tiny-ImageNet上达2.78%,显示出异构蒸馏中的鲁棒性和通用性。代码即将发布。

英文摘要:

Heterogeneous distillation is an effective way to transfer knowledge from cross-architecture teacher models to student models. However, existing heterogeneous distillation methods do not take full advantage of the dark knowledge hidden in the teacher's output, limiting their performance.To this end, we propose a novel framework named Multi-Level Decoupled Relational Knowledge Distillation (MLDR-KD) to unleash the potential of relational distillation in heterogeneous distillation. Concretely, we first introduce Decoupled Finegrained Relation Alignment (DFRA) in both logit and feature levels to balance the trade-off between distilled dark knowledge and the confidence in the correct category of the heterogeneous teacher model. Then, Multi-Scale Dynamic Fusion (MSDF) module is applied to dynamically fuse the projected logits of multiscale features at different stages in student model, further improving performance of our method in feature level. We verify our method on four architectures (CNNs, Transformers, MLPs and Mambas), two datasets (CIFAR-100 and Tiny-ImageNet). Compared with the best available method, our MLDR-KD improves student model performance with gains of up to 4.86% on CIFAR-100 and 2.78% on Tiny-ImageNet datasets respectively, showing robustness and generality in heterogeneous distillation. Code will be released soon.

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

↑