mlsanityData to 14 Oct 2026

When to Ponder: Adaptive Compute Allocation for Code Generation via Test-Time Training

arXiv 2601.00894 ↗First posted 31 Dec 2025cs.LG cs.CLcited by 1

Not on the map. Why?

Abstract (arXiv)

Large language models apply uniform computation to all inputs, regardless of difficulty. We propose PonderTTT, a gating strategy using the TTT layer's self-supervised reconstruction loss to selectively trigger Test-Time Training (TTT) updates. The gating decision itself is training-free--requiring no learned classifier or auxiliary networks; only a single scalar threshold is initially calibrated on unlabeled data and continuously adapted via EMA to maintain target update rates. Our experiments with GPT-2 models (124M to 1.5B) on code language modeling (The Stack v2, teacher-forced perplexity) demonstrate that this signal is inference-compatible, requiring no ground-truth labels. Our Reconstruction Gating achieves 82-89% Oracle Recovery while being fully training-free, significantly outperforming Random Skip baselines (up to 16% lower loss on OOD languages).

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 6 of its 9 references that are in the map. First posted on arXiv 31 Dec 2025.

Builds on

6 of 9 references.

References · 6

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  2. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
  3. 2018Universal Transformerscited by 647
  4. 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
  5. 2022Confident Adaptive Language Modelingcited by 310
  6. 2021PonderNet: Learning to Pondercited by 168

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.