arXiv:2505.11897cs.CV2025-05被引 2

通过高频细节转移提升细粒度图像识别的模型压缩效果

FiGKD: Fine-Grained Knowledge Distillation via High-Frequency Detail Transfer

  • 用离散小波变换分解教师模型输出,分离内容与细节信号
  • 只传递高频细节信息,显著提升在细粒度数据集上的准确率
  • 无需中间特征,适配各类模型结构,适合资源受限场景

知识蒸馏(KD)是一种将高容量教师模型的知识迁移到小型学生模型的常用技术,通过对其输出分布进行对齐实现。然而,在细粒度视觉识别任务中,现有方法表现不佳,因为它们将教师输出的logits视为单一信号,假设所有信息都同等有益于学生模型,导致学生被冗余信号淹没,难以捕捉教师精细的决策边界。为此,我们提出细粒度知识蒸馏(FiGKD),一种基于频率感知的框架,利用离散小波变换(DWT)将模型logits分解为低频(内容)和高频(细节)分量。FiGKD仅选择性地传输高频分量,这些分量编码了教师的语义决策模式,而丢弃已由真实标签覆盖的低频内容。该方法简单、与架构无关,且无需访问中间特征图。在CIFAR-100、TinyImageNet及多个细粒度识别基准上的大量实验表明,无论教师-学生配置如何,FiGKD始终优于当前主流的基于logits和基于特征的蒸馏方法。结果证实,频率感知的logits分解能实现更高效、更有效的知识迁移,尤其在资源受限环境下表现突出。

原文摘要 · Abstract (English)

Knowledge distillation (KD) is a widely adopted technique for transferring knowledge from a high-capacity teacher model to a smaller student model by aligning their output distributions. However, existing methods often underperform in fine-grained visual recognition tasks, where distinguishing subtle differences between visually similar classes is essential. This performance gap stems from the fact that conventional approaches treat the teacher's output logits as a single, undifferentiated signal-assuming all contained information is equally beneficial to the student. Consequently, student models may become overloaded with redundant signals and fail to capture the teacher's nuanced decision boundaries. To address this issue, we propose Fine-Grained Knowledge Distillation (FiGKD), a novel frequency-aware framework that decomposes a model's logits into low-frequency (content) and high-frequency (detail) components using the discrete wavelet transform (DWT). FiGKD selectively transfers only the high-frequency components, which encode the teacher's semantic decision patterns, while discarding redundant low-frequency content already conveyed through ground-truth supervision. Our approach is simple, architecture-agnostic, and requires no access to intermediate feature maps. Extensive experiments on CIFAR-100, TinyImageNet, and multiple fine-grained recognition benchmarks show that FiGKD consistently outperforms state-of-the-art logit-based and feature-based distillation methods across a variety of teacher-student configurations. These findings confirm that frequency-aware logit decomposition enables more efficient and effective knowledge transfer, particularly in resource-constrained settings.

知识蒸馏细粒度识别小波变换模型压缩

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