In plain words: Common words get their own hidden weights while rare words share one set, avoiding separate weights for every word. At the same hidden size, it predicted text better than a standard recurrent network, using only a small fraction of the full per-word version's parameters.
Abstract · Restricted Recurrent Neural Tensor Networks: Exploiting Word Frequency and Compositionality
Increasing the capacity of recurrent neural networks (RNN) usually involves augmenting the size of the hidden layer, with significant increase of computational cost. Recurrent neural tensor networks (RNTN) increase capacity using distinct hidden layer weights for each word, but with greater costs in memory usage. In this paper, we introduce restricted recurrent neural tensor networks (r-RNTN) which reserve distinct hidden layer weights for frequent vocabulary words while sharing a single set of weights for infrequent words. Perplexity evaluations show that for fixed hidden layer sizes, r-RNTNs improve language model performance over RNNs using only a small fraction of the parameters of unrestricted RNTNs. These results hold for r-RNTNs using Gated Recurrent Units and Long Short-Term Memory.
Alexandre Salle, Aline Villavicencio
arXiv:1704.00774 · cs.CL · submitted Apr 3, 2017 · updated May 11, 2018
abstract · pdf · html