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

arXiv:2601.00894 · cs.LG, cs.CL · Submitted 2026-08-20 · Read on arXiv

Gihyeon Sim

cs.LG, cs.CL

Submitted: 2026-08-20

Updated: 2026-08-24

Terminology

• Adaptation — 9× in this paper · explained in 1 episode
• Perplexity — 8× in this paper · explained in 3 episodes
• Test-Time Training — 8× in this paper
• Generalization — 6× in this paper · explained in 13 episodes
• Transformer — 6× in this paper · explained in 2 episodes
• Ablation — 4× in this paper · explained in 1 episode
• Accuracy — 4× in this paper · explained in 1 episode
• Calibration — 4× in this paper · explained in 7 episodes
• Explainability — 4× in this paper · explained in 3 episodes
• Distribution Shift — 3× in this paper · explained in 5 episodes
• Meta-learning — 3× in this paper · explained in 2 episodes
• Verification — 3× in this paper · explained in 2 episodes
• Adaptive Compute Allocation — 2× in this paper · explained in 1 episode
• Benchmark — 2× in this paper · explained in 2 episodes
• Contextual Bandits — 2× in this paper · explained in 3 episodes
• Adam — 1× in this paper · explained in 2 episodes
• Adaptability — 1× in this paper · explained in 1 episode
• Consistency — 1× in this paper · explained in 3 episodes
• Interpretability — 1× in this paper · explained in 15 episodes
• Low-Rank Adaptation (LoRA) — 1× in this paper · explained in 9 episodes
• Mixture-of-Experts (MoE) — 1× in this paper · explained in 7 episodes
• Model-Agnostic — 1× in this paper · explained in 2 episodes
• Online Adaptation — 1× in this paper · explained in 3 episodes
• Sample Efficiency — 1× in this paper · explained in 4 episodes
• Transformer Architecture — 1× in this paper · explained in 2 episodes
• Transformer Models — 1× in this paper · explained in 2 episodes

Sources

• PonderNet: Learning to Ponder
• Gemma 3 Technical Report
• StarCoder 2 and The Stack v2: The Next Generation

Related papers

• Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
• AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
• Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
• Convergence issues in Relational Concept Analysis based on AOC-posets
• Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
• Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks