Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

CT-Net: Channel Tensorization Network for Video Classification

About

3D convolution is powerful for video classification but often computationally expensive, recent studies mainly focus on decomposing it on spatial-temporal and/or channel dimensions. Unfortunately, most approaches fail to achieve a preferable balance between convolutional efficiency and feature-interaction sufficiency. For this reason, we propose a concise and novel Channel Tensorization Network (CT-Net), by treating the channel dimension of input feature as a multiplication of K sub-dimensions. On one hand, it naturally factorizes convolution in a multiple dimension way, leading to a light computation burden. On the other hand, it can effectively enhance feature interaction from different channels, and progressively enlarge the 3D receptive field of such interaction to boost classification accuracy. Furthermore, we equip our CT-Module with a Tensor Excitation (TE) mechanism. It can learn to exploit spatial, temporal and channel attention in a high-dimensional manner, to improve the cooperative power of all the feature dimensions in our CT-Module. Finally, we flexibly adapt ResNet as our CT-Net. Extensive experiments are conducted on several challenging video benchmarks, e.g., Kinetics-400, Something-Something V1 and V2. Our CT-Net outperforms a number of recent SOTA approaches, in terms of accuracy and/or efficiency. The codes and models will be available on https://github.com/Andy1621/CT-Net.

Kunchang Li, Xianhang Li, Yali Wang, Jun Wang, Yu Qiao• 2021

Related benchmarks

TaskDatasetResultRank
Action RecognitionSomething-Something v2 (val)
Top-1 Accuracy64.5
535
Action RecognitionUCF101
Accuracy96.2
365
Action RecognitionSomething-Something v2
Top-1 Accuracy64.5
341
Action RecognitionSomething-Something v2 (test)
Top-1 Acc67.8
333
Video ClassificationKinetics 400 (val)
Top-1 Acc79.8
204
Action RecognitionSomething-something v1 (test)
Top-1 Accuracy56.6
189
Action RecognitionSomething-Something v2 (test val)
Top-1 Accuracy64.5
187
Video ClassificationSomething-Something v2 (test)
Top-1 Acc0.678
169
Action RecognitionSomething-Something V1
Top-1 Acc52.5
162
Video ClassificationKinetics-400
Top-1 Acc79.8
131
Showing 10 of 16 rows

Other info

Code

Follow for update