PrimeTTT: Priming Test-Time Training for LLMs via Meta-Learned Initialization

Xiangqi Wang, Dilinuer Aishan, Yujun Zhou, Yue Huang, Xiangliang Zhang


Abstract
Test-time training (TTT) adapts a frozen large language model (LLM) at inference time by updating prompt-specific parameters before generation. It is especially appealing for long-context reasoning. However, vanilla TTT yields only marginal gains on modern LLMs. We trace this bottleneck to two coupled failures: the inner loop is unprimed, because LLMs are trained as static endpoints rather than initialization points for test-time gradient adaptation; and the proxy objective is misaligned, because prompt-masked next-token prediction (PM-NTP) rewards prompt memorization rather than answer correctness. We propose PrimeTTT, a lightweight TTT method that casts this adaptation problem as meta-learning. A compact hypernetwork emits a prompt-specific LoRA initialization for the inner loop, trained with a MAML-style second-order objective: the inner loop descends the test-time-available PM-NTP loss, while the outer loss is answer-conditioned NTP on the gold answer and is backpropagated through the unrolled inner steps. Across out-of-distribution benchmarks disjoint from training, PrimeTTT consistently outperforms strong TTT baselines and transfers across model families and scales.
Anthology ID:
2026.emnlp-main.271
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
6152–6170
Language:
URL:
https://aclanthology.org/2026.emnlp-main.271/
DOI:
Bibkey:
Cite (ACL):
Xiangqi Wang, Dilinuer Aishan, Yujun Zhou, Yue Huang, and Xiangliang Zhang. 2026. PrimeTTT: Priming Test-Time Training for LLMs via Meta-Learned Initialization. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 6152–6170, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
PrimeTTT: Priming Test-Time Training for LLMs via Meta-Learned Initialization (Wang et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.271.pdf