about
What's the Magic Word? A Control Theory of LLM Prompting (arxiv.org)
1 point by mnk47 on Oct 14, 2024 | hide | past | pdf | discuss on HN

In plain words: Control theory is used to ask which outputs a short prompt can force a model to produce. With prompts of ten tokens or fewer, the correct next token was reachable at least 97% of the time, and the rarest tokens could become most likely.

Abstract

Prompt engineering is crucial for deploying LLMs but is poorly understood mathematically. We formalize LLM systems as a class of discrete stochastic dynamical systems to explore prompt engineering through the lens of control theory. We offer a mathematical analysis of the limitations on the controllability of self-attention as a function of the singular values of the parameter matrices. We present complementary empirical results on the controllability of a panel of LLMs, including Falcon-7b, Llama-7b, and Falcon-40b. Given initial state $\mathbf x_0$ from Wikitext and prompts of length $k \leq 10$ tokens, we find that the "correct" next token is reachable at least 97% of the time, and that the top 75 most likely next tokens are reachable at least 85% of the time. Intriguingly, short prompt sequences can dramatically alter the likelihood of specific outputs, even making the least likely tokens become the most likely ones. This control-theoretic analysis of LLMs demonstrates the significant and poorly understood role of input sequences in steering output probabilities, offering a foundational perspective for enhancing language model system capabilities.

Aman Bhargava, Cameron Witkowski, Shi-Zhuo Looi, Matt Thomson
arXiv:2310.04444 · cs.CL, cs.AI, cs.LG, cs.NE · submitted Oct 2, 2023 · updated Jul 3, 2024
abstract · pdf · html · 28 pages, 10 figures

add comment on HN
Also discussed: Jun 2024 (1 point, 0 comments)