Lightweight Adaptive Mixture of Neural and N-gram Language Models
Computation and Language
2018-10-29 v2
Abstract
It is often the case that the best performing language model is an ensemble of a neural language model with n-grams. In this work, we propose a method to improve how these two models are combined. By using a small network which predicts the mixture weight between the two models, we adapt their relative importance at each time step. Because the gating network is small, it trains quickly on small amounts of held out data, and does not add overhead at scoring time. Our experiments carried out on the One Billion Word benchmark show a significant improvement over the state of the art ensemble without retraining of the basic modules.
Keywords
Cite
@article{arxiv.1804.07705,
title = {Lightweight Adaptive Mixture of Neural and N-gram Language Models},
author = {Anton Bakhtin and Arthur Szlam and Marc'Aurelio Ranzato and Edouard Grave},
journal= {arXiv preprint arXiv:1804.07705},
year = {2018}
}