arXiv AI By Pratyay Dutta, Kowshik Thopalli, Vivek Narayanaswamy

COMPASS: Finding Where Reasoning Lives in Language Models

Read the original on arXiv AI →

The paper introduces COMPASS, an inference-time steering technique that enhances large language model performance on mathematical reasoning tasks. By using the correctness of a model’s direct answer attempts as a latent signal, COMPASS identifies a small subset of attention heads whose activations can be steered to elicit reasoning. Experiments across three model families and multiple math benchmarks show that COMPASS improves GSM8K accuracy by an average of 16 percentage points and achieves near Chain‑of‑Thought performance while generating 20–70% fewer tokens.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv Computation and Language
Aug 25

Mechanistic Interpretability of Chain-of-Thought Reasoning via Sequential Activation Patching

The paper introduces a sequential activation patching framework to study how Chain-of-Thought (CoT) prompting influences large language models over multiple generated tokens. By tracking CoT-conditioned attention-head activations across token positions and aggregating them with Part-of-Speech guidance, the authors identify distributed head sets that jointly contribute to answer generation. Targeted zero-ablation experiments confirm that these heads are functionally important, affecting mechanisms such as reasoning-trajectory maintenance, answer anchoring, exemplar-target separation, and numerical generation.

By Murat Dura, Serkan \"Ozt\"urk, Selma Tekir