AKT通过精炼特征蒸馏提升低比特量化模型性能,突破边缘计算瓶颈。
Advanced Knowledge Transfer: Refined Feature Distillation for Zero-Shot Quantization in Edge Computing
- 利用空间与通道注意力优化特征蒸馏,精准传递关键信息
- 在CIFAR-10上3/5比特下达最优精度,显著优于现有方法
- 解决低比特模型梯度爆炸问题,适合资源受限场景部署
我们提出AKT(Advanced Knowledge Transfer),一种新方法以增强零样本量化(ZSQ)中低比特量化(Q)模型的训练能力。现有研究主要依赖全精度(FP)模型生成高质量数据,但在低比特量化时因信息容量受限导致学习能力下降。为此,我们提出替代数据生成的有效训练策略:分析表明,精炼特征蒸馏过程中的特征图是知识迁移的关键。基于此,AKT高效地将核心信息从FP模型传递至Q模型。它是首个在ZSQ中同时利用空间与通道注意力进行特征蒸馏的方法。该方法解决了低比特量化模型的根本梯度爆炸问题。在CIFAR-10和CIFAR-100上的实验验证了其有效性,显著提升了现有生成模型性能。尤其在CIFAR-10的3、5比特场景下达到当前最优结果。代码已公开于https://github.com/Inpyo-Hong/AKT-Advanced-knowledge-Transfer。
原文摘要 · Abstract (English)
We introduce AKT (Advanced Knowledge Transfer), a novel method to enhance the training ability of low-bit quantized (Q) models in the field of zero-shot quantization (ZSQ). Existing research in ZSQ has focused on generating high-quality data from full-precision (FP) models. However, these approaches struggle with reduced learning ability in low-bit quantization due to its limited information capacity. To overcome this limitation, we propose effective training strategy compared to data generation. Particularly, we analyzed that refining feature maps in the feature distillation process is an effective way to transfer knowledge to the Q model. Based on this analysis, AKT efficiently transfer core information from the FP model to the Q model. AKT is the first approach to utilize both spatial and channel attention information in feature distillation in ZSQ. Our method addresses the fundamental gradient exploding problem in low-bit Q models. Experiments on CIFAR-10 and CIFAR-100 datasets demonstrated the effectiveness of the AKT. Our method led to significant performance enhancement in existing generative models. Notably, AKT achieved significant accuracy improvements in low-bit Q models, achieving state-of-the-art in the 3,5bit scenarios on CIFAR-10. The code is available at https://github.com/Inpyo-Hong/AKT-Advanced-knowledge-Transfer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。