Effect of alternative OPD objectives on pass@k coverage and RL interaction
Investigate whether alternative on-policy distillation objectives—including reverse KL over the full next-token distribution or a top-$K$ truncation, forward KL, and Jensen–Shannon divergence—expand pass@$k$ coverage more effectively than the token-level reverse-KL objective and determine how the resulting cold starts interact with subsequent reinforcement learning.
References
Our analysis does not independently vary teacher distributional alignment and retained uncertainty, leaving their individual contributions and interaction unresolved.
A natural direction for future work is to investigate alternative OPD objectives that may yield a different cold-start for RL, including reverse KL evaluated over the full next-token distribution (or top-$K$), as well as forward KL or Jensen-Shannon divergence (JSD) objectives, which differ from reverse KL in their mode-coverage behavior and effect on generation diversity~\citep{eopd}. Whether these objectives expand pass@$k$ coverage more effectively, and how the resulting cold-start interacts with RL, remain open questions.
Interactions between axes, their dependence on the task, and their side effects therefore remain largely unknown, and, in our opinion, deserve systematic evaluation.
Finally, our coverage results are based on a finite number of samples. Although we use additional sampling and manual inspection to check apparently new solutions, we cannot rule out that larger sampling budgets or different decoding settings would uncover additional correct responses from the initial model.
We leave a systematic study of the effect of $K$ and the resulting top-$K$ approximation error to future work.
As noted earlier, we leave a systematic study of alternative divergence objectives to future work.
For instance, an interesting open question would be to understand if the diversity improvements of our teacher selection scheme can be used to create a better warm start for RL post-OPD.