Exact Record Omission in Delta Attention: A Transport Criterion, Its Cost, and a Replay Certificate
Organizations: Vy Labs, Inc.
Abstract
An assistant can stop repeating a deleted statement while its recurrent memory still carries that statement's influence. We examine this distinction by saving the state difference immediately after a record, carrying this saved difference, or receipt, through later updates, and comparing the corrected state with the state built from the same conversation with the record omitted. Unrolling the recurrence gives an exact criterion: transport reaches this never-stored state if and only if the additional differences created by later updates cancel after transport. We evaluate eleven conditions using 40 prospectively selected synthetic Kimi Linear contexts, with 40 separate calibration contexts and paired uncertainty estimates computed across complete records. Attention masking removed all 40 exact greedy target answers, yet a fixed three-query candidate-scoring attack with known candidate values and log-probability access attained AUC 0.740 (95% interval 0.695--0.809); the measured false-positive rate was 7.5%. Prompt-only forgetting still returned 33 of the 40 targets. Checkpoint replay matched the complete declared active state and audit logits in all 80 contexts, while preserving all 120 retained answers in the evaluation cohort. An independent Qwen cohort and matched original-bf16/8-bit Kimi controls confirmed native transport mismatch beyond the numerical error measured in matched controls. Replay also supported record replacement and successive deletions with a new record added between them. These results provide a practical audit of both the requested memory change and the assistant's remaining behavior, with replay work determined by the surviving suffix.
Figures & tables
| term | meaning | relevant sections |
|---|---|---|
| target record | the record a user asks to delete; one contiguous span of tokens | all |
| never-stored state | state built from the recorded history with the target left out (Eq. 2 ) | all |
| receipt | the state difference at the boundary just after the record | § VI-A , § VI-C |
| transport | carrying the receipt forward through the record-present transitions | § VI-A , § VI-C |
| forcing | new differences created by later updates because the record was present; these are additional to the boundary receipt | § VI-A –§ VI-D |
| imprint (suffix sweep) | present never-stored never-stored per layer; residual without an edit | § VI-C |
| question | comparison or test | evidence | outside the claim |
|---|---|---|---|
| When does receipt transport succeed? | affine recurrence algebra | Theorem 1 , Corollary 1 | non-affine updates |
| Does native KDA meet the criterion? | frozen-input counterfactual vs. native never-stored state | relative state residual; forcing norm ratio | caches outside the recurrence |
| What do the methods cost? | partial recomputation; deferred log vs. checkpoint of recurrent and convolution state | residual vs. fraction recomputed; storage in bytes | other implementations and storage schemes |
| Is residual information detectable? | probes on each corrected state vs. the never-stored reference | lift, rank, greedy, sampling; three-query candidate scoring | untested adaptive attacks; untested generation-only attacks |
| Do next-token distributions match? | exact rebuilt never-stored state | first-token KL divergence; requires probability access | equivalence of full distributions |
| Does replay match the reference? | per-array float32 equality | 94 active arrays (recurrent, convolution, and MLA), offsets, and logits in the prospective cohort (§ VI-G ) | prior outputs; unused allocation; downstream causal effects |
| intervention | intervention setup | observed result |
|---|---|---|
| zero recurrent matrices | leave MLA caches and convolution arrays unchanged | in one synthetic example, the ward code remains rank 1; first-token probability |
| attention masking | hide target attention entries; leave recurrent and convolution state unchanged | nonzero first-token KL on all 216 synthetic and 144 TOFU configuration–cut pairs |
| frozen-input transport | identical suffix transitions and writes | receipt matches the frozen-input record effect to relative error |
| native transport | subtract the transported boundary receipt without correcting later forcing | residual – of record-present suffix-A state norm; forcing norm ratio – |
| checkpoint replay | restore the preceding checkpoint and process the retained suffix again | zero residual on logits and 80 KDA arrays |
| condition | target lift [95% CI] (nats/token) | exact target answers / 40 | AUC [95% CI] | detection % |
|---|---|---|---|---|
| record present | 3.240 [3.062, 3.411] | 40 | 1.000 [1.000, 1.000] | 100 |
| prompt forgetting | 3.121 [2.943, 3.291] | 33 | 1.000 [1.000, 1.000] | 100 |
| attention masked | 0.486 [0.426, 0.546] | 0 | 0.740 [0.695, 0.809] | 30 |
| masked + transport | 0.0178 [0.0108, 0.0245] | 0 | 0.519 [0.511, 0.543] | 7.5 |
| checkpoint replay | 0 [0, 0] | 0 | 0.500 [0.500, 0.500] | 7.5 |
| never stored (reference) | 0 [0, 0] | 0 | 0.500 [0.500, 0.500] | 7.5 |
Appendix figures & tables15 assets
Supplementary material from the paper’s appendix.
Appendix
| Condition | Target log-probability | Retained log-probability | Continuation NLL |
|---|---|---|---|
| Record present | |||
| Attention masking | |||
| Static receipt | |||
| Diagonal ledger | |||
| Receipt transport | |||
| Masking + transport |
| Condition | AUC [95% interval] | TPR, % [two-stage 95% interval] | TPR Wilson 95% interval, % |
|---|---|---|---|
| Record present | |||
| Attention masking | |||
| Static receipt | |||
| Diagonal ledger | |||
| Receipt transport | |||
| Masking + transport |
| quantity | short suffix | long suffix | paired long minus short |
|---|---|---|---|
| native transport effect scaled | |||
| native transport state scaled | |||
| frozen static state scaled | |||
| frozen ledger state scaled |
| condition | derived 8-bit lift | original bf16 lift | bf16 minus 8-bit lift |
|---|---|---|---|
| present | |||
| attention masking | |||
| receipt transport | |||
| masking + transport | |||
| checkpoint replay | |||
| prompt forgetting |
| outcome | count | interval, % |
|---|---|---|
| new answer exact after amendment | 12/12 | 75.8–100 |
| old answer exact after amendment | 0/12 | 0–24.2 |
| all retained answers exact after amendment | 11/12 | 64.6–98.5 |
| all retained targets contained after amendment | 12/12 | 75.8–100 |
| all sequence transitions match fresh state/logits | 6/6 | 61.0–100 |
| both deleted answers absent in both orders | 6/6 | 61.0–100 |
| surface | attempted | qualified | audited unit |
|---|---|---|---|
| Kimi short tables | 8 records | 8 | final logits and 80 KDA arrays per record |
| Kimi 128-record context | 9 records | 9 | same exactness surface plus nine position timings |
| Kimi long notes | 5 records | 4 a | same exactness surface |
| Kimi fixed-position masking vs. replay | 12 records (3 per corpus, 4 corpora) | 11 b | lift after masking and after replay; replay residual on logits and KDA state |
| Kimi suffix sweep, synthetic | 24 configurations | 24 | 20 layers at 10 suffix lengths; probes and 100 sampled draws per condition |
| Kimi suffix sweep, TOFU | 16 configurations | 16 | same surface |
| Synthetic: 24 configurations | TOFU: 16 configurations | |||||||
| condition | lift | rank | greedy % | KL | lift | rank | greedy % | KL |
| record present | 2.16 | 1 | 100 | 4.55 | 1.34 | 1 | 25 | 0.549 |
| static receipt | 2.15 | 1 | 100 | 4.64 | 1.31 | 1 | 25 | 0.457 |
| diagonal ledger | 2.16 | 1 | 100 | 4.52 | 1.33 | 1 | 25 | 0.563 |
| full-matrix receipt | 2.15 | 1 | 100 | 4.62 | 1.34 | 1 | 25 | 0.536 |
| attention masked | 0.0622 | 89 | 0 | 0.0354 | 0.0472 | 1 | 0 | 0.00419 |
| Synthetic: 2,400 draws | TOFU: 1,600 draws | |||
|---|---|---|---|---|
| condition | 0 tokens | 4,096 | 0 tokens | 4,096 |
| record present | 2349 | 2287 | 358 | 325 |
| static receipt | 2314 | 2278 | 245 | 326 |
| attention masked | 1 | 1 | 2 | 11 |
| masked + static receipt | 1 | 0 | 1 | 10 |
| never-stored state | 1 | 0 | 0 | 12 |
| corpus | position | present lift | lift after mask | replay residual | suffix tokens |
|---|---|---|---|---|---|
| TOFU | early | ||||
| TOFU | middle | ||||
| TOFU | late | ||||
| MIMIC-CDS | early | ||||
| MIMIC-CDS | middle | ||||
| MIMIC-CDS | late |
| correction | population | result |
|---|---|---|
| checkpoint replay | 21 qualified records | logits and 80 arrays: residual 0 |
| full-matrix receipt | 36 layer–suffix checks | forcing norm ratio 0.064–1.010 |
| example | intervention | greedy answer |
| synthetic ward code | recurrent matrices zeroed | “OBSIDIAN-TWO” |
| synthetic ward code | checkpoint replay | “The ward code for patient 5182 is not provided.” |
| TOFU biography | attention rows masked | “Hsiao Yun-Hwa identifies as a woman.” |
| TOFU biography | checkpoint replay | “Hsiao Yun-Hwa identifies as a woman.” |
| TOFU biography | never stored | “Hsiao Yun-Hwa identifies as a woman.” |
| position | present lift | after attention mask | replay residual | replay seconds |
|---|---|---|---|---|
| early | ||||
| middle | ||||
| late |
| state | operation | greedy answer |
|---|---|---|
| original | ingest old statement | “Breast cancer.” |
| masked | mask the statement’s attention rows | “The patient did not mention anything about their mother’s health condition.” |
| amended | restore, replace, replay | “Benign breast lump” |
| reference | corrected from the start | “Benign breast lump” |
| synthetic | MIMIC-CDS | MIMIC-Note | ||||
| write rule | raw | ledger | raw | ledger | raw | ledger |
| additive | — | — | — | |||
| per-channel decay | 0.007–0.066 | 0.008–0.078 | 0.015–0.109 | |||
| KDA delta rule | 0.151–0.419 | 0.139–0.418 | 0.116–0.300 | 0.089–0.300 | 0.122–0.489 | 0.081–0.494 |
| family | static | ledger | full- matrix | native ratio median |
|---|---|---|---|---|
| Mamba-2 | ||||
| Falcon-H1 | ||||
| RWKV-7 |
LLM Unlearning
Large language model unlearning studies how to remove specified knowledge or behaviors while preserving a model’s unrelated capabilities.
See all 191 papers