PSPO: Prompt-Level Prioritization and Experience-Weighted Smoothing for Efficient Policy Optimization

Reinforcement Fine-tuning (RFT) methods such as Group Relative Policy Optimization (GRPO) have demonstrated strong capabilities in aligning Large Language Models with human preferences. However, these approaches often suffer from limited data efficiency, necessitating extensive on-policy rollouts to maintain competitive performance. We propose PSPO (Prompt-Level Prioritization and Experience-Weighted Smoothing for Efficient Policy Optimization), a lightweight yet effective enhancement to GRPO that improves training stability and sample efficiency through two complementary techniques. First, we introduce an experience-weighted reward smoothing mechanism, which uses exponential moving averages to track group-level reward statistics for each prompt. This enables more stable advantage estimation across training steps without storing entire trajectories, allowing the model to capture historical reward trends in a lightweight and memory-efficient manner. Second, we adopt a prompt-level prioritized sampling strategy, which is an online data selection method inspired by prioritized experience replay. It dynamically emphasizes higher-impact prompts based on their relative advantages, thereby improving data efficiency. Experiments on multiple mathematical reasoning benchmarks and models show that PSPO achieves comparable or better accuracy than GRPO, while significantly accelerating convergence, and maintaining low computational and memory overhead.

Authors

• Xinxin Zhu Ant Group
• Ying He Nanyang Technological University
• Haowen Hou Guangdong Laboratory of Artificial Intelligence and Digital Economy (SZ), Shenzhen, China
• Ruichong Zhang Tsinghua University, China
• Nianbo Zeng College of Computer Science and Software Engineering, Shenzhen University, China Guangdong Laboratory of Artificial Intelligence and Digital Economy (SZ), Shenzhen, China
• Yulin Peng College of Computer Science and Software Engineering, Shenzhen University, China
• Jiongfeng Fang Guangdong Laboratory of Artificial Intelligence and Digital Economy (SZ), Shenzhen, China
• F. Richard Yu Carleton University

Keywords

No keywords are indexed for this paper.