本文已被:浏览 346次 下载 1246次
Received:April 02, 2025 Revised:April 27, 2025
Received:April 02, 2025 Revised:April 27, 2025
中文摘要: 知识蒸馏通过传递教师模型知识提升学生模型性能. 然而对于轻量化的学生模型而言, 全盘吸收教师特征图内的隐含知识是困难的, 为此本文提出一种基于空间-通道注意力精炼的协同蒸馏方法(SCAR-KD), 通过从原始特征图中提炼关键判别信息, 缓解学生与教师的语义分歧. 具体而言, 本文采用了一种多尺度空间-通道注意力模块(SCSA), 从师生特征图的通道和空间维度中精炼出具有判别性的注意力增强特征进行蒸馏, 同时解耦出空间注意力图加权给原始特征图进行动态蒸馏. 该方法实现了双重知识迁移. 实验结果表明, YOLOv8n-SCAR-KD相较于基线YOLOv8n在VOC和VisDrone数据集上mAP@0.5:0.95分别从64.1%提升至65.3%, 从20.7%提升至21.6%, 超过了现有的主流蒸馏方法, 验证了方法的有效性.
Abstract:Knowledge distillation enhances student model performance by transferring knowledge from the teacher model. However, for lightweight student models, it is challenging to fully absorb the implicit knowledge contained in the teacher feature maps. To address this, this study proposes a spatial-channel attention refinement-based collaborative distillation (SCAR-KD) method, which extracts key discriminative information from the original feature maps to mitigate the semantic discrepancy between the student and teacher. Specifically, we adopt a multi-scale spatial-channel attention (SCSA) module to refine discriminative attention-enhanced features from the teacher feature maps along both the channel and spatial dimensions, while decoupling the spatial attention map to weight the original feature maps for dynamic distillation. This approach achieves dual knowledge transfer. Experimental results demonstrate that the mAP@0.5:0.95 of YOLOv8n-SCAR-KD model improved from 64.1% to 65.3% on the VOC dataset and from 20.7% to 21.6% on the VisDrone dataset, outperforming current mainstream distillation methods and validating the effectiveness of the proposed method.
keywords: knowledge distillation (KD) object detection feature imitation attention mechanism semantic discrepancy
文章编号: 中图分类号: 文献标志码:
基金项目:
引用文本:
叶超越,钟良琪,闫胜业.空间-通道注意力协同的精炼特征知识蒸馏.计算机系统应用,2025,34(11):270-278
YE Chao-Yue,ZHONG Liang-Qi,YAN Sheng-Ye.Refined Feature Knowledge Distillation Based on Spatial-channel Collaborative Attention.COMPUTER SYSTEMS APPLICATIONS,2025,34(11):270-278
叶超越,钟良琪,闫胜业.空间-通道注意力协同的精炼特征知识蒸馏.计算机系统应用,2025,34(11):270-278
YE Chao-Yue,ZHONG Liang-Qi,YAN Sheng-Ye.Refined Feature Knowledge Distillation Based on Spatial-channel Collaborative Attention.COMPUTER SYSTEMS APPLICATIONS,2025,34(11):270-278

