论文

论文提出基于标签引导的知识蒸馏方法优化3D-CNN在动作识别中的性能

Label-Guided Knowledge Distillation for 3D-CNNs in Action Recognition

精选理由

这篇论文分享了一种新的模型压缩技术,能提升3D-CNN在视频动作识别上的效果,感兴趣的可以看看。

本文提出Label-Guided Knowledge Distillation(LGKD)方法,通过样本级和类别级蒸馏,利用标签信息指导学生模型学习教师模型特征。在UCF101和HMDB51数据集上,该方法提升了3D-CNN在动作识别任务中的表现。

原文 · arXiv cs.AI

Label-Guided Knowledge Distillation for 3D-CNNs in Action Recognition

As a key model compression technique, knowledge distillation aims to transfer knowledge from a high-capacity teacher model to a lightweight student model for enhancing the latter's performance. In this work, we reviewed the feature knowledge distillation for 3D-CNNs and observed that most feature distillation methods in video analysis are simple adaptations of those used in image analysis, often neglecting the differences of video features in the temporal dimension. To address this issue, we proposed Label-Guided Knowledge Distillation (LGKD) to guide the distillation of student model features using ground truth labels. Our method entails two components: sample-wise distillation and class-wise distillation, enabling the student model to learn feature representation of the teacher model at two levels. Sample-wise distillation utilizes label information and the teacher's probability distribution to guide the learning of features that significantly impact temporal accuracy while mitigating noise. Meanwhile, class-wise feature distillation employs a prototype network to further capture the relational knowledge among samples within the same category, enhancing the student's ability to learn higher-dimensional semantic information and improving model generalization. To demonstrate the effectiveness and superiority of our method, we conducted comprehensive experiments on two benchmark action recognition datasets, UCF101 and HMDB51, achieving competitive results.