English

DeCoVec: Building Decoding Space based Task Vector for Large Language Models via In-Context Learning

Computation and Language 2026-04-14 v1

Abstract

Task vectors, representing directions in model or activation spaces that encode task-specific behaviors, have emerged as a promising tool for steering large language models (LLMs). However, existing approaches typically require fine-tuning or invasive manipulation of internal states, limiting their flexibility and scalability. We propose \textsc{DeCoVec} (Decoding Space based Task Vector), a training-free and non-invasive framework that constructs task vectors directly in the \textit{decoding space} by leveraging in-context learning (ICL). Specifically, \textsc{DeCoVec} captures the task essence as the difference between the output logit distributions of few-shot and zero-shot prompts, then steers generation by injecting this vector into the decoding process. Experiments across seven LLMs (0.5B--9B) on TruthfulQA, Math-500, and AQUA-RAT show that \textsc{DeCoVec} consistently outperforms standard few-shot baselines, with gains up to +5.50 average accuracy. Further analysis demonstrates that \textsc{DeCoVec} effectively suppresses generation degeneration and logical flaws while exhibiting strong robustness to demonstration ordering, all without incurring additional input token costs. Our method offers a training-free and non-invasive solution for LLM steering without requiring weight updates or auxiliary models.

Keywords

Cite

@article{arxiv.2604.11129,
  title  = {DeCoVec: Building Decoding Space based Task Vector for Large Language Models via In-Context Learning},
  author = {Feiyang Li and Yile Wang},
  journal= {arXiv preprint arXiv:2604.11129},
  year   = {2026}
}

Comments

Accepted to ACL 2026 Findings

R2 v1 2026-07-01T12:05:49.599Z