Self-Supervised Spatiotemporal Feature Learning via Video Rotation Prediction

28 Nov 2018  ·  Longlong Jing, Xiaodong Yang, Jingen Liu, YingLi Tian ·

The success of deep neural networks generally requires a vast amount of training data to be labeled, which is expensive and unfeasible in scale, especially for video collections. To alleviate this problem, in this paper, we propose 3DRotNet: a fully self-supervised approach to learn spatiotemporal features from unlabeled videos... A set of rotations are applied to all videos, and a pretext task is defined as prediction of these rotations. When accomplishing this task, 3DRotNet is actually trained to understand the semantic concepts and motions in videos. In other words, it learns a spatiotemporal video representation, which can be transferred to improve video understanding tasks in small datasets. Our extensive experiments successfully demonstrate the effectiveness of the proposed framework on action recognition, leading to significant improvements over the state-of-the-art self-supervised methods. With the self-supervised pre-trained 3DRotNet from large datasets, the recognition accuracy is boosted up by 20.4% on UCF101 and 16.7% on HMDB51 respectively, compared to the models trained from scratch. read more

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Self-Supervised Action Recognition HMDB51 3D RotNet (3D ResNet-18) Top-1 Accuracy 33.7 # 22
Pre-Training Dataset Kinetics400 # 1
Self-Supervised Action Recognition UCF101 3D RotNet (3D ResNet-18) 3-fold Accuracy 62.9 # 26
Pre-Training Dataset Kinetics400 # 1