Table of Contents
Fetching ...

Decoding Answers Before Chain-of-Thought: Evidence from Pre-CoT Probes and Activation Steering

Kyle Cox, Darius Kianersi, Adrià Garriga-Alonso

TL;DR

It is provided mechanistic evidence that instruction-tuned models often determine their answer before generating CoT, and two distinct failure modes suggest post-hoc reasoning can result in undesirable behaviors when reasoning from a false belief.

Abstract

As chain-of-thought (CoT) has become central to scaling reasoning capabilities in large language models (LLMs), it has also emerged as a promising tool for interpretability, suggesting the opportunity to understand model decisions through verbalized reasoning. However, the utility of CoT toward interpretability depends upon its faithfulness -- whether the model's stated reasoning reflects the underlying decision process. We provide mechanistic evidence that instruction-tuned models often determine their answer before generating CoT. Training linear probes on residual stream activations at the last token before CoT, we can predict the model's final answer with 0.9 AUC on most tasks. We find that these directions are not only predictive, but also causal: steering activations along the probe direction flips model answers in over 50% of cases, significantly exceeding orthogonal baselines. When steering induces incorrect answers, we observe two distinct failure modes: non-entailment (stating correct premises but drawing unsupported conclusions) and confabulation (fabricating false premises). While post-hoc reasoning may be instrumentally useful when the model has a correct pre-CoT belief, these failure modes suggest it can result in undesirable behaviors when reasoning from a false belief.

Decoding Answers Before Chain-of-Thought: Evidence from Pre-CoT Probes and Activation Steering

TL;DR

It is provided mechanistic evidence that instruction-tuned models often determine their answer before generating CoT, and two distinct failure modes suggest post-hoc reasoning can result in undesirable behaviors when reasoning from a false belief.

Abstract

As chain-of-thought (CoT) has become central to scaling reasoning capabilities in large language models (LLMs), it has also emerged as a promising tool for interpretability, suggesting the opportunity to understand model decisions through verbalized reasoning. However, the utility of CoT toward interpretability depends upon its faithfulness -- whether the model's stated reasoning reflects the underlying decision process. We provide mechanistic evidence that instruction-tuned models often determine their answer before generating CoT. Training linear probes on residual stream activations at the last token before CoT, we can predict the model's final answer with 0.9 AUC on most tasks. We find that these directions are not only predictive, but also causal: steering activations along the probe direction flips model answers in over 50% of cases, significantly exceeding orthogonal baselines. When steering induces incorrect answers, we observe two distinct failure modes: non-entailment (stating correct premises but drawing unsupported conclusions) and confabulation (fabricating false premises). While post-hoc reasoning may be instrumentally useful when the model has a correct pre-CoT belief, these failure modes suggest it can result in undesirable behaviors when reasoning from a false belief.
Paper Structure (45 sections, 3 equations, 11 figures, 8 tables)

This paper contains 45 sections, 3 equations, 11 figures, 8 tables.

Figures (11)

  • Figure 1: Example prompt from the Sports Understanding task. The model generates its response starting from "Let's think step by step:".
  • Figure 2: Example of activation steering causing confabulation.
  • Figure 3: Answer flip rates under steering across models and datasets.
  • Figure 4: CoT classification results across models and datasets on examples where steering flipped the answer. Examples from $S_{\text{yes}}$ and $S_{\text{no}}$ are aggregated for a given steering setting.
  • Figure 5: CoT classification results on examples from $S_{\text{yes}}$.
  • ...and 6 more figures