arrow
返回

TT-MLP: Tensor Train Decomposition on Deep MLPs

delete2023-01-01
delete2
delete
OA
AI
J
Jiale Yan
K
Kota Ando
J
Jaehoon Yu
M
Masato Motomura *
DOI:10.1109/ACCESS.2023.3240784delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Deep multilayer perceptrons (MLPs) have achieved promising performance on computer vision tasks. Deep MLPs consist solely of fully-connected layers as the conventional MLPs do but adopt more sophisticated network architectures based on mixer layers composed of token-mixing and channel-mixing components. These architectures enable deep MLPs to have global receptive fields, but the significant increase of parameters becomes a massive burden on practical applications. To tackle this problem, we focus on using tensor-train decomposition (TTD) for compressing deep MLPs. At first, this paper analyzes deep MLPs under conventional TTD methods, especially using various designs of a macro framework and micro blocks: The former is how to concatenate mixer layers, and the latter is how to design a mixer layer. Based on the analysis, we propose a novel TTD method named Train-TTD-Train. The proposed method exerts the learning capability of channel-mixing components and improves the trade-off between accuracy and size. In the evaluation, the proposed method showed a better trade-off than conventional TTD methods on ImageNet-1K and achieved a 0.56% higher inference accuracy with a 15.44% memory reduction on Cifar-10.
Keyword:
Tensor-train decomposition
low-rank approximation
deep neural networks
deep multilayer perceptron
network parameter compression

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

I
Institute of Science Tokyo
学者数:
3.2W
论文数: 2.7W
被引数: 117
T
Tokyo Institute of Technology
学者数:
1.1W
论文数: 9.0K
被引数: 1.9W
引用论文

引用论文

SPECTRAL TENSOR-TRAIN DECOMPOSITION谱张量-训练分解
err2016-01-01
err87
errOAAI
errBigoni, Daniele; Engsig-Karup, Allan P.; Marzouk, Youssef M.
err分享
err收藏
An Improved Alpha Beta Filter Using a Deep Extreme Learning Machine
err2021-01-01
err18
errOAAI
errKhan, Junaid; Fayaz, Muhammad; Hussain, Ayyaz; Khalid, Shah; Mashwani, Wali Khan; Gwak, Jeonghwan
err分享
err收藏
Stability of adaptive behaviors in middle-school children with autism spectrum disorders
err2007-10-01
err0
PREAI
errRobin L. Gabriels; Bonnie Jean Ivers; Dina E. Hill; John A. Agnew; John McNeill
err分享
err收藏
Heuristic rank selection with progressively searching tensor ring network
err2021-03-17
err36
errOAAI
errLi, Nannan; Pan, Yu; Chen, Yaran; Ding, Zixiang; Zhao, Dongbin; Xu, Zenglin
err分享
err收藏
Serum concentrations of chlorinated dibenzo-p-dioxins and dibenzofurans among former New Zealand trichlorophenol workers
err2009-09-01
err0
PREAI
errJames J. Collins; Michael Wilken; David McBride; Noel F. Humphry; Peter Herbison; Carol J. Burns; Kenneth Bodner
err分享
err收藏
学者 查看更多内容