AIAny
Icon for item

TTPO: Test-Time Policy Optimization

A test-time method that adapts LLMs without labels by distilling rollouts that agree with majority pseudo-labels and penalizing disagreeing rollouts via grouped RL, improving robustness under frequent pseudo-label errors.

Introduction

Why this matters Most test-time training (TTT) methods for reasoning rely on majority-vote pseudo-labels but fail catastrophically when votes are wrong because every token of a mistaken teacher is reinforced. TTPO observes an asymmetry: rollouts that disagree with the pseudo-label are usually wrong, independent of whether the vote itself is correct. Exploiting that asymmetry lets models adapt at inference time without access to ground truth while avoiding the typical collapse modes of naive self-distillation or RL.

Key Findings
  • Asymmetric objective: TTPO splits updates into two branches — On-Policy Self-Distillation (OPSD) that distills agreeing rollouts, and Grouped RL that penalizes disagreeing rollouts. Token-level selection further focuses distillation on unconverged positions and restricts RL penalties to confident errors.
  • Robust under noisy pseudo-labels: Both branches remain well-grounded even when majority votes are frequently incorrect, because agreeing rollouts are used as positive teachers while disagreeing rollouts act as negative signals rather than corrupting the whole teacher signal.
  • Empirical gains: Without labels, TTPO matches label-supervised OPSD on five competition-level benchmarks, lifts Qwen3-1.7B from 38.0% to 45.2% under TTT, and produces large accuracy gains (+25.2% to +36.4%) in the "without thinking" setting. The method also shows strong cross-task generalization.
  • Practical mechanics: Majority-vote routing tightens self-supervision as the policy improves; token-level selection and grouped RL make the updates stable and targeted rather than globally destructive.
Who it's for and trade-offs

Great fit if you need to improve LLM reasoning performance at inference time without any labeled data, especially for deterministic/verifiable tasks like math and science reasoning where pseudo-label consensus is meaningful. It is appropriate when you can afford some extra inference rollouts and in-place parameter updates. Look elsewhere if your deployment forbids any model updates at test time, if tasks are highly open-ended (no shared canonical answers), or if tight latency budgets disallow the additional rollouts TTPO needs for reliable consensus.

Information

  • Websitearxiv.org
  • OrganizationsAffiliation: Zhejiang University, Affiliation: Alibaba Group{waz,zhengxilu,syl}@zju.edu.cn    [email protected]
  • AuthorsAozhe Wang, Zhengxi Lu, Jianze Wang, Shangke Lv, Ying Liu, Weiming Lu, Jun Xiao, Yueting Zhuang, Hua Yang, Qianglong Chen
  • Published date2026/08/27

More Items

Studies looping shared transformer layers in Mixture-of-Experts models under matched budgets and proposes SMELT: loop the middle half twice while matching per-token FLOPs, non-embedding parameters, and KV cache. Shows 6.8–18.0% training-FLOPs savings on the compute-optimal frontier, stronger downstream gains on code and long-context tasks.

Turns sparse per-student records into individualized simulators that both reproduce a student’s responses and update them under tutor guidance using pooled LLM pretraining followed by per-student specialization; releases StudentSimEval and reference simulators across chess, L2 writing, and math.

Analyzes on-policy distillation for LLM fine-tuning, shows teacher token-level supervision is often noisy and not the main driver of gains, and introduces OPSA, a supervision-free, entropy-adaptive method that suppresses low-probability tokens to improve downstream accuracy.