On the Redundancy in the Rank of Neural Network Parameters and Its Controllability

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.

Automated summary

The paper examines the redundancy in the ranks of neural network parameters and introduces a regularization method to reduce it, resulting in significant improvements in training dynamics and fewer trainable parameters. Experimental results confirm the efficacy of the proposed approach, showcasing a substantial increase in accuracy and speedup in training steps, as well as a reduction in network size by 30.65%.