用积分梯度增强知识蒸馏,让小模型更懂大模型决策。
Model compression using knowledge distillation with integrated gradients
- 将积分梯度图叠加到输入图像上,帮助学生模型理解教师决策。
- 在CIFAR-10上实现92.6%准确率,压缩4.1倍,比非蒸馏模型高1.1个百分点。
- 预计算梯度图,训练时无需额外开销,适合边缘设备部署。
模型压缩对在资源受限设备上部署深度学习模型至关重要。本文提出一种新方法,将积分梯度(IG)作为数据增强策略融入知识蒸馏。训练时将IG热力图叠加于输入图像,使学生模型获得教师模型决策过程的深层洞察。在CIFAR-10上的大量实验表明,该方法在4.1倍压缩比下达到92.6%测试准确率,相比未蒸馏模型(91.5%)提升1.1个百分点(p<0.001),推理时间从140毫秒降至13毫秒。该方法预先计算IG图,将高运行成本转化为一次性预处理。实验涵盖:(1)与注意力转移对比,二者可互补;(2)蒙特卡洛模拟验证统计稳健性;(3)系统评估压缩比(2.2x–1122x)与精度权衡;(4)在与CIFAR-10类别一致的ImageNet子集上验证泛化能力。大量消融实验证实,基于IG的知识蒸馏在不同架构和压缩比下均优于传统方法,具备真实部署潜力。
原文摘要 · Abstract (English)
Model compression is critical for deploying deep learning models on resource-constrained devices. We introduce a novel method enhancing knowledge distillation with integrated gradients (IG) as a data augmentation strategy. Our approach overlays IG maps onto input images during training, providing student models with deeper insights into teacher models' decision-making processes. Extensive evaluation on CIFAR-10 demonstrates that our IG-augmented knowledge distillation achieves 92.6% testing accuracy with a 4.1x compression factor-a significant 1.1 percentage point improvement ($p<0.001$) over non-distilled models (91.5%). This compression reduces inference time from 140 ms to 13 ms. Our method precomputes IG maps before training, transforming substantial runtime costs into a one-time preprocessing step. Our comprehensive experiments include: (1) comparisons with attention transfer, revealing complementary benefits when combined with our approach; (2) Monte Carlo simulations confirming statistical robustness; (3) systematic evaluation of compression factor versus accuracy trade-offs across a wide range (2.2x-1122x); and (4) validation on an ImageNet subset aligned with CIFAR-10 classes, demonstrating generalisability beyond the initial dataset. These extensive ablation studies confirm that IG-based knowledge distillation consistently outperforms conventional approaches across varied architectures and compression ratios. Our results establish this framework as a viable compression technique for real-world deployment on edge devices while maintaining competitive accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。