On the redundancy in the rank of neural network parameters and its controllability

Chanhee Lee, Young Bum Kim, Hyesung Ji, Yeonsoo Lee, Yuna Hur, Heuiseok Lim

Research output: Contribution to journalArticlepeer-review

Abstract

In this paper, we show that parameters of a neural network can have redundancy in their ranks, both theoretically and empirically. When viewed as a function from one space to another, neural networks can exhibit feature correlation and slower training due to this redundancy. Motivated by this, we propose a novel regularization method to reduce the redundancy in the rank of parameters. It is a combination of an objective function that makes the parameter rank-deficient and a dynamic low-rank factorization algorithm that gradually reduces the size of this parameter by fusing linearly dependent vectors together. This regularization-by-pruning approach leads to a neural network with better training dynamics and fewer trainable parameters. We also present experimental results that verify our claims. When applied to a neural network trained to classify images, this method provides statistically significant improvement in accuracy and 7.1 times speedup in terms of number of steps required for training. Furthermore, this approach has the side benefit of reducing the network size, which led to a model with 30.65% fewer trainable parameters.

Original languageEnglish
Article number725
Pages (from-to)1-15
Number of pages15
JournalApplied Sciences (Switzerland)
Volume11
Issue number2
DOIs
Publication statusPublished - 2021 Jan 2

Keywords

  • Matrix rank
  • Neural network
  • Pruning
  • Redundancy
  • Regularization

ASJC Scopus subject areas

  • Materials Science(all)
  • Instrumentation
  • Engineering(all)
  • Process Chemistry and Technology
  • Computer Science Applications
  • Fluid Flow and Transfer Processes

Fingerprint Dive into the research topics of 'On the redundancy in the rank of neural network parameters and its controllability'. Together they form a unique fingerprint.

Cite this