JA EN
LearnInference & Serving
·★ MEMBER·PAPER·11 min read

TTPO Explained: Training a Model Mid-Exam, With No Answer Key

Majority-vote pseudo-labels are wrong on roughly 85% of competition math prompts. So why does training still work? A ground-up walkthrough of TTPO's asymmetric design: distill the rollouts that agree, penalize the ones that disagree.

ModalitytextTaskinference

TTPO: Test-Time Policy Optimization

Primary source — what this article is built on

undefined2026-08-27undefined2026-08-31same month

TTPO: Test-Time Policy OptimizationAozhe Wang, Zhengxi Lu, Jianze Wang et al. · 2026-08-27 · v1arXiv:2608.27448Paper page·PDF
undefined

Recent prominent post-training methods, such as Reinforcement Learning (RL) and On-Policy Self-Distillation (OPSD), have driven rapid progress in mathematical reasoning for large language models, yet their reliance on ground-truth labels precludes test-time training (TTT). Replacing ground truth with majority-vote pseudo-labels is a natural alternative, yet it is fragile: an incorrect vote corrupts the teacher and misleads every token. We observe that this failure mode is asymmetric: rollouts that disagree with the pseudo-label are typically wrong regardless of whether the vote itself is correct. Building on this observation, we propose Test-Time Policy Optimization (TTPO), an asymmetric objective that distills agreeing rollouts via OPSD and penalizes disagreeing rollouts with Grouped RL. Token-level selection further refines both branches: distillation down-weights already-converged positions, while RL penalizes only confident errors. Both updates remain well-grounded even under frequent pseudo-label errors, and majority-vote routing yields tighter self-supervision as the model improves. Without any labels, TTPO matches label-supervised OPSD on five competition-level benchmarks, raises Qwen3-1.7B from 38.0% to 45.2% in TTT, yields +25.2% to +36.4% without thinking, and shows strong cross-task generalization.


Getting better mid-exam, with no answer key

Picture yourself halfway through a mock exam. There is time left on the clock, but nobody will hand you the marking scheme. You can still do something useful: solve the same problem several different ways, treat whichever answer came up most often as "probably the right one," and use it to audit your own reasoning.

Machine learning calls this test-time training (TTT) — improving on the very problems you are being asked to solve, without ever seeing the labels. It throws away the usual assumption that training data and deployment data are separate things, and uses the test set itself as the training set.

TTPO (Test-Time Policy Optimization) brings that setting to mathematical reasoning in large language models. The claim is blunt: without using a single ground-truth label, it matches or beats methods that do use them (§1).

The setup: both mainstream post-training recipes demand labels

The first family is reinforcement learning from verifiable rewards (RLVR). Let the model solve the problem, give it a reward of 1 if the final answer matches the ground truth and 0 otherwise. But that reward is a single scalar per rollout, broadcast uniformly across every token, so it never says which particular step was the good one. The paper calls it "coarse" (§1).

The second is on-policy self-distillation (OPSD), and the idea is stranger. You build a teacher by conditioning the same model on the answer. The teacher, knowing where the solution lands, re-reads the student's own text token by token and produces a distribution saying "here is what I would have written." Where RL gave one number per rollout, this gives a supervision signal at every token — far denser (§2.2).

The problem is that both need the answer: RLVR to verify, OPSD as privileged context for the teacher. In TTT, where labels never arrive, neither transfers (§1).

What breaks when the majority vote becomes the teacher

If there is no label, have the model make one. Sample several rollouts per problem and take the most common answer as a pseudo-label — a majority vote. TTRL used exactly this as a reward and improved reasoning with no labels at all (§2.1).

So why not drop the pseudo-label into OPSD wherever the ground truth used to go? Here the paper lands a sharp line: a corrupted reward misleads once per trajectory; a corrupted teacher misleads at every token (§1). Dense supervision is powerful when it is right, and equally dense in its damage when it is wrong.

A note before we go on: a teacher's "distribution" is peaked when the model is confident and flat when it is unsure. That peakedness becomes the heart of TTPO later, so get a feel for it with your hands first.

FIG 1Same scores, different shape — a peaked distribution is a confident token, a flat one is an uncertain token. TTPO sorts tokens by exactly this (entropy)

The paper's observation: the errors are asymmetric

Then come the measurements. Running test-time training on Qwen3-1.7B over AIME 2026, the pseudo-label is wrong for roughly 85% of prompts (§3.2, Figure 1a). A teacher that is wrong four times out of five sounds unusable.

But the same measurement yields a second fact. Even when the pseudo-label is wrong, about 79% of the rollouts that disagreed with it produce an answer that is neither the pseudo-label nor the ground truth (§3.2).

That is the crux. Penalizing a disagreeing rollout never touches the content of the pseudo-label. It uses one bit only: this rollout is not in the majority cluster. So the penalty is usually correct whether or not the vote was. Distilling toward the pseudo-label has no such tolerance — it consumes the answer itself, so the error flows straight into the teacher distribution.

Penalties do not depend on the pseudo-label; distillation does. That asymmetry is TTPO's entire design principle. The paper ties it to negative learning from noisy labels, where saying "this is not a dog" stays reliable long after "this is a cat" has stopped being (§1).

Put differently: one signal, the majority vote, is decomposed into a trustworthy use and an untrustworthy one. Prior methods applied that signal uniformly to every rollout, so the blast radius of a bad vote scaled with the number of rollouts. TTPO confines the branch where errors can enter to the agreeing rollouts only — and, as the next section shows, arranges for the error to be defanged even there. What follows is the actual loss, and then the token-level selection layered on top of it.

The mechanism: distill the agreers, penalize the disagreers

For a problem xx, sample KK rollouts and extract their final answers aka_k. Cluster answers by mathematical equivalence, take the largest cluster as the pseudo-label a^\hat{a} (§3.1), then split the rollouts in two.

P={k:aka^},N={k:ak≢a^}\mathcal{P}=\{k: a_k\equiv\hat{a}\},\qquad \mathcal{N}=\{k: a_k\not\equiv\hat{a}\}
(1)

P\mathcal{P} is the set of rollouts that agreed with the vote; N\mathcal{N} the set that disagreed.

Put in words: the KK rollouts are sorted into two piles on a single bit — did this one land on the majority answer, or not. Nothing about the reasoning inside a rollout, or where it went off the rails, enters the split.

is the number of valid response tokens, the teacher's distribution, the student's, a per-token weight described below. Stripped of notation it is a rule which says: measure the gap between the probabilities the teacher and the student assigned, token by token, scale each gap by how much that token matters, and average

What's behind this

§

Members-only from here

371 walkthroughs, 26 textbook chapters, 48 student units and 6 close readings — all included for $4.99/mo, with three new explainers every day. Cancel any time; access runs to the end of the period.

Already a member? Sign in to keep reading

References

  1. Aozhe Wang, Zhengxi Lu, Jianze Wang, Shangke Lv et al.. (2026-08-27) TTPO: Test-Time Policy Optimization. arXiv:2608.27448Paper page·PDF

This article is written from the source paper above. Where they differ, the original is authoritative.

Comments

Sign in to comment