Model Whisper: Steering Vectors Unlock Large Language Models’ Potential in Test-Time

Authors

  • Xinyue Kang Tsinghua Shenzhen International Graduate School, Tsinghua University
  • Diwei Shi Research Institute of Tsinghua University in Shenzhen Tsinghua Shenzhen International Graduate School, Tsinghua University
  • Li Chen Tsinghua Shenzhen International Graduate School, Tsinghua University

DOI:

https://doi.org/10.1609/aaai.v40i37.40403

Abstract

It is a critical challenge to efficiently unlock the powerful reasoning potential of Large Language Models (LLMs) for specific tasks or new distributions. Existing test-time adaptation methods often require tuning model parameters, which is not only computationally expensive but also risks degrading the model's pre-existing abilities.To address this, we introduce a lightweight component, Test-Time Steering Vectors (TTSV), which is prepended to the input while keeping the LLM's parameters entirely frozen. By optimizing the TTSV on test data to minimize the model's output entropy, we steer the model towards an internal state of higher confidence, activating its inherent abilities most relevant to the current task. TTSV is both lightweight and highly efficient to optimize, making it a true plug-and-play enhancement. Extensive experiments validate our approach's effectiveness on both base models and reasoning-enhanced models. For instance, on the MATH500 task, TTSV achieves a 45.88% relative performance gain on the Qwen2.5-Math-7B model and a 16.22% relative gain on the Qwen3-4B model. Furthermore, our approach exhibits robust generalization, with its steering vectors proving highly transferable across diverse tasks.

Downloads

Published

2026-03-14

How to Cite

Kang, X., Shi, D., & Chen, L. (2026). Model Whisper: Steering Vectors Unlock Large Language Models’ Potential in Test-Time. Proceedings of the AAAI Conference on Artificial Intelligence, 40(37), 31392–31400. https://doi.org/10.1609/aaai.v40i37.40403

Issue

Section

AAAI Technical Track on Natural Language Processing II