Multi-Level Decoupled Relational Distillation for Heterogeneous Architectures

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Yaoxin, Ye, Peng, Lin, Weihao, Li, Kangcong, Wen, Yan, Hao, Jia, Chen, Tao
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929706672062464
author Yang, Yaoxin
Ye, Peng
Lin, Weihao
Li, Kangcong
Wen, Yan
Hao, Jia
Chen, Tao
author_facet Yang, Yaoxin
Ye, Peng
Lin, Weihao
Li, Kangcong
Wen, Yan
Hao, Jia
Chen, Tao
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2502_06189
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Multi-Level Decoupled Relational Distillation for Heterogeneous Architectures
Yang, Yaoxin
Ye, Peng
Lin, Weihao
Li, Kangcong
Wen, Yan
Hao, Jia
Chen, Tao
Computer Vision and Pattern Recognition
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.
title Multi-Level Decoupled Relational Distillation for Heterogeneous Architectures
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.06189