Weekly archive · Qwen Math on MATH

Weekly research report

Two-circuit pipeline-parallel RL at longer horizons

This week I finished the long-run check that had been overdue. The exact-k PRF fast circuit completed the full horizon without a collapse, kept its fixed activation budget, and held the MATH validation gain. The useful question now is why fresh random compression survives where a reused PowerSGD Q direction can fail.

full long run completed fixed compression budget held no reward or entropy cliff matched dense run still pending

This week, in one line

Fresh PRF masks can carry the compressed fast circuit through a long RL run while learning, but a reused Q subspace can turn the same training loop into a directional failure. The evidence points to persistent direction, not compression magnitude alone.

Important correction. Dense actor KL also rises during successful RL. A rising reference KL is normal policy movement. The failure signal is the way KL growth couples to clipping pressure, falling entropy, a widening sampler-to-trainer gap, and finally a score cliff.

Start with the comparison

The two Q runs use the same directional codec. Their main difference is how long the anchor state is allowed to remain stale. The PRF run redraws the compression direction instead.

collapsed

PowerSGD Q, slower refresh

The same preferred subspace remains active long enough for the mismatch to become a training direction.

reuse Q aligned codec error clipping pressure rises entropy and score fall
survived

PowerSGD Q, faster refresh

This run still becomes rough, but the next anchor arrives before the transient locks into a collapse.

reuse Q briefly drift starts refresh changes correction score recovers
long run held

Fresh exact-k PRF mask

The instantaneous error is large, but its direction changes with each mask and does not become a second objective.

fresh mask new error direction first-order cancellation learning continues

Why the failure can look sudden

GRPO is a repeated feedback loop. The dense sampler generates answers, verifiable rewards rank them, and the trainer changes token probabilities so better answers become more likely. Those weights return to the sampler and the loop repeats. Compression sits inside the trainer view of that loop.

dense sampler answers and rewards relative advantages clipped policy update new sampler weights

PPO clipping limits one update relative to the stored old policy. It does not limit the mismatch that can accumulate across many outer RL steps between the dense sampler and the compressed trainer.

Upper clipping

good token probability rises too far more promotion is capped

Lower clipping

bad token probability falls too far more suppression is capped

Only the combined standard clip fraction was logged in these runs. The loss also contains a separate dual-clip guard for extreme negative-advantage cases, but there is no W&B series for that branch here.

The slow Q bias can therefore accumulate while the sampled reward still looks healthy. Once enough token ratios cross the clipping bounds together, more of the surrogate becomes flat. The useful update signal shrinks, the compressed actor becomes sharply concentrated, and the next dense rollout can become narrower. That weakens the within-group reward contrast that GRPO needs. The visible score drop is sudden even though the directional drift was building gradually.

Clipping is a guardrail, not a long-horizon controller. The faster Q refresh interrupts the correlation, so clipping can spike and recover. PRF changes the error direction before it can accumulate. This is why clip fraction is useful as a coupled warning signal, but not as a universal collapse threshold.

How the metrics expose the transition

  • critic/score/mean: mean verifiable reward of sampled answers, not a learned critic value.
  • actor/entropy: sharpness of the compressed trainer view. PRF is not directly comparable with Q here.
  • actor/kl_loss: movement from the reference policy inside the trainer view. Dense KL rises too.
  • rollout_corr/kl: gap between dense sampling and compressed training. Trend matters more than level.
  • actor/pg_clipfrac: share of tokens whose PPO surrogate is clipped. The run logger combines the ordinary upper and lower branches.
  • Coupled reading: the collapse is the joint turn in score, entropy, mismatch, and clipping, not one magic value.

The same three runs, with every logged point

The shaded region marks the failed Q run from the first clear score cliff through the entropy trough. Curves are reconstructed from W&B without smoothing.

PowerSGD Q, slower refresh PowerSGD Q, faster refresh fresh PRF mask dense, KL inset only

Mean verifiable score

The slower Q run learns first, then loses answer quality abruptly. The faster refresh and PRF runs hold.

Actor entropy

The failed Q trainer view becomes extremely sharp. PRF stays diffuse because its mask changes the view.

Actor KL loss

Dense KL also rises. The failed Q run matters because its rise couples to the other failure signals.

Sampler-to-trainer gap

PRF starts with a large but stationary wedge. The failed Q wedge accelerates with the score cliff.

PPO clipped-token fraction

Clipping pressure rises before the failed Q score cliff. The logger combines upper and lower clipping.

Read the clip plot with the RL loop

coherent Q drift more ratios clipped less useful gradient narrower samples

A spike is not enough by itself. The faster Q run also sees clipping pressure, but refresh breaks the persistent direction and the score survives.

Large mismatch alone does not cause collapse. PRF carries a large gap from the beginning but keeps it nearly stationary. The slower-refresh Q run turns a smaller initial gap into a coherent, time-correlated drift. The faster-refresh Q run takes a rough path and recovers. Open the real W&B runs for slower Q, faster Q, PRF, and dense.

Why Q is directional and PRF is not, at first order

PowerSGD does not make the whole model gradient low rank. It inserts a projector at each compressed pipeline boundary. While the basis is fixed, both the activation and the boundary gradient pass through the same preferred subspace.

PowerSGD-like Q projection

A AQ AQQT
à = A PQ,   PQ = QQT
∂L/∂A = (∂L/∂Ã) PQ   while Q is fixed
same Q same discarded directions errors can add

What is directional: unless the activation already lies in the Q subspace, the same residual is repeatedly removed. Reusing Q makes this error coherent across RL updates.

Exact-k PRF mask

h mt/q (mt/q)⊙h
t = (mt/q) ⊙ ht
Em[h̃t | ht] = ht
fresh mask different discarded directions errors can cancel

What is mean-preserved: each boundary coordinate is preserved over fresh masks, so the first-order activation error has no persistent direction. The full model output and parameter gradient are not guaranteed to be unbiased because the network is nonlinear.

The practical distinction is bias times coherence. A reused Q can keep removing and reinforcing the same directions. PRF injects larger instantaneous noise, but redraws the direction. Its first-order errors can cancel instead of adding coherently. This is a random-walk explanation, not a claim that neural updates are independent.

How this becomes train-inference mismatch

The dense sampler chooses tokens with full weights, but the trainer scores them through a compressed pipeline. With Q, the trainer repeatedly sees the policy through the same lens and can move the weights toward that lens. The sampling policy and training policy then separate in a directed way. With PRF, each trainer view is noisy but the noise distribution is stationary. The dense sampler can still sharpen while the codec-side view stays diffuse. This is why PRF entropy can remain high without meaning that the actual sampled policy failed to learn.

What the weight-space report adds

The weight evidence does not support calling RLVR movement simply messy, high rank, or sparse. The total movement is small in both Q runs, yet the failing run shifts its movement toward earlier blocks while the surviving run stays more centered and PRF remains later-heavy. The difference is spatial direction, not simply how much the weights moved.

The long-run PRF result

The modified exact-k PRF run keeps a fixed sparse payload, redraws the mask each step, and uses constant inverted rescaling. It completed the full horizon without a reward or entropy cliff. Validation improved early and stayed inside the learned band.

MATH validation through the long run

The final checkpoint comes from the training log because W&B omitted the last row.

Capability improves and holds. The late movement stays inside the noise expected from a single-sample validation set.
Reserved: matched dense parity This panel will hold the in-domain MATH curve and the out-of-domain benchmark comparison when the matched dense checkpoint series is available.
What this result does not say. The long run uses a different training recipe from the older comparison runs. It validates long-horizon survival of the exact-k PRF design. It does not yet prove matched parity with dense training or guarantee the same result across seeds.

What changes next

There is no single collapse band that works across codecs. Dense KL can rise and succeed. PRF can carry a large stationary sampler gap and succeed. The controller has to watch a multivariate health envelope and react to a coordinated turn.

  • Keep exact-k PRF as the safe baseline. The codec search found that more structured variants reduced the unusual codec entropy but increased reference drift.
  • Add a dense-view control plane. Periodically measure actor-to-reference KL without compression and adjust the KL coefficient against that clean signal.
  • Use coupled tripwires. React when dense-view drift, score, entropy, mismatch slope, and clipping pressure turn together. Do not chase one codec-specific level.
  • Reduce variance without restoring direction. The next codec should preserve the no-persistent-direction property while spreading the byte budget more evenly.
  • Run the missing matched evidence. Dense and out-of-domain curves are still needed before claiming parity.
Current direction. Randomness is the mechanism that prevents a fixed compression view from becoming a second objective. The forward path is to keep that lack of directional bias, lower its variance, and control true dense-policy drift directly.

Communication-efficient RLVR weekly report · real W&B trajectories · no smoothing