Abstract
Today’s probabilistic language generators fall short when it comes to producing coherent and fluent text despite the fact that the underlying models perform well under standard metrics (e.g., perplexity). This discrepancy has puzzled the language generation community for the last few years. In this work, we posit that the abstraction of natural language generation as a discrete stochastic process—which allows for an information-theoretic analysis—can provide new insights into the behavior of probabilistic language generators, for example, why high-probability texts can be dull or repetitive. Humans use language as a means of communicating information, aiming to do so in a simultaneously efficient and error-minimizing manner; in fact, psycholinguistics research suggests humans choose each word in a string with this subconscious goal in mind. We formally define the set of strings that meet this criterion: Those for which each word has an information content close to the expected information content, namely, the conditional entropy of our model. We then propose a simple and efficient procedure for enforcing this criterion when generating from probabilistic models, which we call locally typical sampling. Automatic and human evaluations show that, in comparison to nucleus and top-k sampling, locally typical sampling offers competitive performance (in both abstractive summarization and story generation) in terms of quality while consistently reducing degenerate repetitions.
1 Introduction
Modern probabilistic models have repeatedly demonstrated their prowess at modeling natural language, placing high probability on held-out corpora from many different domains (Brown et al., 2020; Hoffmann et al., 2022; Chowdhery et al., 2022). Yet when used as text generators, their performance is far from perfect. One of the largest determinants of the generated text’s quality is the choice of decoding strategy—that is, the decision rule used to extract strings from a model. Perhaps surprisingly, for many language generation tasks, decoding strategies that aim to find the highest-probability strings produce text that is undesirable (Holtzman et al., 2020; See et al., 2019; Eikema and Aziz, 2020; Zhang et al., 2021; DeLucia et al., 2021). For instance, Stahlberg and Byrne (2019) report that in their neural machine translation experiments, the highest- probability string is usually the empty string. On the other hand, stochastic strategies, which take random samples from the model, often lead to text with better qualitative properties (Fan et al., 2018; Holtzman et al., 2020; Basu et al., 2021). However, stochastic strategies still have a host of other problems, while not entirely dispensing with those seen in maximization-based approaches.1
At first glance, it is unintuitive that high- probability strings are often neither desirable nor human-like. Due to this pathology, a number of studies have concluded that there must be faults in the training objective or architecture of the probabilistic models behind language generators (Welleck et al., 2020; Guan et al., 2020; Li et al., 2020, inter alia). Yet, this conclusion is at odds with these models’ performance in terms of other metrics. The fact that modern models can place high probability on held-out text suggests that they provide good estimates (in at least some aspects) of the probability distribution underlying human language. We posit that looking at language generation through an information-theoretic lens may shed light on this paradox.
Communication via natural language can intuitively be cast in information-theoretic terms. Indeed, there is a long history of studying language through the lens of information theory (Shannon, 1948, 1951; Hale, 2001; Piantadosi et al., 2011; Pimentel et al., 2020, inter alia). In this paradigm, linguistic strings are messages used to convey information, and their information content can be quantified as a function of their probability of being uttered—often driven by context. Assuming that humans use language in order to transmit information in an efficient yet robust manner (Zaslavsky et al., 2018; Gibson et al., 2019), the subset of strings typically used by humans should encode information at some (perhaps near-optimal) rate.2 In fact, prior works studying the uniform information density hypothesis (Levy and Jaeger, 2007; Mahowald et al., 2013) empirically observed this property in humans’ use of natural language.
These insights lead us to re-think what it means to be a probabilistic language generator. First, we contend that language generators, in some cases, can be thought of as discrete stochastic processes. This, in turn, allows us to cleanly define typicality (and the typical set) for these processes. We argue, however, that due to discrepancies between the model behind these generators and the true distribution over natural language strings, directly sampling from the typical set is not a good idea. Indeed, for language generators that do not use an end-of-string (eos) state, this is exactly what is done by ancestral sampling—a decoding strategy not known for providing high-quality text. Inspired by research on human sentence processing, we then define the more restrictive notion of local typicality, and argue that if we want text generated from a model to be “human-like,” we should perhaps enforce this information-theoretic criterion in generations ourselves. To this end, we develop a new algorithm, which we call locally typical sampling. Concretely, we hypothesize that for text to be perceived as natural, each word should have an information content close to its expected information content given prior context. When sampling from probabilistic language generators, we should limit our options to strings that adhere to this property. In experiments on abstractive summarization and story generation, we observe that, compared to nucleus and top-k sampling: (i) locally typical sampling reduces the number of degenerate repetitions, giving a rep value (Welleck et al., 2020) on par with human text, and (ii) text generated using typical sampling is generally closer in quality to that of human text.3
2 Two Views of Language Modeling
In this work, we discuss language models4 in an information-theoretic light. Our first step towards this goal is to re-frame their presentation. Concretely, we put forth that there are actually two lenses through which we can view language modeling productively. Under the traditional lens, we can think of a language model as a distribution over full strings: A language model constitutes the distribution of a single string-valued random variable. Under an alternative lens, we can think of a language model as a discrete stochastic process: a collection of indexed random variables. We compare and contrast these views formally, and then show how to use the language process view to derive a new sampling algorithm in §5.
2.1 A Single String-Valued Random Variable
We codify the traditional view of language modeling in the following definition. Let be an alphabet—a non-empty, finite set.
Alanguage modelpis a probability distribution over all strings.5Under this view, we can think of a language model as describing a single-valued random variable.
2.2 A Discrete Stochastic Process
Interestingly, the factorization in Eq. (1) suggests that we might view language models, not as a single string-valued random variable, but rather as a collection of random variables , namely, as a discrete stochastic process.8 Under this view, we arrive at the following definition of what we term a language process, to distinguish it from the definition of a language model given above.
Alanguage processoveris a discrete stochastic processwhere eachYt is -valued. The process is described by a distributionp, and we denote its conditional distribution asp(Yt = yt∣Y <t =y <t) fort > 0. In slight abuse of notation but out of convention, we takeYtfort ≤ 0 to bebos, i.e., conditioningpon justbossignifies the initial distribution of the process.
Definition 2.2 is very generic. In words, it just says that a language process is any discrete process where we sample a new word9 given the previously sampled words. The first question that naturally comes to mind is when the definitions of a language model and a language process coincide. As it turns out, there is a simple answer.
In words, tightness says that a language process must not leak probability mass to infinite strings. Because a language model must be a (valid) probability distribution, it must also be tight.
Letbe a language process over alphabetwith distributionpand let. ThenYis tight if and only ifpt = 1 for someor.
Note that pt is the probability of sampling eos at exactly step t given that the history of the string is of length (t − 1).
Case 1: Suppose pt = 1 for some . Then, Y is clearly tight as no probability mass is leaked to strings beyond length t, where .
Case 2: Now suppose pt < 1 for all t. In this case, we have that the probability of all infinite-length strings is given by . However, by a standard result (see, e.g., Knopp, 1954, Ch. 12), we have that , provided pt < 1.
Both cases together complete the proof.
We can now see that language processes are strictly more general than language models: Eq. (1) shows us that any language model can be written as a language process, but Proposition 2.4 shows the converse is not necessarily true. Indeed, Proposition 2.4 allows us to easily construct a simple language process (example given below) that cannot be converted to a language model, which motivates the formalism.
Let. Define a language process over such that eachYtis distributed according toand. Note that we keep the convention thatYt = bosfort ≤ 0, and thusp0 = 0. We have, so, by Proposition 2.4, Yis not a language model. Computing the infinite productshowsYleaks ≈ .58 to infinite strings.
Life after eos?
Proposition 2.4 further hints at the more intuitive difference between language models and language processes—what happens after eos? In the traditional definition of a language model (Definition 2.1), life ends at eos. That is, any string with symbols after eos would not be a valid sample from a language model because such strings are not in the model’s support. On the other hand, a language process offers a more chipper view: Once we hit eos, we can just generate another symbol. A language process is better thought of as an infinite babbler than a distribution over any sort of strings. At some level, this is indeed the implicit view that is adopted by some when language modeling, as many language models do not have eos in the traditional sense. For the rest of this paper we will also take this view, and consider language processes for which we can continue generating after sampling an eos symbol.
2.3 Other Useful Properties
Next, we discuss some other properties about language processes that are important for understanding the theoretical results presented in §3.
Many language processes are explicitly defined to be Markov, for example, ones based on n-gram language models. However, many language processes based on recurrent neural networks are, in principle, non-Markov. Yet despite being capable of learning non-Markov distributions, researchers have found that recurrent neural language models tend to learn Markov distributions. For instance, Khandelwal et al. (2018) show that a recurrent neural language model’s memory is empirically bounded at roughly 200 words. Thus, we can still generally assume this property when working with language processes parameterized by such models.11
While not theoretically Markovian, human language is generally considered stationary, that is, the probability distribution over the next word should not depend on absolute position, but rather the history.
A language processisergodicif its statistical properties (e.g., ensemble averages) can be deduced from a single, sufficiently long, random sample of the process.
The above definition is informal, as ergodicity is a complex property that would take time to treat rigorously (see, e.g., McMillan, 1953; Breiman, 1957). One of the important implications of ergodicity for language processes, however, is rather straightforward. If our language process is over alphabet with distribution p and is ergodic, then for every symbol and for every history , there must exist an extension y <t′ =y <t,yt,⋯ ,yt′ − 1 such that p(Yt′ = y∣Y <t′ =y <t′) > 0. In plain terms, this just says that we can always reach every word in our alphabet via some path no matter where we currently are. In our context, ergodicity also relates to the problem with eos. If we convert a language model into a language process (as discussed in §2.1) and make the eos state absorbing,12 this language process must be non-ergodic, as once it encounters eos, no other state is reachable.
2.4 Estimating a Language Model from Data
3 Information-Theoretic Properties of Language Processes
The view of language modeling as a discrete stochastic process naturally lends itself to an analysis through the lens of information theory. Indeed, much of information theory is concerned with the study of discrete stochastic processes (see, e.g., Cover and Thomas, 2012, Ch. 4). In this section, we review standard information-theoretic definitions in §3.1 and build on these to introduce our own notion of local typicality in §3.2.
3.1 Typicality
An important definition in the study of stochastic processes is entropy rate, which generalizes the notion of entropy from a random variable to a stochastic process.
The entropy rate is useful in that it tells us, in the limit, how spread out (i.e., entropic) the distribution is. Another interpretation is that it quantifies the complexity of Y. In the case of an i.i.d. process, the entropy rate and the entropy coincide, making the entropy rate a true generalization of the entropy. Using entropy rate, we can define the notion of the typical set.
In informal terms, the typical set is the set of all samples that we would expect when sampling from p. To give the reader intuition about typicality, we now turn to a classical example.13
Consider an i.i.d. stochastic processwhereYtis defined as the outcome of flipping a biased coin: we havep(heads) = .6 andp(tails) = .4. If we flip 100 coins, the most likely outcome is the sequence of 100 heads. However, this would be a surprising outcome to most people, who would intuitively expect the sequence to consist of roughly 60% heads and 40% tails. Indeed, even for relatively largeε, the sequence of 100 heads is not in thetypical set; its average symbol probability is .6 ≫ 2−H(Yt) ≈ 0.51.
The above example demonstrates that the typical set often does not contain the most likely sequence. Additionally, the typical set is interesting because, as , it contains nearly all the probability mass; we formalize this property in a proposition.
Letbe a stationary, ergodic discrete stochastic process where eachYtfollows distributionpand takes on values in a finite support. For everyε > 0, for sufficiently largeT, the following conditions hold:
- i)
- ii)
In words, as we take, the probability mass covered by the typical set is nearly 1 and the number of elements in it is nearly 2T·H(Y).
See Breiman (1957) for proof.
What’s Wrong with the Typical Set?
Let Y be a stationary, ergodic language process. By the conditions of Definition 3.2, we know that Y has a typical set. We have motivated the typical set, intuitively, as the subset of strings that are usual or typical among all strings. Under this intuition, it makes sense that—when using Y as a language generator—this is the set from which we would like to select a string. A relatively straightforward corollary of Proposition 3.4 is that ancestral sampling should pull from just this set. To see this, we can turn to (i) in Proposition 3.4: since ancestral sampling provides an i.i.d. sample from Y, the probability of getting an element not in as is (1 − ε), that is, practically never. However, there is the confound that our models are not perfect representations of the true distribution behind the “human” natural language process. Perhaps for this reason (and the reasons discussed in §4), ancestral sampling is not known to result in samples that humans judge to be high quality in the task of language generation; rather it often leads to text that humans perceive as incoherent (Holtzman et al., 2020). Furthermore, the typical set’s definition relies on Y being a stationary and ergodic language process. As we saw in §2.2, however, a language model that we convert into a language process will be non-ergodic by definition (at least if we keep eos as an absorbing state). Thus, while the typical set is a natural starting point, it does not actually get us to our end goal of defining a set of strings that humans would find typical. To remedy this problem, we introduce the new concept of local typicality.
3.2 Local Typicality
A core contribution of this work is to define a more restrictive notion of typicality—termed here local typicality—which we subsequently motivate as useful in the context of describing the set of strings humans typically produce.
In comparison to the typical set, the locally typical set further restricts the set of samples to those for which each individual symbol yt has probability near the local conditional entropy, that is, the entropy of the distribution p(·∣y <t). In general, there is no strong theoretical relationship between the typical set and the locally typical set. However, in the case of an i.i.d. stochastic process we can prove that the latter constitutes a subset of the former.
Letbe an i.i.d. discrete stochastic process, then.
Since Y is i.i.d., we have that H(Y) = H(Yt∣Y <t) = H(Yt). Let y be an element of . Then, . Thus, by the triangle inequality, , which implies , which implies .
A natural question to ask at this point is why the definition of local typicality is useful in the context of a language process. Our argument, presented in the following section, is cognitive in nature.
4 Local Typicality in Natural Language
To motivate our definition of local typicality in the context of natural language, we must first look at language through an information-theoretic lens. We will consider two distributions in this section: , the distribution that a speaker of the language is assumed to generate strings from, and our language process that approximates —albeit, perhaps not perfectly. In this setting, we view a natural language string y as a means of communicating some information, where each word yt is a symbol via which we construct our message. The information content of y is then defined as its negative log-probability under a specified distribution: . Following the chain rule of probability, this quantity can be decomposed over words, that is, the information content of a word is its negative log-probability given prior context: .
4.1 Properties of Human Communication
Given the above definitions, we can now ask a question at the heart of this work: What are the information-theoretic characteristics of natural language typically produced by humans. In other words, what do strings sampled from look like, from the perspective of , our trained language process? Research in psycholinguistics suggests that a core component of what makes text human-like is its per-unit information content.
To motivate this conclusion, we first consider a language user’s objective. When using natural language, humans aim to transmit information efficiently while also minimizing the risk of miscommunication (Zipf, 1949). In order to achieve this goal, speakers avoid producing words with either very high or very low information content (Fenk and Fenk, 1980; Aylett and Turk, 2004; Levy and Jaeger, 2007; Mahowald et al., 2013, inter alia), a behavior in line with theories of efficient and robust communication.14 Indeed, cross-linguistic research has shown that languages trade off information content and speech rate, perhaps aiming at a specific (optimal) information rate (Coupé et al., 2019; Pimentel et al., 2021). Further, not using words in a context where they have very high or low information content avoids characteristics that appear to negatively impact traditional grammaticality judgments: An ideal natural language string would not compensate for unusually near-zero probability in the first half (e.g., syntactic error) with unusually high probability in the second half (e.g., especially frequent words) (Schütze, 2016; Lau et al., 2017).
4.2 An Information-Theoretic Formalization
The definition of local typicality presented in §3.2 can be viewed as an embodiment of the characteristics of human language just described above. One logical interpretation of these behaviors is that, at every time step, natural-sounding language should have per-symbol information content close to the expected (average) per-symbol information content.15 We formalize the relationship between natural language and local typicality in the following hypothesis.
Samplesy = y0⋯yTfrom a human language process with distributiontend to belong to the process’slocally typical setfor large enoughTand someε > 0. In words, this means that we should expect every word in natural-sounding sentences to be close to theexpectedinformation content under, i.e., the conditional entropy given prior context.
We verify this relationship empirically using data from human language processes. In Figure 1, we show the distribution of the difference between the information content of yt and the expected information content of Yt, namely, , according to the model on human-generated text. The peaked nature of the distributions in Figure 1 reveals that human language indeed tends to have per-word information content quite close a specific value; the centering of these distributions around ≈ 0 suggests that this value is H(Yt∣Y <t =y <t). Notably, Meister et al. (2022) shows the same is not true for text generated by models according to a number of different popular decoding schemes, which instead produce strings with much higher probability, that is, with lower information content.
In an ideal situation, such a property of natural language would be reflected in , in which case sampling from the typical set should be sufficient to ensure human-like language. However, our models are by no means perfect. The failure to capture the property of human language expounded in Hypothesis 4.1 may come from a number of possible modeling deficiencies, for example, poor ability to capture the tails of these distributions. We hypothesize that, when using language models to generate text, enforcing this local-typicality criterion explicitly may serve as a patch for this shortcoming.
5 Sampling from a Language Process
In this section, we describe how to sample from a language process parameterized by the distribution p,16 or in more commonly used terminology, how to decode from p. There are many different algorithms one could employ to sample from p. The most intuitive strategy is ancestral sampling,17 which works as follows: Sample yt ∼ p(·∣y <t) for each history y <t successively until some chosen criterion, for example, the eos symbol is sampled or a maximum length is reached. Note that in the case of the former criterion, this procedure is equivalent to sampling entire strings according to the distribution p. Perhaps the most popular set of techniques for sampling fall under a paradigm we call truncated sampling, where the vocabulary at a time step is truncated to a core subset of words. For instance, Fan et al. (2018) propose limiting the sampling space to the top-k most likely words in each decoding step, and Holtzman et al. (2020) consider the smallest nucleus (i.e., subset) of words whose cumulative probability mass exceeds a chosen threshold η.
5.1 Shortcomings of Existing Algorithms
To motivate sampling based on the locally typical set, we must first better understand the shortcomings of current decoding strategies. While strings generated using stochastic strategies may have lower probability according to , they often outperform those decoded using maximization-based strategies in terms of qualitative metrics. A number of recent works have tried to offer explanations for this phenomenon. Some have attributed it to a diversity–quality trade-off (Zhang et al., 2021; Basu et al., 2021), while others blame shortcomings of model architectures or training strategies (Welleck et al., 2020; Li et al., 2020).
Our analysis from §4 inspires an alternative explanation, motivated by information theory and psycholinguistics, for why models that perform so well (in terms of metrics such as perplexity) can still exhibit such undesirable behavior when used to generate text. First, the connection between probability and information content may explain why high-probability text is often dull or generic (Holtzman et al., 2020; Eikema and Aziz, 2020); its low information content likely makes for boring (i.e., uninformative) text. This connection also offers a potential explanation for the rather strange behavior that, when a string has a repetitive loop, language models often assign increasingly higher probability to the repeated substring (Holtzman et al., 2020); the substring conveys less and less information after each occurrence.
A further implication of this framing is the equivalence between decoding strings from a probabilistic language generator and sampling messages from the natural language communication channel. If we wish to solely sample from the subset of messages that a human would typically construct, that is, that are human-like, then we should begin by narrowing down this subset to those messages that meet at least some of the same criteria as human-generated messages. In this work, we have identified the criterion that such messages tend to be in the locally typical set. This observation motivates a new decoding strategy in which our information-theoretic criterion is explicitly enforced, which we subsequently present.
5.2 Locally Typical Sampling
We now introduce our novel sampling algorithm, which we entitle locally typical sampling.
In words, Algorithm 5.3 limits the sampling distribution to only those words with negative log-probability within a certain absolute range from the conditional entropy (expected information content) of the model at that time step. In the spirit of nucleus sampling, this range is determined by a hyperparameter τ, the amount of probability mass from the original distribution that we wish to consider.
Interestingly, Algorithm 5.3 does not imply that high-probability words should not be chosen. Indeed, in the situation where conditional entropy is low, namely, when the model places most of the probability mass on a small subset of words, it is likely the case that only high-probability words fall into the locally typical set.
Computational Complexity.
From a practical perspective, locally typical sampling can be implemented with the same efficiency as nucleus or top-k sampling. First, we compute the conditional entropy, which is an operation. Second, we sort words by their absolute distance from , which can be done in time with standard sorting algorithms. Finally, we greedily take words from this list until their cumulative probability exceeds the threshold τ, which again takes time. Thus, creating our altered distribution has time complexity .18
Relationship to Other Decoding Strategies.
Notably, we already see motivation for this criterion in the performance of several well-known decoding strategies. For example, beam search is the predominant decoding strategy for machine translation models (Wu et al., 2016; Edunov et al., 2018; Ng et al., 2019; Meister et al., 2020b), a setting in which beam search (incidentally) often already enforces this criterion.19 Yet, when used in more open-ended tasks, where the entropy of the language model is higher, beam search can lead to low-quality text (Li et al., 2016; Holtzman et al., 2020; Welleck et al., 2020; Meister et al., 2022). Locally typical sampling is also closely related to nucleus sampling. When the probability distribution over the vocabulary has low conditional entropy, that is, when there are only a few reasonable choices for the next word according to our model, nucleus and typical will have the same truncation set. Locally typical sampling and Mirostat (Basu et al., 2021) likewise have similar decision rules for truncation. Mirostat decodes strings such that they have a perplexity (or, equivalently, a per-word information content) close to a target value. In contrast to Mirostat, however, locally typical sampling does not require a specific target information content to be defined. Rather, locally typical sampling defines this quantity as the conditional entropy, choosing it dynamically (per word) and making it less sensitive to hyperparameter choice. Finally, locally typical sampling is also related to Braverman et al. (2020) strategy, which proposes a look- ahead decoding algorithm that generates text with a similar entropy rate to that of human-generated text. Our strategy’s motivation is similar—to match the tendencies in information content exhibited by human-generated text—albeit without requiring the computational overhead of a look- ahead strategy.
6 Experiments
In this section, we explore the efficacy of our decoding strategy on two natural language generation tasks: abstractive summarization and story generation. We assess performance with respect to several other stochastic decoding strategies: nucleus sampling, top-k sampling, temperature sampling,20 beam search, and Mirostat. Our evaluation includes both automatic metrics and human ratings.
6.1 Setup
Models and Data.
We use the HuggingFace framework (Wolf et al., 2020) for reproducibility, employing their implementations of nucleus, top-k, temperature sampling, and beam search. We rely on the implementation of Mirostat provided by its authors. For story generation, we finetune the medium and large versions of GPT-2 (Radford et al., 2019) from checkpoints made available by OpenAI on the WritingPrompts dataset (Fan et al., 2018). We use the medium checkpoint finetuned on WikiText-103 (Merity et al., 2017) to produce the data used in Figure 1. For abstractive summarization, we use BART (Lewis et al., 2020) finetuned on the CNN/DailyMail dataset (Nallapati et al., 2016).21 All reported metrics are computed on the respective test sets.
Hyperparameters.
In a preliminary hyperparameter sweep using Mauve22 (Pillutla et al., 2021), we found k = {30,40}, η = {0.9,0.95} and τ = 3.0 to be the best performing hyperparameters for top-k sampling, nucleus sampling and Mirostat, respectively. For locally typical sampling, we found τ = 0.2,τ = 0.95 to provide the best results for story generation and abstractive summarization, respectively. Standard values according to the literature for other hyperparameters (i.e., for beam search and temperature sampling) were employed. We use these values in our human evaluations and in computation of automatic metrics.
Automatic Quality Metrics.
As automatic quality metrics, we evaluate the generated text’s perplexity—under both the model used to generate the text (PPL(g)) and an independent, i.e., not finetuned, LM (PPL(i)), namely GPT-2 large (Radford et al., 2019). Several prior works have shown that neither low nor high perplexity (Zhang et al., 2021; Nadeem et al., 2020; Pillutla et al., 2021) are direct indicators of text quality. Rather, human-like text often has perplexity within a certain range. Consequently, we report the difference in this metric from the reference text as well. We additionally evaluate using Mauve23 (Pillutla et al., 2021) with the reference text.
Automatic Diversity Metrics.
Human Evaluations.
We use the AmazonMechanical Turk framework to obtain human judgments of text quality from 5 different annotators on 200 examples per decoding strategy, per task. We use solely MTurk Master Workers in order to maximize the quality of our ratings. We follow DeLucia et al. (2021) in setting up our evaluations. Each Human Intelligence Task (HIT) consists of either a single prompt from which a story should be generated or a single news article to be summarized. The raters are first presented with the different rating criteria, along with descriptions of the type of text that meets these criteria at different levels of the scale. Raters are additionally provided several examples of stories/ summarizations that both meet and fail to meet the rating criteria. They are then presented with the respective prompt/news article and the corresponding stories/summaries generated by different decoders and by the reference in random order. For abstractive summarization, we ask annotators to score on fluency and relevance, while for story generation, annotators score on fluency, coherence, and interestingness, each using a scale from 1 to 5. We choose these criteria following recommendations from van der Lee et al. (2019).
For each story/summarization and each of the criteria, we take the median score across raters as the final score.24 Workers are paid $1.50 per HIT for the abstractive summarization task and $2 per HIT for the story generation task, for which entries were longer. Note that these rates translate to >$15/hour.
6.2 Results
Quantitative Performance.
Tables 1 and 2 show the results of our different evaluation metrics. Human scores are averaged across the qualitative metrics to give an aggregate score; the value in parentheses is the standard error of the estimate. We show full breakdowns of score distributions in Table 5. We see that in general, locally typical sampling performs on par with or better than other sampling techniques, producing text with human quality ratings closest to that of the reference among the stochastic decoding strategies. Interestingly, beam search still outperforms locally typical sampling in abstractive summarization, albeit by a small margin. This could perhaps be attributed to the deterministic nature of beam search, which suggests that an interesting direction for future research may be a deterministic version of locally typical sampling, for example, where the highest-probability word within the truncated set is always chosen. Importantly, all the strategies we explore are quite close to human- level performance—in some cases even surpassing human references in terms of ratings. At this level, it is perhaps only reasonable to expect that the differentiation between the top strategies is small. Accordingly, we also consider how robust locally typical sampling is to hyperparameter choice. Figure 2 shows rep measurements for different values of the hyperparameters k, η, and τ for top-k, nucleus, and locally typical sampling, respectively. Interestingly, rep appears to be far less sensitive to τ than to k and η. While many values of k and η appear to lead to degenerate repetitions in story generation, most values of τ lead to text with a rep value on par with human text, demonstrating that an advantage of our technique is its robustness to hyperparameter choice. See Figure 3 in the Appendix for a larger exploration of how other quality metrics vary as a function of τ.
Qualitative Performance.
We present some examples of text generated according to each of the decoding strategies in Tables 3 and 4. For both of the tasks, we choose the example with ID 1 in the respective test set and provide examples from each of the decoding strategies, employing the hyperparameter values that lead to the best human scores in Tables 2 and 1. For the summarization task, we see that locally typical sampling provides a comprehensive and coherent summary of the article, quite similar to that of beam search. In comparison, the text produced by temperature sampling is not necessarily coherent; text from nucleus sampling and top-k sampling misses some of the important information in the article, for example, the charges of burglary and arson. While the qualitative performance in story generation is much more subjective, locally typical sampling arguably provides the most fluent story among all the decoding strategies. Other stories lack coherence and, even within the first few sentences, we see repeated phrases and words. Together, these results suggest that locally typical sampling may indeed produce more desirable text.
Abstractive Summarization (CNN/DailyMail) . | |
---|---|
Prompt | (CNN) The attorney for a suburban New York cardiologist charged in what authorities say was a failed scheme to have another physician hurt or killed is calling the allegations against his client “completely unsubstantiated.” Appearing Saturday morning on CNN’s “New Day,” Randy Zelin defended his client, Dr. Anthony Moschetto, who faces criminal solicitation, conspiracy, burglary, arson, criminal prescription sale and weapons charges in connection to what prosecutors called a plot to take out a rival doctor on Long Island. “None of anything in this case has any evidentiary value,” Zelin told CNN’s Christi Paul... |
Reference | A lawyer for Dr. Anthony Moschetto says the charges against him are baseless. Moschetto, 54, was arrested for selling drugs and weapons, prosecutors say. Authorities allege Moschetto hired accomplices to burn down the practice of former associate. |
Beamk = 5 | Dr. Anthony Moschetto faces criminal solicitation, conspiracy, burglary, arson and weapons charges. “None of anything in this case has any evidentiary value,” his attorney says. |
Nucleusη = 0.95 | Dr. Anthony Moschetto, 54, pleaded not guilty to charges Wednesday. Two men – identified as James Chmela and James Kalamaras – were named as accomplices. |
Top-kk = 30 | Dr. Anthony Moschetto is accused of providing police with weapons and prescription drugs. Authorities say he was part of a conspiracy to harm or kill a rival doctor. His attorney calls the allegations against his client “completely unsubstantiated” |
Typicalτ = 0.95 | Dr. Anthony Moschetto is charged with crimes including arson, conspiracy, burglary, prescription sale, weapons charges. His attorney says “none of anything in this case has any evidentiary value” |
Abstractive Summarization (CNN/DailyMail) . | |
---|---|
Prompt | (CNN) The attorney for a suburban New York cardiologist charged in what authorities say was a failed scheme to have another physician hurt or killed is calling the allegations against his client “completely unsubstantiated.” Appearing Saturday morning on CNN’s “New Day,” Randy Zelin defended his client, Dr. Anthony Moschetto, who faces criminal solicitation, conspiracy, burglary, arson, criminal prescription sale and weapons charges in connection to what prosecutors called a plot to take out a rival doctor on Long Island. “None of anything in this case has any evidentiary value,” Zelin told CNN’s Christi Paul... |
Reference | A lawyer for Dr. Anthony Moschetto says the charges against him are baseless. Moschetto, 54, was arrested for selling drugs and weapons, prosecutors say. Authorities allege Moschetto hired accomplices to burn down the practice of former associate. |
Beamk = 5 | Dr. Anthony Moschetto faces criminal solicitation, conspiracy, burglary, arson and weapons charges. “None of anything in this case has any evidentiary value,” his attorney says. |
Nucleusη = 0.95 | Dr. Anthony Moschetto, 54, pleaded not guilty to charges Wednesday. Two men – identified as James Chmela and James Kalamaras – were named as accomplices. |
Top-kk = 30 | Dr. Anthony Moschetto is accused of providing police with weapons and prescription drugs. Authorities say he was part of a conspiracy to harm or kill a rival doctor. His attorney calls the allegations against his client “completely unsubstantiated” |
Typicalτ = 0.95 | Dr. Anthony Moschetto is charged with crimes including arson, conspiracy, burglary, prescription sale, weapons charges. His attorney says “none of anything in this case has any evidentiary value” |
Story Generation (WritingPrompts) . | |
---|---|
Prompt | A kid doodling in a math class accidentally creates the world’s first functional magic circle in centuries. |
Reference | It was dark and Levi was pretty sure he was lying on his back. There was firelight flickering off of what was left of a ceiling. He could hear something but it was muffled. He was almost positive it was screaming. When he tried to move he felt an excruciating pain in his left side that caused him to cry out... |
Nucleusη = 0.95 | “Why do you draw a circle?” He asked as he handed the piece of paper over to her. She rolled her eyes and looked down at the picture, “Because I like math!” “Then what?” “I like to draw the circles.” She ran a hand through her curls and looked away. “See, I have three circles.” “Hmph, I didn’t think I saw that in your class.” A younger student brought up the ceiling and murmured... |
Top-kk = 30 | “Aha. Aah! That makes three.” Professor Gosh sighed. “Oh my God, you had to have been there.” “Well, I’m not sure if that’s the right term, but this circle is going to work.” “But I just told you that it wasn’t working, didn’t you, you little sh**? I’m not kidding! That circle is gon na work!” “So what now?”... |
Tempτ = 1.0 | It was teachers duty to ensure that our kingdom is pure and pure and successful, however it does not mean we do not try to be as rebellious as any other kingdom. Most of our teachers are suitable for rulership, being knowledgeable in sadistic rec thunking of time stopping magic circle created by several apprentices... |
Mirostatτ = 3 | ready to go home, when a little kid starts to sketch a circle. “No. Just stop. It’s not working.” “But the circle turns to smoke and then to a flame. It’s not working, I know it is!” Another kid, who was also drawing a circle with his phone, starts to scream. The teacher, looking at the circle, lost his patience and started to yell... |
Typicalτ = 0.2 | As soon as he got the clock on his desk he jumped up. The noise it was making had changed his day forever. His hair, normally unkempt, now glistened and he had to have more hair because of it. “Yes!” He had declared triumphantly as he watched his father fumble around in the backpack in search of a little compass he always kept with him. He took it from the desk and quickly walked out the door to see what had happened. He wasn’t the first child in the world to draw something... |
Story Generation (WritingPrompts) . | |
---|---|
Prompt | A kid doodling in a math class accidentally creates the world’s first functional magic circle in centuries. |
Reference | It was dark and Levi was pretty sure he was lying on his back. There was firelight flickering off of what was left of a ceiling. He could hear something but it was muffled. He was almost positive it was screaming. When he tried to move he felt an excruciating pain in his left side that caused him to cry out... |
Nucleusη = 0.95 | “Why do you draw a circle?” He asked as he handed the piece of paper over to her. She rolled her eyes and looked down at the picture, “Because I like math!” “Then what?” “I like to draw the circles.” She ran a hand through her curls and looked away. “See, I have three circles.” “Hmph, I didn’t think I saw that in your class.” A younger student brought up the ceiling and murmured... |
Top-kk = 30 | “Aha. Aah! That makes three.” Professor Gosh sighed. “Oh my God, you had to have been there.” “Well, I’m not sure if that’s the right term, but this circle is going to work.” “But I just told you that it wasn’t working, didn’t you, you little sh**? I’m not kidding! That circle is gon na work!” “So what now?”... |
Tempτ = 1.0 | It was teachers duty to ensure that our kingdom is pure and pure and successful, however it does not mean we do not try to be as rebellious as any other kingdom. Most of our teachers are suitable for rulership, being knowledgeable in sadistic rec thunking of time stopping magic circle created by several apprentices... |
Mirostatτ = 3 | ready to go home, when a little kid starts to sketch a circle. “No. Just stop. It’s not working.” “But the circle turns to smoke and then to a flame. It’s not working, I know it is!” Another kid, who was also drawing a circle with his phone, starts to scream. The teacher, looking at the circle, lost his patience and started to yell... |
Typicalτ = 0.2 | As soon as he got the clock on his desk he jumped up. The noise it was making had changed his day forever. His hair, normally unkempt, now glistened and he had to have more hair because of it. “Yes!” He had declared triumphantly as he watched his father fumble around in the backpack in search of a little compass he always kept with him. He took it from the desk and quickly walked out the door to see what had happened. He wasn’t the first child in the world to draw something... |
7 Conclusion
In this work, we analyze decoding from probabilistic language generators in the information- theoretic framework. We equate a language model to a discrete stochastic process, and use the known properties of such processes to quantitatively describe the samples we should expect. Motivated by results in psycholinguistics, we hypothesize that—with the goal of communicating efficiently and robustly—humans produce text whose per- word information content is within a close range of the expected information content of a word given prior context. Current language models may fall short in capturing this property, which is a possible explanation for why the corresponding language processes often do not lead to human- like text. Yet, this observation provides a simple new criterion for decoding more human-like text from probabilistic language generators: Constraining the sampling space to words that meet this criterion. In experiments on two language generation tasks, we find that our strategy—called locally typical sampling—leads to text of comparable or better quality than other stochastic decoding strategies according to human ratings. Further, when compared to these other decoding strategies, several quantitative properties of typically- sampled text more closely align with those of human text.
Acknowledgments
We would like to thank Jason Eisner, Tim Vieira, Jennifer White, and Ari Holtzmann for early conversations about the relationship between information theory and sampling. We would also like to thank Ehud Reiter, who served as our TACL action editor, and the the anonymous reviewers for their insightful feedback during the review process. Further, we are grateful to Eleanor Chodroff, Clément Guerner, and Lucas Torroba Hennigen for their feedback on the manuscript of this work.
Ethical Concerns
In order to complete our human evaluation, we used a crowdsourcing platform. For each task, we made sure that the crowdworkers would be paid (at minimum) a wage of $15 per hour.
Another ethical consideration worth discussing concerns the use of language models for text generation. Text generated by these models may contain malicious content, either by design of the user or as a byproduct of the training data/algorithm. While we hope the results of our work will not be misused, they may nonetheless provide insights for those employing these models with ill-intent as to how machine-generated text can be made more “human-like,” and thus more convincing.
A Additional Results
Notes
While maximization-based strategies can produce text that is generic or degenerate, stochastic strategies occasionally produce nonsensical text. Both types of strategies tend to eventually fall into repetitive loops.
Information rate may be defined with respect to time (as is the case with spoken language) or with respect to a specific linguistic unit, such as a word (as is the case with text).
An implementation of typical sampling can be found in the HuggingFace Transformers library (Wolf et al., 2020).
Here we use the term language model to refer to any (valid) probability distribution over natural language strings. We subsequently specify the necessary conditions for validity. Note that this distribution may also be conditioned on an input.
The Kleene closure of a set is defined as .
Some authors erroneously omit eos from their definition. However, we require a distinguished symbol eos to be able to locally normalize the language model and make it a valid probability distribution.
This process is discrete both in time and in value.
One could just as easily define a language process over subwords, morphemes, or characters.
Also known as a Markov chain.
Note that, in principle, human language is not Markov, in so far as many linguists believe human language is capable of arbitrarily deep center-embeddings (Chomsky, 1957, 1995). Yet research suggests that humans do not make use of this property in practice (Reich, 1969; Karlsson, 2010), and so we do not consider the Markovian property of most models as a limitation to their ability to model natural language in practice.
This would be done by setting the transition probability p(Yt = eos∣Y <t =y <t) = 1 if yt − 1 = eos.
See Dieleman (2020) for further discussion of the concept of typicality in the context of generative modeling.
See Gibson et al. (2019) for an in-depth review of how efficiency has shaped the evolution of language.
The standard definition of (Shannon) entropy for a random variable X with support is equivalent to the expected information of X: .
Here we only consider locally normalized p, i.e., processes in which sampling is done on a word-by-word basis.
Another natural option would be to choose words which maximize the probability assigned by p to the resulting string, but this work focuses on stochastic strategies.
For each of the truncation sampling algorithms, the truncation set can also be identified using the selection algorithm (no sorting required) in time. We provide the analysis using sorting as that is the standard implementation.
When trained without label-smoothing, which artificially inflates conditional entropies, machine translation models tend to have quite low conditional entropies (see, e.g., Figure 3 in Meister et al., 2020a). Therefore, at each decoding step, the set of words with negative log-probability near the conditional entropy of the model are typically those with high probability—the same as those chosen by beam search.
Temperature sampling is defined as ancestral sampling after local renormalization with an annealing term τ.
As we are interested in getting as close an estimate of p as possible with our models , all fine-tuning is done without label-smoothing. Note that label-smoothing may also artificially inflate conditional entropy estimates, as it pushes the learned distribution towards the most entropic distribution: the uniform distribution (Pereyra et al., 2017).
We use the default settings given by the authors for all Mauve computations, although we employ different LMs in our parameter sweep vs. reported results (standard GPT-2 vs. GPT-2 large, respectively) to reduce bias in the final results. Notably, Mauve presents similar performances when used with these two pretrained LMs (Pimentel et al., 2022).
We use the implementation provided by the authors.
We use an attention check in each HIT. Responses where the attention check has been failed are thrown out.
References
Author notes
Action Editor: Ehud Reiter