Back to AI News
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.

AI Brief

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.

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.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!