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

Trains long-horizon web researchers by combining agentic mid-training, synthetic trajectories, and reinforcement learning. A 30.5B-parameter model activates 3.3B parameters per token and learns to plan, search, and synthesize evidence.

Explains how base-model behavior and GRPO normalization distort claims about reasoning gains. Introduces Dr. GRPO to reduce length bias, with a 7B training recipe reaching 43.3% on AIME 2024.

Improves language-model problem solving by generating candidate answers, retaining verified successes, and fine-tuning on them. Repeats this loop with an expectation-maximization interpretation instead of collecting more human-written solutions.