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 ygreedy=arg⁡max⁡yπθ(y∣x)y^{\text{greedy}} = \arg\max_y \pi_\theta(y\mid x): A^ReMax(x,y)=R(x,y)−R(x,ygreedy)\hat{A}_{\text{ReMax}}(x, y) = R(x, y) - R(x, y^{\text{greedy}})

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

  1. ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Models ↗

    Ziniu Li, Tian Xu, Yushun Zhang, Zhihang Lin, Yang Yu, Ruoyu Sun et al. · 2023

    All 7 authors

    Ziniu Li, Tian Xu, Yushun Zhang, Zhihang Lin, Yang Yu, Ruoyu Sun, Zhi-Quan Luo