Prefix-Tuning for Arbitrary Output Sequences on Pretrained Transformers

arXiv:2610.24123 · cs.CL, cs.AI, cs.LG · Submitted 2026-10-16 · Read on arXiv

cs.CL, cs.AI, cs.LG

Submitted: 2026-10-16

Updated: 2026-10-16

Comments: Accepted at NeurIPS 2026. Code: https://github.com/wzii/prefix_tuning. Online demo: https://wzii.github.io/prefix_tuning/

License: http://creativecommons.org/licenses/by-nc-sa/4.0/

The gist: Prefix-tuning is a parameter-efficient fine-tuning method for large language models (LLMs) that steers model outputs by prepending task-specific prefix representations to the input sequence.

Terminology

Abstract

Prefix-tuning is a parameter-efficient fine-tuning method for large language models (LLMs) that steers model outputs by prepending task-specific prefix representations to the input sequence. Despite its practical success, the theoretical understanding of the expressiveness of prefixes remains incomplete. In particular, the following question remains open: Given a pretrained autoregressive Transformer, for any input-output sequence pair (,), does there always exist a prefix S such that prepending S to induces the model to generate the target output ? In this work, we prove that under the assumption that is in the range of the pretrained Transformer, such a prefix S exists and can be explicitly computed by solving multiple linear systems. We further show that this assumption is necessary, in the sense that such a prefix may not exist when the assumption fails, thereby providing a complete answer to the above question. Our analysis is constructive and the obtained theoretical results hold under fairly general assumptions on model architecture. Extensive experiments on GPT-2 and Qwen3-4B demonstrate that our algorithm produces prefixes with strong control over model outputs.

Related papers