Bayesian Sparsification of Recurrent Neural Networks
Machine Learning
2017-08-02 v1 Computation and Language
Machine Learning
Abstract
Recurrent neural networks show state-of-the-art results in many text analysis tasks but often require a lot of memory to store their weights. Recently proposed Sparse Variational Dropout eliminates the majority of the weights in a feed-forward neural network without significant loss of quality. We apply this technique to sparsify recurrent neural networks. To account for recurrent specifics we also rely on Binary Variational Dropout for RNN. We report 99.5% sparsity level on sentiment analysis task without a quality drop and up to 87% sparsity level on language modeling task with slight loss of accuracy.
Keywords
Cite
@article{arxiv.1708.00077,
title = {Bayesian Sparsification of Recurrent Neural Networks},
author = {Ekaterina Lobacheva and Nadezhda Chirkova and Dmitry Vetrov},
journal= {arXiv preprint arXiv:1708.00077},
year = {2017}
}
Comments
Published in Workshop on Learning to Generate Natural Language, ICML, 2017