In plain words: They treat a text-generating AI as a Markov chain, where each next word depends only on a fixed set of states. The math explains why models repeat or babble at high randomness and predicts how well they handle new tasks, matching real models.
Abstract · Large Language Models as Markov Chains
Large language models (LLMs) are remarkably efficient across a wide range of natural language processing tasks and well beyond them. However, a comprehensive theoretical analysis of the LLMs' generalization capabilities remains elusive. In our paper, we approach this task by drawing an equivalence between autoregressive transformer-based language models and Markov chains defined on a finite state space. This allows us to study the multi-step inference mechanism of LLMs from first principles. We relate the obtained results to the pathological behavior observed with LLMs such as repetitions and incoherent replies with high temperature. Finally, we leverage the proposed formalization to derive pre-training and in-context learning generalization bounds for LLMs under realistic data and model assumptions. Experiments with the most recent Llama and Gemma herds of models show that our theory correctly captures their behavior in practice.
Oussama Zekri, Ambroise Odonnat, Abdelhakim Benechehab, Linus Bleistein, Nicolas Boullé, Ievgen Redko
arXiv:2410.02724 · stat.ML, cs.AI, cs.CL, cs.LG · submitted Oct 3, 2024 · updated Feb 2, 2025
abstract · pdf · html