arrow
返回

Nonlinear tensor train format for deep neural network compression

delete2021-12-01
delete21
PRE
AI
D
Dingheng Wang
G
Guangshe Zhao *
H
Hengnu Chen
Z
Zhexian Liu
邓
邓磊 (Lei Deng)
Guoqi Li 封面图
Guoqi Li (Guoqi Li) *
DOI:10.1016/j.neunet.2021.08.028delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Deep neural network (DNN) compression has become a hot topic in the research of deep learning since the scale of modern DNNs turns into too huge to implement on practical resource constrained platforms such as embedded devices. Among variant compression methods, tensor decomposition appears to be a relatively simple and efficient strategy owing to its solid mathematical foundations and regular data structure. Generally, tensorizing neural weights into higher-order tensors for better decomposition, and directly mapping efficient tensor structure to neural architecture with nonlinear activation functions, are the two most common ways. However, the considerable accuracy loss is still a fly in the ointment for the tensorizing way especially for convolutional neural networks (CNNs), while the number of studies in the mapping way is comparatively limited and corresponding compression ratio appears to be not considerable. Therefore, in this work, by researching multiple types of tensor decompositions, we realize that tensor train (TT), which has specific and efficient sequenced contractions, is potential to take into account both of tensorizing and mapping ways. Then we propose a novel nonlinear tensor train (NTT) format, which contains extra nonlinear activation functions embedded in sequenced contractions and convolutions on the top of the normal TT decomposition and the proposed TT format connected by convolutions, to compensate the accuracy loss that normal TT cannot give. Further than just shrinking the space complexity of original weight matrices and convolutional kernels, we prove that NTT can afford an efficient inference time as well. Extensive experiments and discussions demonstrate that the compressed DNNs in our NTT format can almost maintain the accuracy at least on MNIST, UCF11 and CIFAR-10 datasets, and the accuracy loss caused by normal TT could be compensated significantly on large-scale datasets such as ImageNet. (C) 2021 Published by Elsevier Ltd.
Keyword:
Tensor train decomposition
Nonlinear tensor train
Sequenced contractions
Sequenced convolutions
Neural network compression
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Neural Networks 封面图
Neural Networks
IF:
6.3
论文数:
8.2K
被引数:
3.0W

机构

X
xi'an jiaotong university
学者数:
9.3W
论文数: 6.7W
被引数: 75
T
tsinghua university
学者数:
11.9W
论文数: 10.0W
被引数: 137
引用论文

引用论文

Body Pose Prediction Based on Motion Sensor Data and Recurrent Neural Network
err2021-03-01
err51
PREAI
errWozniak, Marcin; Wieczorek, Michal; Silka, Jakub; Polap, Dawid
err分享
err收藏
Compressing 3DCNNs based on tensor train decomposition基于张量训练分解的3dcnn压缩算法
err2020-11-01
err22
errOAAI
errWang, Dingheng; Zhao, Guangshe; Li, Guoqi; Deng, Lei; Wu, Yang
err分享
err收藏
Functional Diversity of Microbial Communities in Soils in the Vicinity of Wanda Glacier, Antarctic Peninsula
err2012-01-01
err0
errOAAI
errIgor Stelmach Pessi; Susana de Oliveira Elias; Felipe Lorenz Simões; Jefferson Cardia Simões; Alexandre José Macedo
err分享
err收藏
学者 查看更多内容