EliteReplay: Preserving Diverse Solutions in RL Post-Training via Quality-Diversity Search· EliteReplay:RL 后训练保持解多样性
Reinforcement learning has become a common post-training technique to improve the accuracy of LLMs on verifiable tasks. However, such training can narrow the set of solutions the model can produce. Several approaches have attempted to address this behavior-sharpening effect, but they often optimize metrics that do not reflect the diversity of correct solutions. Ideally, we want a model that can produce multiple good solutions, not only a single reliable one. We propose EliteReplay, a post-training recipe that pairs GRPO with an "elite" archive built by quality-diversity search. Before training, we use the frozen base model to collect solutions that span distinct approaches; during training, GRPO is combined with a replay loss on this archive. The result is a post-trained model that improve
RL 后训练方法,保持解的多样性
- 核心方法
- 使用质量多样性搜索构建“精英”解档案,并结合 GRPO 与重放缓冲区损失进行训练
- 适合谁读
- 研究者 / 工程师
- 要解决的问题
- 强化学习后训练过程中,模型解的多样性减少问题
- 关键实验
- 在多个需要多样性解的任务上进行了实验验证,具体数据未提供
- 主要贡献
- 提出 EliteReplay,有效保持了模型解的多样性,同时提高了任务性能
- 意义与局限
- 有助于避免模型过度优化单一解,拓宽了解的应用场景。局限在于增加了解决方案收集和管理的复杂性。