VPO: Strengthening Reasoning Performance Through Diversity Learning
Key point
VPO is an RL algorithm that trains models to generate diverse solutions, improving inference-time search performance.
Details
Existing LLM post-training relies on optimizing a single scalar reward, which causes the model's response distribution to become monotonous. This degrades Inference-time search performance, which requires exploring diverse paths during the reasoning stage.
Vector Policy Optimization (VPO) is a new reinforcement learning (RL) algorithm that treats rewards as vector values, training the model to generate diverse solutions.
Key features are as follows:
- GRPO Replaceable: A drop-in method that can replace GRPO's Advantage estimator.
- Ensuring Diversity: Induces the model to generate solutions with different trade-offs within the vector reward space.
- Search Performance Advantage: Outperforms existing scalar RL models on pass@k and best@k metrics, with the performance gap widening as the search budget increases.
- Overcoming Limitations: Can solve Evolutionary search problems that GRPO models fail to solve.
This summary was generated automatically by AI. Check the original for the author's claims and context. Copyright belongs to the original author.
Our guide explains how the AI works. Report summary errors, attribution issues, or removal requests via Contact.