提出跨模型知识蒸馏新方法,保留空间信息提升性能
SFKD: Spatial--Frequency Joint-Aware Heterogeneous Knowledge Distillation via Multi-Level Wavelet Spectral Interaction

- 用多层小波变换分离空间与频域特征
- 在多个数据集上超越现有跨模型蒸馏方法
- 适合需要跨架构知识迁移的场景
现有知识蒸馏多聚焦同构模型(如CNN到CNN),忽视了异构模型间知识迁移的灵活性与潜力。由于异构模型固有的归纳偏置差异导致表征空间分布不一致,以往方法常弱化或丢弃空间信息。然而,表征中的空间信息往往编码可迁移的全局结构语义及架构特异的局部细节,不应被直接忽略。为此,我们提出空间-频率联合感知的异构知识蒸馏框架SFKD。通过利用小波变换的空间局部性与傅里叶表示对全局能量分布的刻画优势,首先对表征进行多级离散小波变换,显式解耦空间信息。生成的小波子带经双流双阶段精炼模块优化后,再结合高斯滤波频率损失,选择性捕获有信息量的全局特征。在多个基准数据集上,无论同构还是异构模型设置下,实验均验证了本方法的优越性。
原文摘要 · Abstract (English)
Most existing knowledge distillation methods focus on homogeneous models (e.g., CNN-to-CNN), thereby overlooking the flexibility and potential of knowledge transfer across heterogeneous models. Due to intrinsic inductive bias discrepancies between heterogeneous models that cause spatial distribution inconsistencies, prior heterogeneous distillation methods often weaken or discard spatial information in heterogeneous representations. However, the spatial information in representations often encodes transferable global structural semantics as well as architecture-specific local details, and therefore should not be directly ignored. To better leverage the spatial information encoded in heterogeneous representations, we propose a Spatial-Frequency Joint-Aware Heterogeneous Knowledge Distillation framework (SFKD). By leveraging the complementary properties of wavelet transform spatial locality and Fourier representations in characterizing global energy distributions, we first apply multi-level discrete wavelet transform to explicitly decouple spatial information. The resulting wavelet sub-bands are further refined by a dual-stream dual-stage refinement module, and finally combined with a Gaussian-filtered frequency loss to selectively capture informative global information. Extensive experiments on multiple benchmark datasets under both homogeneous and heterogeneous models demonstrate the superiority of our method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。