arXiv · 2310.12688
Compression of Recurrent Neural Networks using Matrix Factorization
Abstract
Compressing neural networks is a key step when deploying models for real-time or embedded applications. Factorizing the model's matrices using low-rank approximations is a promising method for achieving compression. While it is possible to set the rank before training, this approach is neither flexible nor optimal. In this work, we propose a post-training rank-selection method called Rank-Tuning that selects a different rank for each matrix. Used in combination with training adaptations, our method achieves high compression rates with no or little performance degradation. Our numerical experiments on signal processing tasks show that we can compress recurrent neural networks up to 14x with at most 1.4% relative performance reduction.
Explore related subjects
Keep this discovery
Lucas Maison, Hélion du Mas des Bourboux, Thomas Courtat. 2023-10-19. Compression of Recurrent Neural Networks using Matrix Factorization. https://arxiv.org/abs/2310.12688
Cite the original work for its findings. Save a collection to share your selection of sources.