mlsanityData to 14 Oct 2026

Tool Calling is Linearly Readable and Steerable in Language Models

arXiv 2605.07990 ↗First posted 8 May 2026cs.CL cs.AI cs.LGcited by 10

On the map: LLM agents and tool use → Tool learning and function calling

Abstract (arXiv)

Language-model agents can take real actions by calling tools, so choosing the wrong tool can cause errors that are difficult to undo. Most evaluations only observe the tool choice after the model generates a call. We read this choice from the model before generation, and we steer it. For each tool, we average the model's hidden states from a few example requests to obtain a tool vector. Comparing a new request with these tool vectors predicts which tool it needs. The difference between two tool vectors gives a steering direction that can move the model toward a chosen tool without retraining. Across eight instruction-tuned models from 4B to 27B parameters, steering changes the generated call to a chosen target tool in 56-78% of held-out tool pairs, depending on the model. We also observe steering on real APIs from τ-bench and ToolBench, though less reliably. At the final layer, steering could work simply by raising the score of the target tool name. To test this, we compare the steering direction with a direction that only raises that score, layer by layer. In the middle layers, on the three models we examine in depth, the steering direction switches more calls than the score-raising direction. So the tool vectors capture part of the tool choice before the final layer. The same tool vectors also help identify likely tool-selection errors before generation. Calls are more likely to be wrong when a request is similarly close to two tool vectors. Across four models, this signal achieves an AUROC of 0.61-0.78 and outperforms a first-token confidence baseline on three of them. Together, these results suggest that the model's hidden state provides a way to read, steer, and check tool choice before a call is made.

Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Placed in Tool learning and function calling, one of 16 directions in LLM agents and tool use. Builds on 25 of its 37 references that are in the map. First posted on arXiv 8 May 2026.

Builds on

25 of 37 references.

Closest ancestors · 6

Same direction, newest first

  1. 2026ASA: Backbone-Training-Free Representation Engineering for Tool-Calling Agents4 Feb
  2. 2023ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIs31 Jul
  3. 2023Gorilla: Large Language Model Connected with Massive APIs24 May
  4. 2023ToolkenGPT: Augmenting Frozen Language Models with Massive Tools via Tool Embeddings19 May
  5. 2023API-Bank: A Comprehensive Benchmark for Tool-Augmented LLMs14 Apr
  6. 2023Toolformer: Language Models Can Teach Themselves to Use Tools9 Feb

Other references · 19

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  3. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
  4. 2024τ-bench: A Benchmark for Tool-Agent-User Interaction in Real-World Domainscited by 1,402
  5. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  6. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  7. 2024Refusal in Language Models Is Mediated by a Single Directioncited by 1,032
  8. 2023Steering Language Models With Activation Engineeringcited by 922
  9. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
  10. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
  11. 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 528
  12. 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 394
7 more
  1. 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 319
  2. 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 300
  3. 2023An Overview of Catastrophic AI Riskscited by 222
  4. 2024Not All Language Model Features Are One-Dimensionally Linearcited by 212
  5. 2020Causal Mediation Analysis for Interpreting Neural NLP: The Case of Gender Biascited by 191
  6. 2023Linear Representations of Sentiment in Large Language Modelscited by 163
  7. 2026Internal Representations as Indicators of Hallucinations in Agent Tool Selectioncited by 19

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 10.

10 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Loud Failures, Quiet Failures: Fault Detection and Recovery in Tool-Using Language Model Agents
  2. 2026ToolRobustBench: Stage-Wise Perturbation Evaluation and Failure Diagnosis for Tool-Calling Agents
  3. 2026Feedback That Backfires: Why Small Language Model Agents Repeat the Call They Just Watched Fail
  4. 2026Calibration is the Bottleneck: An Action-Class Diagnostic of Multi-Turn Tool-Calling
  5. 2026Benchmarking the Benchmarks: A Validity Audit of Tool-Calling Evaluation
  6. 2026ToolFailBench: Diagnosing Tool-Use Failures in LLM Agents
  7. 2026The Calls are Coming from Inside the Model: Investigating Probe-based Detection of Tool-Calling Errors in LLMs
  8. 2025Tool Preferences in Agentic LLMs are Unreliable