Imagine you're filling in a crossword puzzle, but instead of working one clue at a time, you can see tentative letters for every square simultaneously. Some letters you're confident about — the 'T' in a three-letter word ending in 'HE' — and others are shaky guesses. The question isn't whether to fill in squares in parallel (you can), but WHICH squares to commit to ink right now versus leaving in pencil for another pass. Commit too eagerly and you lock in errors that cascade. Too cautiously and you waste time re-checking letters you already had right. That's the core problem with masked diffusion language models like LLaDA and Dream. These models generate all tokens simultaneously by iteratively 'unmasking' predictions, but the default strategy — unmask at a fixed rate per step — is blunt. Learned unmasking policies add a small auxiliary network that decides which predictions are safe to commit. The literature has proposed several, but they differ in supervision signals, input features, and model size, making it impossible to tell which design choices actually matter. This paper runs the controlled experiment the field needed. The authors decompose learned unmasking into four independent axes: what the teacher labels as committable (the supervision target), what states the student policy trains on (the data distribution), what the policy observes (confidence scores, hidden states, or both), and how much compute reduction is needed to justify the policy's overhead. They test each axis in isolation. The headline findings are sharp. First, the popular 'match-the-final-output' labeling strategy is too permissive — it tells the policy to commit predictions that happen to match the final answer but would have changed on a different trajectory, leading to error accumulation at low step counts. Second, hidden states alone are weak predictors of commitment quality, but they complement token-level confidence meaningfully. Third, the overhead break-even point is surprisingly low: roughly a 10% reduction in number of function evaluations (NFE) suffices to pay for policies up to 7.5M parameters. Armed with these findings, the authors propose 'trajectory-contraction targets' — a supervision signal that only labels a prediction as committable if it stays correct across ALL remaining steps of a reference denoising trajectory, not just at the final output. This is a stricter, more causally honest label. The resulting policy uses both confidence and hidden-state features, weighs in at only 856K parameters, and trains on trajectories sampled from the base model itself. On LLaDA-8B-Instruct, this policy achieves the best accuracy-vs-NFE frontier on GSM8K (grade-school math), is competitive at moderate NFE targets on MATH-500 and HumanEval (code generation), and — crucially — transfers to Dream-7B-Instruct without retraining the policy from scratch. The transfer result is important because it suggests the policy is learning something about the geometry of the denoising process, not just memorizing one model's quirks. The integrity picture is mixed but honest. The authors compare against multiple recent learned-unmasking baselines (including confidence-based and hidden-state-based variants) and report results where their method ties or trails, not just where it wins. MATH-500 and HumanEval results are 'competitive' rather than dominant, which they state plainly. The benchmarks (GSM8K, MATH-500, HumanEval) are community standards, not custom tasks. However, there is no code release mentioned, no pre-registration, and no independent replication — all standard for a preprint but worth tracking. The bigger picture: diffusion language models are an active alternative to autoregressive generation, and their viability hinges on whether parallel decoding can match sequential quality at lower latency. This paper doesn't answer that question, but it sharpens the tooling. If learned unmasking policies can reliably cut 10-30% of steps without accuracy loss, diffusion LMs become substantially more competitive on inference cost. The next test is whether this scales — both to larger base models and to harder tasks where error accumulation at low NFE would be catastrophic.
