Compressing Fully Connected Layers using Kronecker Tensor Decomposition

被引:0
|
作者
Chen, Shaowu [1 ]
Sun, Weize [1 ]
Huang, Lei [1 ]
Yang, Xin [1 ]
Huang, Junhao [1 ]
机构
[1] Shenzhen Univ, Guangdong Lab Artificial Intelligence & Cyber Eco, Shenzhen, Peoples R China
来源
PROCEEDINGS OF 2019 IEEE 7TH INTERNATIONAL CONFERENCE ON COMPUTER SCIENCE AND NETWORK TECHNOLOGY (ICCSNT 2019) | 2019年
关键词
deep neural network; fully connected layers; kronecker tensor decomposition;
D O I
10.1109/iccsnt47585.2019.8962432
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
In recent years, deep neural networks have made great achievements in many fields such as data science and image processing. Models of deep neural networks require pretty large memory storage resources with high computational complexity. In particular, the great number of parameters of fully connected layers prevents the application of deep neural networks in mobile devices. It is shown redundancy and low rank structure inside weight matrices of fully connected layers, thus methods of tensor decomposition can be applied to reduce the number of parameters. In this paper, a novel method for decomposition, Kronecker Tensor Decomposition with single pattern and multiple patterns is used to decompose the weight matrices of fully connected layers by using a series of sub-tensors to represent the original tensor with less amount of parameters. Experimental results showed that the proposed scheme can achieve a high compression ratio without significant accuracy loss.
引用
收藏
页码:308 / 312
页数:5
相关论文
共 7 条
  • [1] Compressing Deep Models using Multi Tensor Train Decomposition
    Yang, Xin
    Sun, Weize
    Huang, Lei
    Chen, Shaowu
    ICCAIS 2019: THE 8TH INTERNATIONAL CONFERENCE ON CONTROL, AUTOMATION AND INFORMATION SCIENCES, 2019,
  • [2] Deep fully-connected tensor network decomposition for multi-dimensional signal recovery
    Su, Ruoyang
    Zhao, Xi-Le
    Wu, Wei-Hao
    Liu, Sheng
    He, Junhua
    SIGNAL PROCESSING, 2025, 233
  • [3] In Defense of Fully Connected Layers in Visual Representation Transfer
    Zhang, Chen-Lin
    Luo, Jian-Hao
    Wei, Xiu-Shen
    Wu, Jianxin
    ADVANCES IN MULTIMEDIA INFORMATION PROCESSING - PCM 2017, PT II, 2018, 10736 : 807 - 817
  • [4] AutoFCL: automatically tuning fully connected layers for handling small dataset
    S. H. Shabbeer Basha
    Sravan Kumar Vinakota
    Shiv Ram Dubey
    Viswanath Pulabaigari
    Snehasis Mukherjee
    Neural Computing and Applications, 2021, 33 : 8055 - 8065
  • [5] AutoFCL: automatically tuning fully connected layers for handling small dataset
    Basha, S. H. Shabbeer
    Vinakota, Sravan Kumar
    Dubey, Shiv Ram
    Pulabaigari, Viswanath
    Mukherjee, Snehasis
    NEURAL COMPUTING & APPLICATIONS, 2021, 33 (13): : 8055 - 8065
  • [6] Overfitting remedy by sparsifying regularization on fully-connected layers of CNNs
    Xu, Qi
    Zhang, Ming
    Gu, Zonghua
    Pan, Gang
    NEUROCOMPUTING, 2019, 328 : 69 - 74
  • [7] Impact of fully connected layers on performance of convolutional neural networks for image classification
    Basha, S. H. Shabbeer
    Dubey, Shiv Ram
    Pulabaigari, Viswanath
    Mukherjee, Snehasis
    NEUROCOMPUTING, 2020, 378 (378) : 112 - 119