针对动作识别的蒸馏方法,能自适应区分难易样本并优化训练效率。
Sample-level Adaptive Knowledge Distillation for Action Recognition
- 按样本难易度动态调整蒸馏强度,提升知识迁移效果。
- 在两个视频数据集上性能超越传统方法,且训练成本更低。
- 适合资源受限设备部署,尤其对复杂视频任务有帮助。
知识蒸馏通过从预训练大模型(教师)中学习小模型(学生)来压缩神经网络。尽管图像领域已有诸多研究,但视频分析因需训练更大模型,难以在资源受限设备上部署。传统方法忽视两个关键问题:一是教师与学生能力差距导致部分难迁移样本的知识无法正确传递,甚至损害学生性能;二是随着训练进行,难易样本的可学性会动态变化。为此,本文提出样本级自适应知识蒸馏(SAKD)框架,包含样本蒸馏难度评估模块和样本自适应蒸馏模块。前者通过随机丢弃或打乱帧来增强样本学习难度,从而更准确判断其蒸馏难度;后者在样本层面自适应调节蒸馏比例,使易迁移样本以蒸馏损失为主,难迁移样本以原始损失为主。更重要的是,仅选取低蒸馏难度且高多样性的样本用于学生训练,显著降低计算开销。在两个视频基准和一个图像基准上的实验表明,该方法在性能与效率间取得良好平衡。
原文摘要 · Abstract (English)
Knowledge Distillation (KD) compresses neural networks by learning a small network (student) via transferring knowledge from a pre-trained large network (teacher). Many endeavours have been devoted to the image domain, while few works focus on video analysis which desires training much larger model making it be hardly deployed in resource-limited devices. However, traditional methods neglect two important problems, i.e., 1) Since the capacity gap between the teacher and the student exists, some knowledge w.r.t. difficult-to-transfer samples cannot be correctly transferred, or even badly affects the final performance of student, and 2) As training progresses, difficult-to-transfer samples may become easier to learn, and vice versa. To alleviate the two problems, we propose a Sample-level Adaptive Knowledge Distillation (SAKD) framework for action recognition. In particular, it mainly consists of the sample distillation difficulty evaluation module and the sample adaptive distillation module. The former applies the temporal interruption to frames, i.e., randomly dropout or shuffle the frames during training, which increases the learning difficulty of samples during distillation, so as to better discriminate their distillation difficulty. The latter module adaptively adjusts distillation ratio at sample level, such that KD loss dominates the training with easy-to-transfer samples while vanilla loss dominates that with difficult-to-transfer samples. More importantly, we only select those samples with both low distillation difficulty and high diversity to train the student model for reducing computational cost. Experimental results on two video benchmarks and one image benchmark demonstrate the superiority of the proposed method by striking a good balance between performance and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。