Cast a Wider Net: Coordinated Pass@K Policy Optimization for Code Reasoning
Abstract
Repeated sampling with a verifier is the standard way to allocate test-time compute for code generation, with pass@ as the canonical metric. Yet the standard policy class draws independent samples from a single answer distribution, so attempts often collapse onto near-duplicate reasoning paths and waste the budget on redundant rollouts. This failure is costly in competitive programming, where many problems admit multiple distinct algorithmic strategies and pass@ requires only one correct attempt. We propose Coordinated Pass@ Policy Optimization (CPPO), which turns pass@ generation into joint exploration over strategies: a planner emits a tuple of alternative high-level methods, and a shared solver attempts one solution per method. CPPO trains this joint policy with a multiplicative planner reward, , assigning credit only to valid strategy tuples that lead to verifier-confirmed pass@ success. Across APPS, CodeContests, and LiveCodeBench-v6, CPPO improves pass@ over direct sampling, planning baselines, planner-only SFT, and pass@-oriented RL under the same solver-attempt budget, with statistically significant gains on six of nine model--benchmark cells. The largest single gain is on Qwen3.5-9B LiveCodeBench-v6 over the strongest baseline, PKPO (; paired bootstrap, ).
Cite
@article{arxiv.2605.27000,
title = {Cast a Wider Net: Coordinated Pass@K Policy Optimization for Code Reasoning},
author = {Yilong Li and Suman Banerjee and Tong Che},
journal= {arXiv preprint arXiv:2605.27000},
year = {2026}
}
Comments
Code reasoning; pass@K optimization; coordinated planning; verifiable rewards; strategy diversity