arrow
Return

Compressing 3DCNNs based on tensor train decomposition

delete2020-11-01
delete22
delete
OA
AI
D
Dingheng Wang
G
Guangshe Zhao
Guoqi Li cover
Guoqi Li (Guoqi Li) *
邓
邓磊 (Lei Deng)
Y
Yang Wu
DOI:10.1016/j.neunet.2020.07.028delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Three-dimensional convolutional neural networks (3DCNNs) have been applied in many tasks, e.g., video and 3D point cloud recognition. However, due to the higher dimension of convolutional kernels, the space complexity of 3DCNNs is generally larger than that of traditional two-dimensional convolutional neural networks (2DCNNs). To miniaturize 3DCNNs for the deployment in confining environments such as embedded devices, neural network compression is a promising approach. In this work, we adopt the tensor train (TT) decomposition, a straightforward and simple in situ training compression method, to shrink the 3DCNN models. Through proposing tensorizing 3D convolutional kernels in TT format, we investigate how to select appropriate TT ranks for achieving higher compression ratio. We have also discussed the redundancy of 3D convolutional kernels for compression, core significance and future directions of this work, as well as the theoretical computation complexity versus practical executing time of convolution in TT. In the light of multiple contrast experiments based on VIVA challenge, UCF11, UCF101, and ModelNet40 datasets, we conclude that TT decomposition can compress 3DCNNs by around one hundred times without significant accuracy loss, which will enable its applications in extensive real world scenarios. (C) 2020 Elsevier Ltd. All rights reserved.
Keywords:
Tensor train decomposition
3DCNN
Neural network compression
Tensorizing
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Neural Networks cover
Neural Networks
IF:
6.3
Papers:
8.2K
Citations:
3.0W

Organization

X
xi'an jiaotong university
Scholars:
9.3W
Papers: 6.7W
Citations: 75
T
tsinghua university
Scholars:
11.9W
Papers: 10.0W
Citations: 137
U
University of California Santa Barbara
Scholars:
1.2W
Papers: 9.6K
Citations: 3.6W
University of California System cover
University of California System
Scholars:
37.7W
Papers: 33.8W
Citations: 6.6K
researcher View more organizations
Cited Papers

Cited Papers

Efficient Tensor Completion for Color Image and Video Recovery: Low-Rank Tensor Train
err2017-05-01
err339
errOAAI
errBengua, Johann A.; Phien, Ho N.; Hoang Duong Tuan; Do, Minh N.
errShare
errSave
Dense Trajectories and Motion Boundary Descriptors for Action Recognition
err2013-03-06
err1.3K
errOAAI
errWang, Heng; Klaeser, Alexander; Schmid, Cordelia; Liu, Cheng-Lin
errShare
errSave
errShare
errSave
A Bag of Expression framework for improved human action recognition
err2018-02-01
err38
errOAAI
errNazir, Saima; Yousaf, Muhammad Haroon; Nebel, Jean-Christophe; Velastin, Sergio A.
errShare
errSave
researcher View more