过去并未过去:记忆增强的动态奖励塑形
阅读原文· arxiv.org针对大语言模型强化学习中采样多样性降低、策略重复生成相似错误的问题,本文提出MEDS(记忆增强动态奖励塑形)框架。该方法通过存储中间模型表示捕捉历史rollout特征,利用密度聚类识别高频错误模式,并对重复错误施加更重惩罚,从而在鼓励探索的同时减少重复犯错。在五个数据集和三个基础模型上的实验表明,MEDS较基线平均性能显著提升,pass@1和pass@128最高分别提升4.13和4.37点,有效增强了采样多样性。
Despite the success of reinforcement learning for large language models, a common failure mode is reduced sampling diversity, where the policy repeatedly generates similar erroneous behaviors. Classical entropy regularization encourages randomness under the current policy, but does not explicitly discourage recurrent failure patterns across rollouts. We propose MEDS, a Memory-Enhanced Dynamic reward Shaping framework that incorporates historical behavioral signals into reward design. By storing and leveraging intermediate model representations, we capture features of past rollouts and use density-based clustering to identify frequently recurring error patterns. Rollouts assigned to more prevalent error clusters are penalized more heavily, encouraging broader exploration while reducing repeated mistakes. Across five datasets and three base models, MEDS consistently improves average performance over existing baselines, achieving gains of up to 4.13 pass@1 points and 4.37 pass@128 points. Additional analyses using both LLM-based annotations and quantitative diversity metrics show that MEDS increases behavioral diversity during sampling.