Ústav teorie informace a automatizace

Jste zde

Bibliografie

Conference Paper (international conference)

Stable Low-Rank Tensor Decomposition for Compression of Convolutional Neural Network

Phan A. H., Sobolev K., Sozykin K., Ermilov D., Gusak J., Tichavský Petr, Glukhov V., Oseledets I., Cichocki A.

: ECCV 2020, p. 522-539 , Eds: Vedaldi Andrea, Bischof Horst, Brox Thomas, Frahm Jan-Michael

: European Conference on Computer Vision 2020 /16./, (Glasgow, GB, 20200823)

: Convolutional neural network acceleration, Low-rank tensor decomposition, Degeneracy correction

: 10.1007/978-3-030-58526-6_31

: http://library.utia.cas.cz/separaty/2020/SI/tichavsky-0534541.pdf

(eng): Most state-of-the-art deep neural networks are overparameterized and exhibit a high computational cost. A straightforward approach to this problem is to replace convolutional kernels with its low-rank tensor approximations, whereas the Canonical Polyadic tensor Decomposition is one of the most suited models. However, fitting the convolutional tensors by numerical optimization algorithms often encounters diverging components, i.e.,extremely large rank-one tensors but canceling each other. Such degeneracy often causes the non-interpretable result and numerical instability for the neural network ne-tuning. This paper is the first study on degeneracy in the tensor decomposition of convolutional kernels. We present a novel method, which can stabilize the low-rank approximation of convolutional kernels and ensure efficient compression while preserving the high quality performance of the neural networks. We evaluate our approach on popular CNN architectures for image classification and show that our method results in much lower accuracy degradation and provides consistent performance.\n

: JD

: 20201

07.01.2019 - 08:39