One Task Vector is not Enough: A Large-Scale Study for In-Context Learning

Pavel Tikhonov, Ivan Oseledets, Elena Tutubalina


Abstract
In-context learning (ICL) enables Large Language Models (LLMs) to adapt to new tasks using few examples, with task vectors, defined as specific hidden state activations hypothesized to encode task information. Existing studies are limited by small-scale benchmarks, restricting comprehensive analysis. We introduce QᴜɪᴛᴇAFᴇᴡ, a novel dataset of 3,096 diverse few-shot tasks, each with 30 input-output pairs derived from the Alpaca dataset. Experiments with Llama-3-8B on QᴜɪᴛᴇAFᴇᴡ reveal: (1) task vector performance peaks at an intermediate layer (e.g., 15th), (2) effectiveness varies significantly by task type, and (3) complex tasks rely on multiple, subtask-specific vectors rather than a single vector, suggesting distributed task knowledge representation.
Anthology ID:
2026.acl-srw.127
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Santosh T.Y.S.S., Juan Diego Rodriguez, Ona de Gibert
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
1451–1463
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.acl-srw.127/
DOI:
Bibkey:
Cite (ACL):
Pavel Tikhonov, Ivan Oseledets, and Elena Tutubalina. 2026. One Task Vector is not Enough: A Large-Scale Study for In-Context Learning. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026), pages 1451–1463, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
One Task Vector is not Enough: A Large-Scale Study for In-Context Learning (Tikhonov et al., ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.acl-srw.127.pdf