ASA: Training-Free Representation Engineering for Tool-Calling Agents

Youjin Wang, Run Zhou, Rong Fu, Shuaishuai Cao, Hongwei Zeng +4 more
2/4/2026
cs.SEcs.AI

No confident code implementation found for this paper.

Abstract

Adapting LLM agents to domain-specific tool calling remains notably brittle under evolving interfaces. Prompt and schema engineering is easy to deploy but often fragile under distribution shift and strict parsers, while continual parameter-efficient fine-tuning improves reliability at the cost of training, maintenance, and potential forgetting. We identify a critical Lazy Agent failure mode where tool necessity is nearly perfectly decodable from mid-layer activations, yet the model remains conservative in entering tool mode, revealing a representation-behavior gap. We propose Activation Steering Adapter (ASA), a training-free, inference-time controller that performs a single-shot mid-layer intervention and targets tool domains via a router-conditioned mixture of steering vectors with a probe-guided signed gate to amplify true intent while suppressing spurious triggers. On MTU-Bench with Qwen2.5-1.5B, ASA improves strict tool-use F1 from 0.18 to 0.50 while reducing the false positive rate from 0.15 to 0.05, using only about 20KB of portable assets and no weight updates.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{wang2026asa,
  title  = {ASA: Training-Free Representation Engineering for Tool-Calling Agents},
  author = {Youjin Wang and Run Zhou and Rong Fu and Shuaishuai Cao and Hongwei Zeng and Jiaxuan Lu and Sicheng Fan and Jiaqiao Zhao and Liangming Pan},
  year   = {2026},
  eprint = {2602.04935},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.04935v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.04935.svg)](https://codeofpaper.com/paper/2602.04935)

Discussion