Error Correction in Trained TTT Layers
Not on the map. Why?
Abstract (arXiv)
Test-time-training (TTT) layers update fast-weight memory while processing an input sequence. A gradient-based update, however, does not by itself establish how the memory revises an existing association. We study the structure and empirical role of prediction-derived updates in released TTT implementations. For dot-product objectives, we characterize the absence of a prediction-error residual and establish conditions under which inner-loop ascent and descent are equivalent after reparameterization. For the LayerNorm-based squared-error objective used by TTT-Linear and TTT-MLP, we derive the inner gradient and identify a prediction-derived component that cancels at the released initialization in the zero-epsilon limit. We then measure the full normalized prediction's contribution on three trained checkpoints. Removing this contribution while retaining the LayerNorm Jacobian gives a mean gradient norm ratio of 0.25 to 0.32 across the models. On 100,303 tokens from War and Peace, the same intervention with local per-token norm matching raises perplexity by 2.8-2.9% on TTT-Linear-760M and 1.3B, whereas freezing all fast-weight updates raises it by factors of 4.2-4.5. Separate synthetic tests illustrate the dependence of plain delta-rule overwriting on step size. The results distinguish a prediction-derived update from its behavioral value and show why implementation analysis and live interventions are both needed to assess test-time memory.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 12 of its 13 references that are in the map. First posted on arXiv 12 Oct 2026.
Builds on
12 of 13 references.
References · 12
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 502
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
- 2024Titans: Learning to Memorize at Test Timecited by 328
- 2023Uncovering mesa-optimization algorithms in Transformerscited by 100
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 74
- 2025It's All Connected: A Journey Through Test-Time Memorization, Attentional Bias, Retention, and Online Optimizationcited by 64
- 2025End-to-End Test-Time Training for Long Contextcited by 62
- 2025ViT3: Unlocking Test-Time Training in Visioncited by 18
- 2026Test-Time Training with KV Binding Is Secretly Linear Attentioncited by 13
Built on by
Not cited yet.