16
1

NN-gram Is Back: Residual Learning of Neural Text Generation with nn-gram Language Model

Abstract

NN-gram language models (LM) have been largely superseded by neural LMs as the latter exhibits better performance. However, we find that nn-gram models can achieve satisfactory performance on a large proportion of testing cases, indicating they have already captured abundant knowledge of the language with relatively low computational cost. With this observation, we propose to learn a neural LM that fits the residual between an nn-gram LM and the real-data distribution. The combination of nn-gram and neural LMs not only allows the neural part to focus on the deeper understanding of language but also provides a flexible way to customize an LM by switching the underlying nn-gram model without changing the neural model. Experimental results on three typical language tasks (i.e., language modeling, machine translation, and summarization) demonstrate that our approach attains additional performance gains over popular standalone neural models consistently. We also show that our approach allows for effective domain adaptation by simply switching to a domain-specific nn-gram model, without any extra training. Our code is released at https://github.com/ghrua/NgramRes.

View on arXiv
Comments on this paper