arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.02468cs.CV

DL-KDD:用于暗光下动作识别的双光知识蒸馏

DL-KDD: Dual-Light Knowledge Distillation for Action Recognition in the Dark

National Yang Ming Chiao Tung University(国立阳明交通大学)

Chi-Jui Chang, Oscar Tai-Yuan Chen, Vincent S. Tseng

AI总结:

针对暗光视频动作识别中增强方法丢失信息与双流方法计算成本高的问题,提出DL-KDD师生框架,通过知识蒸馏让模型在训练时结合原始与增强视频,推理时仅需原始视频,在多个数据集上性能优于现有方法。

摘要:

暗光视频中的人体动作识别是计算机视觉领域的一项挑战性任务。近期研究侧重于应用暗光增强方法来提升视频的可见度。然而,这种视频处理会导致原始(未增强)视频中关键信息的丢失。相反,传统的双流方法能够从原始视频和处理后的视频中学习信息,但在视频分类任务的推理阶段会导致计算成本显著增加。为了解决这些挑战,我们提出了一种新颖的师生视频分类框架,命名为用于暗光下动作识别的双光知识蒸馏(DL-KDD)。该框架使模型能够从原始视频和增强视频中学习,而不会在推理期间引入额外的计算成本。具体而言,DL-KDD在训练期间利用知识蒸馏策略。教师模型使用增强视频进行训练,学生模型则使用原始视频和教师模型生成的软目标进行训练。这种师生框架允许学生模型在推理期间仅使用原始输入视频预测动作。在我们的实验中,所提出的DL-KDD框架在ARID、ARID V1.5和Dark-48数据集上优于现有最先进的方法。我们在每个数据集上均取得最佳性能,并在Dark-48上实现高达4.18%的提升,且仅使用原始视频输入,从而避免在推理时使用双流框架或增强模块。我们进一步在消融实验中验证了蒸馏策略的有效性。结果突显了我们的知识蒸馏框架在暗光人体动作识别中的优势。

英文摘要:

Human action recognition in dark videos is a challenging task for computer vision. Recent research focuses on applying dark enhancement methods to improve the visibility of the video. However, such video processing results in the loss of critical information in the original (un-enhanced) video. Conversely, traditional two-stream methods are capable of learning information from both original and processed videos, but it can lead to a significant increase in the computational cost during the inference phase in the task of video classification. To address these challenges, we propose a novel teacher-student video classification framework, named Dual-Light KnowleDge Distillation for Action Recognition in the Dark (DL-KDD). This framework enables the model to learn from both original and enhanced video without introducing additional computational cost during inference. Specifically, DL-KDD utilizes the strategy of knowledge distillation during training. The teacher model is trained with enhanced video, and the student model is trained with both the original video and the soft target generated by the teacher model. This teacher-student framework allows the student model to predict action using only the original input video during inference. In our experiments, the proposed DL-KDD framework outperforms state-of-the-art methods on the ARID, ARID V1.5, and Dark-48 datasets. We achieve the best performance on each dataset and up to a 4.18% improvement on Dark-48, using only original video inputs, thus avoiding the use of two-stream framework or enhancement modules for inference. We further validate the effectiveness of the distillation strategy in ablative experiments. The results highlight the advantages of our knowledge distillation framework in dark human action recognition.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑