Consistency of a Recurrent Language Model with Respect to Incomplete Decoding

By Sean Welleck, Ilia Kulikov, Jaedeok Kim, Richard Yuanzhe Pang, Kyunghyun Cho. 2020

76 citations recorded by the source corpus.

Despite strong performance on a variety of tasks, neural sequence models trained with maximum likelihood have been shown to exhibit issues such as length bias and degenerate repetition. We study the related issue of receiving infinite-length sequences from a recurrent language model when using common decoding algorithms. To analyze this issue, we first define inconsistency of a decoding algorithm, meaning that the algorithm can yield an infinite-length sequence that has zero probability under the model. We prove that commonly used incomplete decoding algorithms - greedy search, beam search, top-k sampling, and nucleus sampling - are inconsistent, despite the fact that recurrent language models are trained to produce sequences of finite length. Based on these insights, we propose two remedies which address inconsistency: consistent variants of top-k and nucleus sampling, and a self-terminating recurrent language model. Empirical results show that inconsistency occurs in practice, and that the proposed methods prevent inconsistency.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Natural Language Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Yejin Choi
  2. Mohit Bansal
  3. Percy Liang
  4. Luke S. Zettlemoyer
  5. Sean Welleck
  6. Hannaneh Hajishirzi
  7. Noah A. Smith
  8. Denny Zhou

In Computer Vision and Pattern Recognition

  1. Yejin Choi
  2. Mohit Bansal
  3. Luc Van Gool
  4. Dacheng Tao
  5. Rowan Zellers
  6. Quoc V. Le
  7. Trevor Darrell
  8. Jianfeng Gao

In Natural Language Processing

  1. Yejin Choi
  2. Mohit Bansal
  3. Ronan Le Bras
  4. Chandra Bhagavatula
  5. Sean Welleck
  6. Hannaneh Hajishirzi
  7. Luke S. Zettlemoyer
  8. Noah A. Smith

Citations in the corpus