mlsanityData to 7 Oct 2026

Watch the Model Think: On-Policy Extraction of Activation Steering Vectors

arXiv 2602.14143 ↗First posted 15 Feb 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

When a model solves a problem on one attempt and fails it on the next, what separates the two is rarely the final answer token; it is the trajectory that reached it. Contrastive activation steering leaves that signal unused: CAA, SADI, RepE and ITI build their direction from experimenter-supplied text, recorded while the model reads rather than reasons. That choice also caps what the vector can express, since polarity must be written into the text, and a task judged only by outcome offers nothing to write it with. ROAST makes the trajectory itself the contrast: sample rollouts, let an outcome verifier split them into successes and failures, and contrast the reasoning that worked against the reasoning that did not. A matched teacher-forced control---rollouts, labels, answer text and pair counts held fixed, the trajectory alone stripped---points to the trajectory as what matters: on GSM8K at 0.6B the pairs alone buy +0.12 points while restoring the trajectories buys +6.05, the larger and only seed-robust step. Replacing the trajectory with an equal-length neutral prefix or another question's reasoning falls below no intervention. The two corpora are also far apart geometrically, a median 70+ degrees apart at both Qwen3 scales probed, beyond what a split-half null explains. Reading from rollouts calls for two corrections---keeping the full difference vector rather than Top-10% masking, and giving each question one vote rather than one per pair---and only grouped aggregation beats the unsteered baseline under 20% verifier noise. On parser-free benchmarks (GSM8K, MATH500, IFEval), ROAST is best in all six cells over two models, by up to +9.7, at +6.4% wall-clock and no added context; it also leads on six parser-scored benchmarks across three models. Across nine models (0.6B--122B, four families), ROAST improves on the unsteered model at every scale. Code: https://github.com/TomySu404/ORBIT

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 25 of its 31 references that are in the map. First posted on arXiv 15 Feb 2026.

Builds on

25 of 31 references.

References · 25

  1. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  2. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  3. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  4. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  5. 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,919
  6. 2020Learning to summarize from human feedbackcited by 2,562
  7. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  8. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
  9. 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
  10. 2024ChatGLM: A Family of Large Language Models from GLM-130B to GLM-4 All Toolscited by 1,096
  11. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
  12. 2024Refusal in Language Models Is Mediated by a Single Directioncited by 960
13 more
  1. 2018XNLI: Evaluating Cross-lingual Sentence Representationscited by 913
  2. 2022STaR: Bootstrapping Reasoning With Reasoningcited by 895
  3. 2023Steering Language Models With Activation Engineeringcited by 824
  4. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
  5. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
  6. 2015Sequence Level Training with Recurrent Neural Networkscited by 568
  7. 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 417
  8. 2024Massive Activations in Large Language Modelscited by 277
  9. 2022Extracting Latent Steering Vectors from Pretrained Language Modelscited by 217
  10. 2024Improving Instruction-Following in Language Models through Activation Steeringcited by 149
  11. 2024Semantics-Adaptive Activation Intervention for LLMs via Dynamic Steering Vectorscited by 61
  12. 2023Truth Forest: Toward Multi-Scale Truthfulness in Large Language Models through Intervention without Tuningcited by 45
  13. 2025Steering Language Models with Weight Arithmeticcited by 19

Built on by

Not cited yet.