Calibration, Entropy Rates, and Memory in Language Models

By Mark Braverman, Xinyi Chen, Sham M. Kakade, Karthik Narasimhan, Cyril Zhang, Yi Zhang. 2019

50 citations recorded by the source corpus.

Building accurate language models that capture meaningful long-term dependencies is a core challenge in natural language processing. Towards this end, we present a calibration-based approach to measure long-term discrepancies between a generative sequence model and the true distribution, and use these discrepancies to improve the model. Empirically, we show that state-of-the-art language models, including LSTMs and Transformers, are \emph{miscalibrated}: the entropy rates of their generations drift dramatically upward over time. We then provide provable methods to mitigate this phenomenon. Furthermore, we show how this calibration-based approach can also be used to measure the amount of memory that language models use for prediction.

Subfields: Artificial Intelligence, Cultural Studies

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. Thibault Sellam
  2. Ankur P. Parikh
  3. Karthik Narasimhan
  4. Shunyu Yao
  5. Jiwei Li
  6. Fei Wu
  7. Mohit Bansal
  8. Theodore R. Sumers

Citations in the corpus