arXiv · 2502.06115
Task-driven Layerwise Additive Activation Intervention
Abstract
Modern language models (LMs) have significantly advanced generative modeling in natural language processing (NLP). Despite their success, LMs often struggle with adaptation to new contexts in real-time applications. A promising approach to task adaptation is activation intervention, which steers the LMs' generation process by identifying and manipulating the activations. However, existing interventions are highly dependent on heuristic rules or require many prompt inputs to determine effective interventions. This paper proposes a layer-wise additive activation intervention framework that optimizes the intervention process, thus enhancing the sample efficiency. We benchmark our framework on various datasets, demonstrating improvements in the accuracy of pre-trained LMs and competing intervention baselines.
Explore related subjects
Keep this discovery
Hieu Trung Nguyen, Bao Nguyen, Binh Nguyen, Viet Anh Nguyen. 2025-02-10. Task-driven Layerwise Additive Activation Intervention. https://arxiv.org/abs/2502.06115
Cite the original work for its findings. Save a collection to share your selection of sources.