arXiv · 1511.06732
Sequence Level Training with Recurrent Neural Networks
Abstract
Many natural language processing applications use language models to generate text. These models are typically trained to predict the next word in a sequence, given the previous words and some context such as an image. However, at test time the model is expected to generate the entire sequence from scratch. This discrepancy makes generation brittle, as errors may accumulate along the way. We address this issue by proposing a novel sequence level training algorithm that directly optimizes the metric used at test time, such as BLEU or ROUGE. On three different tasks, our approach outperforms several strong baselines for greedy generation. The method is also competitive when these baselines employ beam search, while being several times faster.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Marc'Aurelio Ranzato, Sumit Chopra, Michael Auli, Wojciech Zaremba. 2016-05-06. Sequence Level Training with Recurrent Neural Networks. https://arxiv.org/abs/1511.06732
Cite the original work for its findings. Save a collection to share your selection of sources.