Alignment is Localized: A Causal Probe into Preference Layers

Reinforcement Learning frameworks, particularly those utilizing human annotations, have become an increasingly popular method for preference fine-tuning, where the outputs of a language model are tuned to match a certain set of behavioral policies or guidelines. Reinforcement Learning through Human Feedback (RLHF) is perhaps the most popular implementation of such a framework, particularly for aligning LMs toward safety and human intent. However, the internal workings of how such alignment is achieved remain largely opaque. In this work, we systematically analyze preference optimization for language model alignment by applying layer-wide causal patching between a base model and its tuned counterpart across human preference pairs. We implement our methodology on \textit{Llama-3.2-1B}, and find that alignment is spatially localized: mid-layer activations encode a distinct subspace that causally determines reward-consistent behavior, while early and late layers remain largely unaffected. Utilizing LASSO regression, we also find that only a small number of layers possess non-zero coefficients linking activation distances to reward gains. Overall, we show that, at least for some language models, alignment from human-based, preferential tuning is a directional, low rank process, rather than diffuse and parameteric.

Paper

References (8)

03A mathematical framework for transformer circuits2021 · Transformer Circuits Thread
04Locating and editing factual associationsin gpt
05(A3) α -Interpolation (Dose–Response). We scale the strength of the causal intervention with a mixing coefficient α ∈ [0 , 1]can be
06(A4) Source Variants. We patch activations corresponding to the chosen , rejected , and ( chosen – rejected ) differences separately, seeing if this changes the alignment signal
07Progress measures for grokking viamechanistic interpretability
08(A5) Low-Rank Reconstruction. We perform singular value decomposition (SVD) on the tuned activations and patch only the top-k principal components

Similar papers

© 2026 NYSGPT2525 LLC