Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Weight Patching: Toward Source-Level Mechanistic Localization in LLMs

About

Mechanistic interpretability seeks to localize model behavior to the internal components that causally realize it. Prior work has advanced activation-space localization and causal tracing, but modules that appear important in activation space may merely aggregate or amplify upstream signals rather than encode the target capability in their own parameters. To address this gap, we propose Weight Patching, a parameter-space intervention method for source-oriented analysis in paired same-architecture models that differ in how strongly they express a target capability under the inputs of interest. Given a base model and a behavior-specialized counterpart, Weight Patching replaces selected module weights from the specialized model into the base model under a fixed input. We instantiate the method on instruction following and introduce a framework centered on a vector-anchor behavioral interface that provides a shared internal criterion for whether a task-relevant control state has been formed or recovered in open-ended generation. Under this framework, the analysis reveals a hierarchy from shallow candidate source-side carriers to aggregation and routing modules, and further to downstream execution circuits. The recovered component scores can also guide mechanism-aware model merging, improving selective fusion across the evaluated expert combinations and providing additional external validation.

Chenghao Sun, Chengsheng Zhang, Guanzheng Qin, Rui Dai, Xinmei Tian• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval
HumanEval Score30.49
93
Language UnderstandingMMLU
MMLU Accuracy55.01
77
Instruction FollowingIFEval
Genuine-Followup Rate53.97
38
Code GenerationMBPP
MBPP Score35.2
32
Mathematical ReasoningMATH
MATH Score14
32
Showing 5 of 5 rows

Other info

Follow for update