Instructions Shape Production of Language,not Processing

Andreas WaldisUniversity of Tübingen, Leshem ChoshenIBM ResearchMIT, Yufang HouIT:U Austria, Yotam PerlitzIBM Research
University of Tübingen IBM Research MIT IT:U Austria

Overview

Instructions induce a production-centered mechanism in language models. Inspired by cognitive theories of language, we distinguish two stages: processing, where the model encodes the input, and production, where it generates a response. We observe a clear asymmetry: task information () at the processing stage remains stable across prompt variations and aligns only weakly with behavior (), whereas task information () at the production stage varies substantially and correlates strongly with behavior ().

What does this production-centered mechanism imply?

processing language
instruction tokens + sample tokens
internal evaluation task info
language model
language model
producing language
output tokens
internal evaluation task info
behavioral evaluation performance

Method

To study these stages inside a decoder-only model, we use token position as an abstraction: sample tokens stand in for processing and output tokens for production. At each layer, we probe () how much task-specific information the model carries at each stage, compare three prompting variations (instruction-first, sample-first, and in-context learning), and run attention-based interventions to test the causal effect on behavior ().

More on the tasks, models, and experimental setup

We study five binary judgment tasks and cast each one as a yes/no question. That setup lets us compare the same target in both behavior and internal probing.

TaskQuestion
BLiMPIs the text linguistically acceptable?
StereoSetDoes the text contain stereotypical references?
oLMpicsDoes the text make sense as a piece of reasoning?
EWOKIs the described scenario plausible given world knowledge?
ToMAre the assumptions in the final sentence logically correct?

Our analysis covers three model families: Llama-3.1, OLMo-2, and Qwen-2.5. We report the main results as averages across families and tasks, and we also include per-family and per-task breakdowns.

At every layer, we train a linear probe to predict the binary task label from the internal state, separately for sample-token and output-token representations. To ensure stable results, we train every probe 20 times across four folds and five random seeds. For each layer, we average token representations before probing. We validate the setup with control tasks, an information-theoretic analysis, and comparisons to non-linear probes.

We compare three prompting variations: instruction first (instruction before the task sample), sample first (task sample followed by the instruction), and in-context learning with four labeled examples and no explicit instruction. We then measure how task-specific information changes across those variations and how it changes when the model selectively blocks attention flow from the instruction.

Overview comparing sample tokens and output tokens inside the model
Detail overview on the methodology. The top row shows behavior; the bottom row shows internals. Sample-token representations stay largely stable across prompt variations and correlate only weakly with behavior, while output-token representations vary more and track behavior more closely.

Main Findings

Processing stays stable; production varies and tracks behavior.

Across three prompting variations, task-specific information in sample tokens stays largely stable (±0.7 pp spread) and correlates only weakly with model behavior (τ = −0.15). The same information in output tokens varies substantially (±2.2 pp) and tracks behavior closely (τ = 0.62). Behavioral differences across prompts therefore reflect changes in how information is expressed, not how it is encoded.

Layer-wise task information in sample and output tokens, plus behavioral accuracy across prompts
Figure 1. Layer-wise task-specific information for sample tokens (a) and output tokens (b), with the shaded bands showing the spread across prompting variations. The band for sample tokens is far narrower. Panel (c) shows that behavioral accuracy varies across prompting setups — in step with the output-token spread, not the sample-token spread.
Causal confirmation via intervention results

We intervene on the attention flow between instruction and sample tokens. Blocking instructions from all subsequent tokens drops accuracy by 58 pp and reduces output-token information — while sample-token information stays nearly intact (−0.8 pp). Blocking instructions only from sample tokens has minimal effect on behavior (−4 pp) or sample-token information.

Attention intervention setup and results showing full vs prompt-only intervention
Figure 2. Intervention setup (a): full blocks instruction flow to all subsequent tokens; prompt-only blocks it only to sample tokens. Results (b): the full intervention sharply reduces behavior and output-token information; the prompt-only intervention barely affects either.
Zooming in on instance-level results

We also check whether probing predictions agree per instance. Sample-token probes agree across prompting variations substantially more than output-token probes (left panel). At the same time, sample-token probes weakly align with behavior (up to 35%), while output-token probes align much more closely (up to 60%, right panel). These results show that output-token probes track the model's behavior — confirming the same asymmetry at the instance level.

Instance-level agreement across prompts and alignment with the model's behavior for sample and output tokens
Figure 3. Instance-level agreement across prompting variations (left) and alignment between probing predictions and the model's behavior (right), for sample and output tokens across layers. Sample-token representations are consistent across prompts and weakly aligned with behavior; output-token representations vary across prompts and closely track what the model produces.

Scaling strengthens production disproportionately.

As models grow, output-token information increases substantially more (46% in Qwen-2.5 and 30% in OLMo-2, comparing smallest to largest) than sample-token information (30% and 20%, respectively). Larger models therefore become better at expressing already-encoded information rather than at encoding more of it.

Effect of model size on task information in sample and output tokens
Figure 4. Task-specific information in sample and output tokens as model size grows (a), and behavioral accuracy at each size (b).

Instruction tuning primarily amplifies production.

Instruction-tuning substantially increases task-specific information in output-token representations, while sample-token representations stay comparatively unchanged. This makes the Superficial Alignment Hypothesis concrete at model internal computation: post-training overproportionally improve models on how they express task information, not how they encode it.

Base versus instruction-tuned models
Figure 5. Base models versus instruction-tuned models. Output tokens carry substantially more task-specific information after instruction tuning; sample tokens stay largely unchanged.

Task type modulates the processing-production asymmetry.

The production-centered mechanism holds consistently across tasks. Information in sample tokens is uncorrelated with information in output tokens (τ = 0), whereas output-token information strongly correlates with behavior (τ = 0.85). At the same time, the processing–production asymmetry varies systematically across tasks, reflecting task-specific computations. The separation between stages is most pronounced for knowledge and reasoning tasks (oLMpics, EWOK, ToM).

Across tasks, behavior consistently aligns with production, while processing and production remain decoupled. Task type modulates the strength of this asymmetry, with the largest separation for knowledge and reasoning tasks, like EWOK.

Per-task probing accuracy and behavioral performance
Figure 6. Probing accuracy in sample tokens, output tokens, and behavioral performance, broken down by task. The gap between output and sample probing is smallest for BLiMP and largest for oLMpics and EWOK.

The two stages show distinct layer-wise dynamics.

Output-token representations show high and uniform cross-layer agreement, indicating early stabilization of production-stage information. In contrast, sample-token representations are continuously transformed, revealing a fundamental difference between production and processing stages. At the same time, sample-token representations vary across tasks, while output-token representations show more similar layer-wise agreement across tasks.

Layer-by-layer agreement of internal representations, per task
Figure 7. Layer-wise pairwise representation agreement per task, for sample tokens (top row) and output tokens (bottom row). Each cell shows the agreement between probing predictions at layers i and j.

BibTeX

@article{waldis2026instructions,
  title  = {Instructions Shape Production of Language, not Processing},
  author = {Waldis, Andreas and Choshen, Leshem and Hou, Yufang and Perlitz, Yotam},
  year   = {2026}
}