Mechanism
Eliminates the value network in PPO and replaces the multi-sample Monte Carlo baseline of RLOO with the reward of a single deterministic greedy-decoded trajectory :
Practical Notes
Requires only 2 forward generations per prompt (1 sampled + 1 greedy) and 0 critic memory, though higher variance than RLOO/GRPO on hard reasoning tasks where greedy decoding consistently scores 0.
References
- ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Models ↗
All 7 authors
Ziniu Li, Tian Xu, Yushun Zhang, Zhihang Lin, Yang Yu, Ruoyu Sun, Zhi-Quan Luo