中文

VidTr:无卷积的视频Transformer

计算机视觉与模式识别 2021-10-19 v2

摘要

我们引入了带有可分离注意力(separable-attention)的Video Transformer(VidTr)用于视频分类。与常用的3D网络相比,VidTr能够通过堆叠注意力聚合时空信息,并以更高效率提供更好性能。我们首先介绍朴素视频Transformer,表明Transformer模块能够从原始像素执行时空建模,但内存消耗大。随后我们提出VidTr,在保持相同性能的同时将内存成本降低3.3×\times。为进一步优化模型,我们提出基于标准差的topK注意力池化(pooltopK_stdpool_{topK\_std}),其通过沿时间维度丢弃非信息性特征来减少计算量。VidTr在五个常用数据集上以更低计算需求取得最先进性能,显示了我们设计的高效性与有效性。最后,误差分析与可视化表明VidTr尤其擅长预测需要长期时序推理的动作。

关键词

引用

@article{arxiv.2104.11746,
  title  = {VidTr: Video Transformer Without Convolutions},
  author = {Yanyi Zhang and Xinyu Li and Chunhui Liu and Bing Shuai and Yi Zhu and Biagio Brattoli and Hao Chen and Ivan Marsic and Joseph Tighe},
  journal= {arXiv preprint arXiv:2104.11746},
  year   = {2021}
}

备注

ICCV 2021 Accepted