用注意力图流让图像模型高效处理视频动作识别。
AM Flow: Adapters for Temporal Processing in Action Recognition
- 通过注意力图流分离空间与时间处理,识别运动相关像素。
- 在多个数据集上实现顶尖性能,训练更快、所需轮次更少。
- 适合想用图像模型快速部署视频动作识别的开发者。
深度学习模型,尤其是图像模型,近年来展现出更强的泛化性和鲁棒性。本文提出将此类进展应用于视频分类任务。现有视频基础模型需大量预训练和长时间训练。为此,我们引入“注意力图流(AM Flow)”,用于识别输入视频帧中与运动相关的像素。根据相机运动情况,提出两种计算AM Flow的方法。该方法可分离空间与时间处理,相比联合时空处理(如传统视频模型)效果更优。利用参数高效迁移学习中的适配器技术,将AM Flow融入预训练图像模型,避免全量微调。我们进一步扩展适配器为“时序处理适配器”,在适配器中引入时序处理单元。实验表明,该方法实现更快收敛,显著减少训练轮次。同时,使图像模型在Kinetics-400、Something-Something v2和Toyota Smarthome等数据集上达到或接近当前最佳性能,大幅降低训练时间和预训练复杂度。
原文摘要 · Abstract (English)
Deep learning models, in particular \textit{image} models, have recently gained generalisability and robustness. %are becoming more general and robust by the day. In this work, we propose to exploit such advances in the realm of \textit{video} classification. Video foundation models suffer from the requirement of extensive pretraining and a large training time. Towards mitigating such limitations, we propose "\textit{Attention Map (AM) Flow}" for image models, a method for identifying pixels relevant to motion in each input video frame. In this context, we propose two methods to compute AM flow, depending on camera motion. AM flow allows the separation of spatial and temporal processing, while providing improved results over combined spatio-temporal processing (as in video models). Adapters, one of the popular techniques in parameter efficient transfer learning, facilitate the incorporation of AM flow into pretrained image models, mitigating the need for full-finetuning. We extend adapters to "\textit{temporal processing adapters}" by incorporating a temporal processing unit into the adapters. Our work achieves faster convergence, therefore reducing the number of epochs needed for training. Moreover, we endow an image model with the ability to achieve state-of-the-art results on popular action recognition datasets. This reduces training time and simplifies pretraining. We present experiments on Kinetics-400, Something-Something v2, and Toyota Smarthome datasets, showcasing state-of-the-art or comparable results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。