Training-Free Task Vectors for LLM Behavioral Control
cs.LG, cs.AI
Submitted: 2026-09-08
Updated: 2026-10-14
Code: https://github.com/EleutherAI/lm-evaluation-harness5https:
License: http://creativecommons.org/licenses/by/4.0/
The gist: Task vectors enable post-training model editing by identifying semantically meaningful directions in weight space, typically computed as the difference between a fine-tuned model and its pretrained
Terminology
Abstract
Task vectors enable post-training model editing by identifying semantically meaningful directions in weight space, typically computed as the difference between a fine-tuned model and its pretrained initialization. However, this reliance on fine-tuning makes discovering such directions costly and limits the practicality of post-training model editing. To address this limitation, we introduce Training-Free Task Vectors (TFTVs), a novel method to compute task-vector-like directions without requiring fine-tuning. Our method maps activation steering vectors to rank-one weight-space edits using only forward-pass statistics, while satisfying arithmetic properties that directly support learning via addition, forgetting via subtraction, and the composition of multiple edits. Empirically, we evaluate TFTVs on large language model behavioral control tasks and show that they consistently amplify, suppress, and compose target behaviors while preserving general knowledge and problem-solving skills. We also validate our method against other editing and steering baselines, experimentally demonstrating that TFTVs achieve stronger trait control with better or competitive utility preservation. We hope our work opens new directions for the community in post-training model editing and broader training-free model control. Code is available on the project website: tftv-llm.github.io.
Sources
- GPT-4 Technical Report
- Emergent Misalignment: Narrow finetuning can produce broadly misaligned LLMs
- Evaluating Large Language Models Trained on Code
- Persona Vectors: Monitoring and Controlling Character Traits in Language Models
- Training Verifiers to Solve Math Word Problems
- Sparse Autoencoders Find Highly Interpretable Features in Language Models
- Decomposing Task Vectors for Refined Model Editing
- Steering Language Models with Weight Arithmetic
- The Llama 3 Herd of Models
- Measuring Massive Multitask Language Understanding
- Editing Models with Task Arithmetic
- Dataless Knowledge Fusion by Merging Weights of Language Models
- The Rogue Scalpel: Activation Steering Compromises LLM Safety
- AdaRank: Adaptive Rank Pruning for Enhanced Model Merging
- Ministral 3
- Mass-Editing Memory in a Transformer
- Fast Model Editing at Scale
- Fine-tuning Aligned Language Models Compromises Safety, Even When Users Do Not Intend To!
- Steer2Edit: From Activation Steering to Component-Level Editing
- Gemma 4 Technical Report
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks