We introduce Video Transformer (VidTr) with separable-attention for video classification. Comparing with commonly used 3D networks, VidTr is able to aggregate spatio-temporal information via stacked attentions and provide better performance with higher efficiency. We first introduce the vanilla video transformer and show that transformer module is able to perform spatio-temporal modeling from raw pixels, but with heavy memory usage. We then present VidTr which reduces the memory cost by 3.3× while keeping the same performance. To further optimize the model, we propose the standard deviation based topK pooling for attention (pool), which reduces the computation by dropping non-informative features along temporal dimension. VidTr achieves state-of-the-art performance on five commonly used datasets with lower computational requirement, showing both the efficiency and effectiveness of our design. Finally, error analysis and visualization show that VidTr is especially good at predicting actions that require long-term temporal reasoning.

Download full-text PDF

Source
http://www.ncbi.nlm.nih.gov/pmc/articles/PMC9093781PMC
http://dx.doi.org/10.1109/iccv48922.2021.01332DOI Listing

Publication Analysis

Top Keywords

video transformer
12
vidtr
6
vidtr video
4
transformer
4
transformer convolutions
4
convolutions introduce
4
introduce video
4
transformer vidtr
4
vidtr separable-attention
4
separable-attention video
4

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!