How Many Different Outputs Can a Transformer Generate?
Abstract
We study how we can leverage only a handful of characteristics of a transformer's architecture to closely predict the number of different sequences it can output, both qualitatively and quantitatively. We provide an upper bound depending on the length of the prompt, which we show empirically to be tight up to a factor less than 10, across architectures and model sizes. Our analysis also provides a theoretical explanation for previously observed empirical failures of transformers on simple sequence tasks—such as copying and cramming. Formally, we prove that (i) the maximal length of accessible sequences (those that the transformer can output for some prompt) grows linearly with the prompt length, (ii) beyond a critical threshold, the proportion of accessible sequences decays exponentially with sequence length, and (iii) the linear coefficient relating prompt length to accessible sequence length admits a theoretical upper bound. Notably, these results hold even with unbounded context and computation time.
Lay Summary
Large language models can suddenly fail on long copying tasks, even when they work well on shorter ones. We show that these failures are not random: they can be predicted from the design of the model itself. In fact, LLMs can generate only a finite number of different sentences, meaning that most possible sentences are structurally inaccessible to them. We also derive simple formulas that closely predict model performance on expressivity tasks, showing that some limits of AI systems are mathematically predictable.