TSM: Temporal Shift Module for Effificient Video Understanding

TSM: Temporal Shift Module for Effificient Video Understanding

TSM: Temporal Shift Module for Effificient Video Understanding

对于实际部署来讲,硬件高效的视频理解方法十分重要。论文提出时也有一些方法,来tradeoff建模与计算量,但是论文提出,已知的方案牺牲了低级的时序建模以换取效率,所以一些重要特征的损失在所难免。

核心亮点TSM模块

TSM模块的额外计算在概念上是没有的,只有涉及数据移动的实际延迟,但该模块有很强大的时空建模能力

时序局部平移策略:只平移一部分channels

在何处使用TSM模块?在残差分支使用,不会对2D CNN的空间特征学习能力造成损害。

为什么要平移?(卷积的替代)

一维上讲,1D卷积可以分解为平移和乘加操作,形式化地可以表示为:

因此,乘加操作就可以表示为:

二维上沿空间维度平移,我对一维的思想做了延申:

对于a12这个位置而言,平移和乘加操作的计算公式为:$a_{22}*w_{11}+a_{13}*w_{21}+a_{02}*w_{22}+a_{11}w_{12}+a_{12}{x}$,但是这个结果成十字形,与原先2D卷积相差较大。

三维上沿时间维度进行平移,如下图所示:

对于每一个插入TSM的模块,其时间感受野都增加了2,因此对其进行2D卷积时,就好像同时在时间维度进行filter size为3的卷积一样。

TSM的好处不言而喻

使用TSM模块可以把任意一个已有的2D网络模块变成伪3D模块,它能够同时持有空间和时间信息

部署来讲是硬件高效且友好的,低延迟推理,低内存消耗,只需支持2D卷积操作即可

支持在线推理,对于实时性要求高的应用很重要:

对于前一帧的特征图,我们截取它的1/8并保存到内存中,在当前帧中用缓存的1/8的特征图替换当前帧特征图的对应部分。从图中我们可以发现,TSM是一种多层次的特征融合模块。

实验结果基线对比

从图中可以看出,对于时序关系相对重要的数据集,加入TSM模块之后的精度提升可达2位数的提升。

泛用性证明

对于不同的2D骨干网络,TSM模块的加入都能提升精度,即使对于NL R-50这种已具备时序建模能力的网络来说。

state-of-arts证明

第一个区使用了late temporal fusion;第二个区域允许中级时序融合;第三个区域允许全层次的时序融合。

精度-性能对比

图中圆的面积代表参数量的大小,从图中可以看出TSM模型是精度高且效率高的。

性能对比

在线推理

由图可知,TSM模块对于低延迟的实时推理十分有用。图二是仅观察一部分的帧后的预测精度,可以看出TSM模型提早预测的能力较强。

相关推荐

快手新号养几天可以直播?快手直播一般多久起号最好?
365bet亚洲唯一官网

快手新号养几天可以直播?快手直播一般多久起号最好?

📅 07-04 👁️ 5074
数字绘画
best365网页登录不上去

数字绘画

📅 07-30 👁️ 7019
饥荒冰魔杖使用心得分享
365sport365中文版

饥荒冰魔杖使用心得分享

📅 08-10 👁️ 4718
火烈鸟的价格
best365网页登录不上去

火烈鸟的价格

📅 11-28 👁️ 9140
插画家︱芊祎:偏爱荒诞的“暗黑系”画家
365bet亚洲唯一官网

插画家︱芊祎:偏爱荒诞的“暗黑系”画家

📅 01-19 👁️ 8380
巴索戈和孙准浩,中超两名外援入选世界杯大名单
best365网页登录不上去

巴索戈和孙准浩,中超两名外援入选世界杯大名单

📅 07-03 👁️ 646