English

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