Beyond Update Counts: A Control Map for Evaluating Test-Time Model Updates
Organizations: Meta AI
Abstract
Test-time updating lets a language model change persistent state after training---such as model weights, adapter parameters, or a memory module---before or while answering queries. Existing evaluations often compare systems using update count or frequency and performance measured immediately afterward. These summaries omit what is updated, where, when, and in what order. The same total can therefore hide an important difference: one system may answer after one update while another answers after two. We introduce the Control Map, an evaluation framework that records the committed updates preceding each scored answer and separately evaluates behavioral changes, immediate learning (acquisition), later recall (retention), preservation of unrelated behavior (locality), and performance in new settings. We prove when counts alone determine final state and show, under stated assumptions, how differences between pre-answer histories limit differences in average scores. Using ControlMap, our experiments reveal three insights. First, update scheduling matters: at a 760M-parameter checkpoint, 24 schedules with the same update count yield 19.7%--34.2% accuracy. Second, learning an edit does not guarantee that it lasts or remains contained: all six selected CounterFact LoRA settings learn the new information, while ROME retains edits at all 13 operating points but fails locality at each. Third, no editor performs best consistently across models, benchmarks, and tests. Evaluations should therefore report update histories and assess acquisition, retention, and locality separately.
Figures & tables
| Choice | Comparison and controls | Observed result | Supported takeaway |
|---|---|---|---|
| Placement | 24 band–period assignments; one MAC-760M checkpoint; 6,000 paired Single-NIAH rows; equal prompt-only writes | Accuracy – ; the band–period model descriptively explains of between-schedule variation | Placement matters in this task and checkpoint; |
| Timing | Same selected events and terminal actions; writes before versus after the focal query | Immediate-minus-deferred acquisition: and nats across four GPT-2 sizes ( Radford et al., 2019 ) ; Pythia-410M ( Biderman et al., 2023 ) : nats, range | The estimated effect is positive in all four GPT-2 assays; the Pythia range crosses zero |
| Dose | Eight objectives, eight actions, 32 stochastic-gradient-descent steps, and terminal adapter–optimizer construction fixed | Across 18 deliberately varied LoRA settings, the best tested doses occur in settings; the best-dose prediction specified before outcomes is confirmed in 7/18 settings | The best tested dose varies across settings |
Appendix figures & tables28 assets
Supplementary material from the paper’s appendix.
Appendix
| Claim / audit question | Study | Timing | Unit and scale | Location |
|---|---|---|---|---|
| Equal counts need not identify equal executed treatments | MAC-760M trace atlas; 24 depth–cadence schedules on eight panels | Outcome-blind exact enumeration | 27,072 benchmark rows; 24 policies per row | Section B ; Table 6 |
| Layer placement can change accuracy at fixed update count | MAC-760M placement surface on Single-NIAH | One fixed checkpoint; additive summary descriptive | 6,000 paired rows; all 24 schedules | Figure 2 ; Section B.4 |
| Order can masquerade as dose–locality structure | GPT-2XL LoRA complete-order correction | Post-hoc deterministic correction | 1,600 order-matched pairs | Section B |
| Query-relative phase can change acquisition | GPT-2 family and Pythia-410M phase studies | Per-model frozen gates; cross-model synthesis descriptive | Four LoRA initializations per model | Section B |
| The best tested dose varies across settings | LoRA/SGD four-dose grids | Purposive, partly adaptive setting collection; within-setting calls frozen | 18 settings; four adapter initializations per setting | Section B.5 |
| Acquisition need not imply feasible memory | LoRA and ROME grids; fixed MEMIT, ROME, and AlphaEdit controls | Cohort-specific frozen or declared timing | Family/block resampling within each assay | Table 7 ; Sections C.2 – C.10 |
| Declared aggregate deviation | Binary-effect cap (points) |
| Study | Recorded boundary | Conformance evidence | Ancestry-coverage claim |
|---|---|---|---|
| MAC trace atlas | Declared deterministic scheduler graph | Complete 27,072-row enumeration; 184 outcome-blind checks | Exact relative to the declared scheduler adapter; no whole-program claim |
| Constructed neural LoRA test | Recorded live state transitions and nonzero score-gradient paths | 16/16 chains validate; ; 16 checks reject and restoration passes | Conformance for recorded paths in one randomly initialized small Llama; unrecorded dependencies remain unverified |
| GPT-2XL LoRA phase study | Declared PEFT-LoRA/SGD event and optimizer-step trace | Separate implementation matches 3,072/3,072 fields; 15/15 tests reject five omission/alias/substitution fault classes | Structural reconstruction only; no live-state observation or completeness theorem |
| WikiRecent | Score-ancestral editor outer calls | 512/512 policy rows match precomputed certificates | EasyEdit-internal false-positive/negative rates unavailable |
| GPT-2XL endpoints | Reconstructed editor outer-call traces | 448/448 compiled cells agree; 1,792 family–order units and 3,584 policy traces match | Editor internals were not observed |
| Llama-3.2-3B endpoints | Declared dose-four/terminal atomic outer-call traces | Post-outcome projection: 3,584/3,584 rows conform; all 1,792 paired traces separate ( ) | EasyEdit internals and cross-bundle correspondence unavailable; diagnostic component only |
| Context | Policy | fail | inconclusive | pass | At |
|---|---|---|---|---|---|
| New 160-family CounterFact panel | MEMIT-high | pass | |||
| New 160-family CounterFact panel | ROME-low | inconclusive | |||
| WikiRecent shared locality | MEMIT-high | fail | |||
| WikiRecent shared locality | ROME-low | fail |
| Panel | Rows | Pair alias | Full rows | Rank/9 | |
|---|---|---|---|---|---|
| SocialIQA | 1,954 | 14.31% | 0.05% (1 row) | 9 | 1954.000 |
| BoolQ | 3,270 | 0.15% | 96.61% | 9 | 1.035 |
| LAMBADA | 5,153 | 0.01% | 99.77% | 9 | 1.002 |
| PIQA | 1,838 | 66.60% | 0.00% | 6 | — |
| WinoGrande | 1,267 | 43.58% | 0.00% | 6 | — |
| HellaSwag | 10,042 | 5.84% | 40.60% | 9 | 2.445 |
| Assay | Acquisition | Retention | Locality | Full gate |
|---|---|---|---|---|
| LoRA, six selected CounterFact settings | 6/6 pass | 0/6 pass | 0/6 pass | 0/6 pass |
| ROME, 13 CounterFact operating points | 13/13 pass | 13/13 pass | 0/13 pass | 0/13 pass |
| MEMIT-high, pilot-disjoint 160-family CounterFact point | pass | pass | pass | pass |
| ROME-low, same 160-family point | pass | pass | inconclusive | inconclusive |
| AlphaEdit, shared-eight-event CounterFact doses | 4/4 pass | 4/4 pass | 4/4 pass | 4/4 pass |
| Setting | Call | Peak | Acq. [95% range] | Ret. [95% range] | Install | Keep install | Locality [95% range] |
|---|---|---|---|---|---|---|---|
| Pythia current | 1 | 1 | |||||
| Pythia step100000 | 1 | 2 | |||||
| GPT-2 Large | 4 | 4 | |||||
| GPT-2XL | 4 | 4 |
| Setting | |||
|---|---|---|---|
| Pythia current | |||
| Pythia step100000 | |||
| GPT-2 Large | |||
| GPT-2XL |
| Model | [95% range] | Retention (nats) / (pp) | ||
|---|---|---|---|---|
| Qwen2.5-0.5B | / | |||
| Llama-3.2-1B | / |
| Editor | Native strength | Frozen -response label | |||
|---|---|---|---|---|---|
| MEMIT | high | dose-insensitive | |||
| ROME | low | inconclusive | |||
| ROME | default | dose-insensitive | |||
| ROME | high | dose-insensitive |
| Dose | [95% range] | Install | Retain/install | Full gate | |
|---|---|---|---|---|---|
| 1.09375 | 0.375 | 1.000 | pass | ||
| 1.09375 | 0.375 | 1.000 | pass | ||
| 1.09375 | 0.375 | 1.000 | pass | ||
| 1.09375 | 0.375 | 1.000 | pass |
| Point | [95% range] | range | Install | Keep install | |
|---|---|---|---|---|---|
| MEMIT-high | |||||
| ROME-low |
| Editor | [95% range] | ( range) | Install | Retain/install | Full gate |
|---|---|---|---|---|---|
| MEMIT-high | 0.4781 | 1.0000 | pass | ||
| ROME-low | 0.4172 | 0.9925 | inconclusive : locality |
| Dose | 95% range for | Install / retain | Gates | ||
|---|---|---|---|---|---|
| 1 | 6.37607 | 6.71875 | 0.3125 / 0.8000 | A, R pass ; L, full inconclusive | |
| 2 | 6.37283 | 6.71875 | 0.3125 / 0.8000 | A, R pass ; L, full inconclusive | |
| 4 | 6.28910 | 6.71875 | 0.3125 / 0.8000 | A, R pass ; L, full inconclusive | |
| 8 | 5.60322 | 6.71875 | 0.3125 / 0.8000 | A, R pass ; L, full inconclusive |
| Sensitivity target | Estimate (pp) | SE (pp) | 95% range (pp) |
|---|---|---|---|
| Family-weighted point, relation-cluster CR1 | 0.9353 | ||
| Equal weight over 33 observed relation means | 0.9405 |
| Writer | Frozen prediction | Realized class | Acquisition range (nats) | Locality damage (pp) | Full-pass doses |
|---|---|---|---|---|---|
| AlphaEdit | safe + insensitive | safe + insensitive | |||
| FT-L | failing + sensitive | failing + inconclusive |
| Context | MEMIT-high | ROME-low | Mean paired difference | Effect status |
|---|---|---|---|---|
| Canonical prompt | 4.1261 | 3.9119 | pass | |
| One rephrase | 2.9104 | 2.1925 | pass |
| Context | Policy | Acquisition | Install | Retain/inst. | Locality | Full |
|---|---|---|---|---|---|---|
| Canonical | MEMIT | 4.1261 | .8867 | .9912 | fail | |
| Canonical | ROME | 3.9119 | .8047 | .9563 | fail | |
| Rephrase | MEMIT | 2.9104 | .3828 | 1.0000 | fail | |
| Rephrase | ROME | 2.1925 | .2070 | .9811 | fail |
| Endpoint | Editor | MEMIT–ROME locality success | ||
|---|---|---|---|---|
| CounterFact | MEMIT-high | 8.3016 | 2.0781 | |
| CounterFact | ROME-low | 7.5166 | 4.6406 | — |
| KnowEdit-zsRE | MEMIT-high | 4.457329 | 31.328125 | |
| KnowEdit-zsRE | ROME-low | 3.880083 | 23.828125 | — |
| KnowEdit-WikiData-CF | MEMIT-high | 3.997292 | 69.726562 | |
| KnowEdit-WikiData-CF | ROME-low | 2.704324 | 53.808594 | — |
| Endpoint | Families | Editor-level [95% range] | Paired [95% range] | Paired [95% range] |
|---|---|---|---|---|
| CounterFact | 160 | M: R: | ||
| KnowEdit-zsRE | 160 | M: R: | ||
| KnowEdit WikiData-CounterFact | 128 | M: R: |
| Record | Evidence | Status |
|---|---|---|
| Resolution | Outer-call reconstruction gives for 640 CounterFact, 640 zsRE, and 512 WikiData-CF family–order pairs. | conditional pass ; diagnostic |
| Effect | Source acquisition/locality contrasts are nats and points; endpoint signs are / . | Acq. : undefined (no directional gate); locality : pass / fail / fail |
| Feasibility | CounterFact: MEMIT passes and ROME is inconclusive ; both fail locality on zsRE and retention/locality on WikiData-CF. Costs are ungated. | source mixed; targets fail |
| Transport | The correspondence fixes GPT-2XL, method identities, strengths, four-call template, dose, and orientation while endpoint data/instruments change. Locality reverses to and . | fail |
| Endpoint | Locality success (points) | Acquisition (nats) |
|---|---|---|
| CounterFact | ||
| KnowEdit-zsRE | ||
| WikiData-CounterFact |
| Bundle | Endpoint | Acquisition (nats) | Locality (points) |
|---|---|---|---|
| GPT-2XL fixed | CounterFact | ||
| zsRE | |||
| WikiData-CF | |||
| Llama native | CounterFact | ||
| zsRE | |||
| WikiData-CF |
| Unsupported shortcut | Relevant evidence | Evaluation alternative |
|---|---|---|
| Equal update counts imply the same pre-answer history | 24 schedule labels produce 1, 4, 12, or 24 history classes | Record the ordered updates before each answer |
| Immediate learning implies durable, local learning | LoRA: acquisition passes 6/6; retention and locality pass 0/6 | Test acquisition, retention, and locality separately |
| A method ordering carries over to a new setting | Observed GPT-2XL and Llama-3.2-3B orderings differ across evaluated bundles; the cross-model mapping is incomplete | Evaluate the same methods directly in the new setting |
| An average improvement implies usability | WikiRecent mean improvement passes while locality fails | Require every prespecified behavioral check to pass |
LLM Evaluation
Large language model evaluation measures what models can do, how reliably they do it, and whether benchmark scores predict real-world performance.
See all 2,408 papers