arXiv:2504.07691cs.LGcs.CV2025-04AAAI被引 10

跨架构知识蒸馏新方法,让CNN与Transformer互学更高效。

Distilling Knowledge from Heterogeneous Architectures for Semantic Segmentation

  • 将异构模型中间特征投影到对齐的逻辑空间,消除架构差异影响。
  • 引入知识混合与评估机制,动态选择适合学生的知识。
  • 在多个主流数据集上超越现有方法,支持多种模型组合蒸馏。

当前语义分割的知识蒸馏方法主要关注同构架构间的知识传递,忽略了不同归纳偏置架构(如CNN与Transformer)所蕴含的多样化知识。为此,本文首次提出一种面向异构架构的通用知识蒸馏方法——HeteroAKD。由于异构架构间存在显著差异,直接跨架构传递知识面临挑战。为此,我们通过将教师与学生模型的中间特征映射到统一的逻辑空间,消除架构特异性信息的影响。同时,设计了教师-学生知识混合机制(KMM)与知识评估机制(KEM),通过评估异构师生知识的可靠性及其差异性,实现定制化知识传输。在三个主流基准测试上,采用多种教师-学生组合进行的大量实验表明,该方法在异构架构间蒸馏效果优于现有最先进方法。

原文摘要 · Abstract (English)

Current knowledge distillation (KD) methods for semantic segmentation focus on guiding the student to imitate the teacher's knowledge within homogeneous architectures. However, these methods overlook the diverse knowledge contained in architectures with different inductive biases, which is crucial for enabling the student to acquire a more precise and comprehensive understanding of the data during distillation. To this end, we propose for the first time a generic knowledge distillation method for semantic segmentation from a heterogeneous perspective, named HeteroAKD. Due to the substantial disparities between heterogeneous architectures, such as CNN and Transformer, directly transferring cross-architecture knowledge presents significant challenges. To eliminate the influence of architecture-specific information, the intermediate features of both the teacher and student are skillfully projected into an aligned logits space. Furthermore, to utilize diverse knowledge from heterogeneous architectures and deliver customized knowledge required by the student, a teacher-student knowledge mixing mechanism (KMM) and a teacher-student knowledge evaluation mechanism (KEM) are introduced. These mechanisms are performed by assessing the reliability and its discrepancy between heterogeneous teacher-student knowledge. Extensive experiments conducted on three main-stream benchmarks using various teacher-student pairs demonstrate that our HeteroAKD outperforms state-of-the-art KD methods in facilitating distillation between heterogeneous architectures.

知识蒸馏异构架构语义分割CNN Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。