To Copy or Not to Copy: Copying Is Easier to Induce Than Recall
Mehrdad Farahani, Franziska Penzkofer, Richard Johansson
TL;DR
This work tackles how large language models arbitrate between parametric knowledge and contextual information in retrieval-augmented settings. It introduces an arbitration vector, computed from residual-stream centroids that distinguish IC from RC across 27 relations, and demonstrates that additive steering can toggle between Copy$\rightarrow$Recall and Recall$\rightarrow$Copy in two architectures (Gemma2-2B and T5Gemma-2B) and two QA benchmarks. The results reveal a robust asymmetry: copying from context is easier to induce than recalling from memory, with mechanistic analyses showing attention reallocation and MLP-driven updates as key drivers. The findings offer a general, non-fine-tuned control signal for directing information sourcing in RAG-like systems, while acknowledging limitations related to prompting, model scale, and end-to-end retriever integration.
Abstract
Language models used in retrieval-augmented settings must arbitrate between parametric knowledge stored in their weights and contextual information in the prompt. This work presents a mechanistic study of that choice by extracting an \emph{arbitration vector} from model activations on a curated dataset designed to disentangle (i) irrelevant contexts that elicit parametric recall and (ii) relevant but false contexts that elicit copying. The vector is computed as the residual-stream centroid difference between these regimes across 27 relations, and is injected as an additive intervention at selected layers and token spans to steer behavior in two directions: Copy$\rightarrow$Recall (suppressing context use) and Recall$\rightarrow$Copy (inducing the model to copy any token from the context). Experiments on two architectures (decoder-only and encoder/decoder) and two open-domain QA benchmarks show consistent behavior shifts under moderate scaling while monitoring accuracy and fluency. Mechanistic analyses of attention routing, MLP contributions, and layer-wise probability trajectories reveal an asymmetry: inducing copying is an easy ``reactivation'' process that can be triggered at different locations in the input, while restoring recall is a ``suppression'' process that is more fragile and strongly tied to object-token interventions.
