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