arXiv:2611.06837v1 Announce Type: cross
Abstract: Under stochastic decoding, a single context can elicit many different answers from a language model. Characterizing this variation typically requires many expensive autoregressive rollouts, each revealing only one realization. We ask whether the activations produced while reading the context can directly forecast the model's rollout activations. We train context--answer metamodels that map a context activation sequence to point or distribution forecasts of mean-pooled answer activations. We find that (i) the four metamodel families we study forecast mean-pooled answer activations with high fidelity, and the distribution metamodels also capture rollout variability; (ii) performance improves as the training set grows to 500,000 prompts, and the metamodels generalize zero-shot to out-of-domain datasets; (iii) their forecasts retain behavior-predictive signal; and (iv) context activations from Qwen2.5 predict mean-pooled answer activations from Qwen3.5 and Gemma 4, suggesting that activation forecasts can transfer across models. Together, these results suggest that context activation traces can support scalable forecasts of behaviors in LLM answers before the first answer token is generated.
arxiv
Metamodels Forecast LLM Answer Activations Before First Token
Researchers propose context-answer metamodels that map context activations to forecasts of mean-pooled answer activations, capturing variability across stochastic rollouts. The metamodels achieve high fidelity, scale with training data up to 500,000 prompts, and generalize zero-shot to out-of-domain datasets. Forecasts retain behavior-predictive signal and transfer across models, suggesting scalable pre-generation behavior prediction.
This introduces a novel method to predict LLM answer behavior without expensive rollouts, with cross-model transfer, offering practical efficiency gains for AI evaluation and safety.
Comments & Questions (0)
No comments yet
Be the first to comment!