On the Redundancy in the Rank of Neural Network Parameters and Its Controllability
- Authors
- Lee, Chanhee; Kim, Young-Bum; Ji, Hyesung; Lee, Yeonsoo; Hur, Yuna; Lim, Heuiseok
- Issue Date
- 1월-2021
- Publisher
- MDPI
- Keywords
- matrix rank; neural network; pruning; redundancy; regularization
- Citation
- APPLIED SCIENCES-BASEL, v.11, no.2, pp.1 - 15
- Indexed
- SCIE
SCOPUS
- Journal Title
- APPLIED SCIENCES-BASEL
- Volume
- 11
- Number
- 2
- Start Page
- 1
- End Page
- 15
- URI
- https://scholar.korea.ac.kr/handle/2021.sw.korea/50623
- DOI
- 10.3390/app11020725
- ISSN
- 2076-3417
- 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.
- Files in This Item
- There are no files associated with this item.
- Appears in
Collections - Graduate School > Department of Computer Science and Engineering > 1. Journal Articles
Items in ScholarWorks are protected by copyright, with all rights reserved, unless otherwise indicated.