Compression of Recurrent Neural Networks for Efficient Language Modeling

  • 2019-02-06 19:49:22
  • Artem M. Grachev, Dmitry I. Ignatov, Andrey V. Savchenko
  • 13

Abstract

Recurrent neural networks have proved to be an effective method forstatistical language modeling. However, in practice their memory and run-timecomplexity are usually too large to be implemented in real-time offline mobileapplications. In this paper we consider several compression techniques forrecurrent neural networks including Long-Short Term Memory models. We makeparticular attention to the high-dimensional output problem caused by the verylarge vocabulary size. We focus on effective compression methods in the contextof their exploitation on devices: pruning, quantization, and matrixdecomposition approaches (low-rank factorization and tensor traindecomposition, in particular). For each model we investigate the trade-offbetween its size, suitability for fast inference and perplexity. We propose ageneral pipeline for applying the most suitable methods to compress recurrentneural networks for language modeling. It has been shown in the experimentalstudy with the Penn Treebank (PTB) dataset that the most efficient results interms of speed and compression-perplexity balance are obtained by matrixdecomposition techniques.

 

Quick Read (beta)

loading the full paper ...