Vector Policy Optimization: Training for Diversity to Improve Test-Time Search

Bahlous-Boldi, Puri, Shenfeld, Kumar, Damani, Risi, Khattab, Hong, Agrawal · MIT Improbable AI Lab, MIT-IBM, Sakana AI · arXiv preprint, May 2026 (not peer reviewed)
arXiv 2605.22817 Interactive page

Same RL, two training units

Toggle the recipe. VPO only changes the advantage computation and what counts as a sample.

Search needs a varied pool

Best-of-N and evolutionary search can only exploit differences between candidates.Illustrative

Reward vectors and the Pareto frontier

Each dot is a candidate scored on two components (e.g. gold vs diamond). A scalarization weight picks one dot.Illustrative points

Scoring a set of three answers

Set score = average over K shared weight draws of the best answer's w·r. Step through it.Illustrative numbers

Why both pieces are needed

Paper's ablation logic as described on the episode, drawn as a 2×2.

Reported results

Numbers quoted from the episode. One training run per table cell, no confidence intervals.Reported

Reward-space diversity

Mean pairwise L1 distance between candidates' reward vectors. Reworded copies score ~0.Toy vectors

When VPO has nothing to diversify

As reward components become collinear, the Pareto front shrinks toward one point. Paper: on-policy correlation 0.95 on UltraFeedback, VPO 0.776 vs GRPO-style 0.779 best@50.Synthetic cloud

Metrics and caveats

References

  1. Vector Policy Optimization (Bahlous-Boldi et al., 2026)
  2. DeepSeekMath (Shao et al., 2024) — GRPO
  3. Yue et al., 2025 — RL vs base model at large k
  4. Large Language Monkeys (Brown et al., 2024)
  5. Kirk et al., 2024 — RLHF and diversity
  6. Roijers et al., 2013 — multi-objective survey
  7. Hayes et al., 2022 — practical MORL guide
  8. Yang, Sun, Narasimhan, 2019
  9. Rewarded Soups (Ramé et al., 2023)
  10. Pass@k Policy Optimization (Walder, Karkhanis, 2025)
  11. Inference-Aware Fine-Tuning for Best-of-N (Chow et al., 2024)
  12. AlphaEvolve (Novikov et al., 2025)
  13. FunSearch (Romera-Paredes et al., 2024)
  14. ShinkaEvolve (Lange et al., 2025)
  15. MAP-Elites / Quality Diversity
  16. Reaching Beyond the Mode — Multi-RLVR (Puri et al., 2026)
  17. KL-Regularized RL is Designed to Mode Collapse (GX-Chen et al., 2025)
  18. Max@k optimisation (Bagirov et al., 2025)
  19. MaxRL (Tajwar et al., 2026)
  20. GDPO (Liu et al., 2026)
  21. Lexicase selection / DALex
  22. BOND / BoNBoN / InfAlign
  23. OpenEvolve and relatives
  24. Rewards-in-Context (Yang et al., 2024)
  25. Rewarded Soups / Directional Preference Alignment
  26. DARLING and DAPO clip-higher