HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

TSM: Temporal Shift Module for Efficient Video Understanding

Lin Ji ; Gan Chuang ; Han Song

TSM: Temporal Shift Module for Efficient Video Understanding

Abstract

The explosive growth in video streaming gives rise to challenges onperforming video understanding at high accuracy and low computation cost.Conventional 2D CNNs are computationally cheap but cannot capture temporalrelationships; 3D CNN based methods can achieve good performance but arecomputationally intensive, making it expensive to deploy. In this paper, wepropose a generic and effective Temporal Shift Module (TSM) that enjoys bothhigh efficiency and high performance. Specifically, it can achieve theperformance of 3D CNN but maintain 2D CNN's complexity. TSM shifts part of thechannels along the temporal dimension; thus facilitate information exchangedamong neighboring frames. It can be inserted into 2D CNNs to achieve temporalmodeling at zero computation and zero parameters. We also extended TSM toonline setting, which enables real-time low-latency online video recognitionand video object detection. TSM is accurate and efficient: it ranks the firstplace on the Something-Something leaderboard upon publication; on Jetson Nanoand Galaxy Note8, it achieves a low latency of 13ms and 35ms for online videorecognition. The code is available at:https://github.com/mit-han-lab/temporal-shift-module.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
3d-action-recognition-on-assembly101TSM
Actions Top-1: 35.27
Object Top-1: 47.45
Verbs Top-1: 58.27
action-classification-on-kinetics-400TSM
Acc@1: 74.7
action-recognition-in-videos-on-somethingTSM (RGB + Flow)
Top-1 Accuracy: 66.6
Top-5 Accuracy: 91.3
action-recognition-in-videos-on-something-1TSM (RGB + Flow)
Top 1 Accuracy: 50.7
action-recognition-in-videos-on-something-1TSMEn
Top 1 Accuracy: 49.7
Top 5 Accuracy: 78.5
action-recognition-in-videos-on-something-1TSM
Top 1 Accuracy: 47.2
Top 5 Accuracy: 77.1
video-object-detection-on-imagenet-vidOnline TSM
MAP : 76.3

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp