Hugging Face Daily PapersYizhuo Li, Jianhao Yan, Yun Luo1 min readpaperadvanced
Rethinking Critic Learning in PPO: Understanding and Mitigating Value Flattening
Summary
PPO critics in reinforcement learning for LLMs suffer from "Value Flattening," where predicted state values are too flat compared to actual values. This paper identifies the causes as an implicit variance penalty and redundant updates, and proposes SParse Proximal Policy Optimization (SP3O) to mitigate it by supervising only a few well-separated states.
- PPO critics exhibit "Value Flattening," where predicted state values are overly flat despite sharp changes in true state values.
- This failure mode is more pronounced in larger state spaces, impacting LLM reinforcement learning.
- Value Flattening is linked to an implicit variance penalty in the critic loss and redundant updates from correlated states.
- SP3O mitigates Value Flattening by applying the value loss to only a few well-separated states per response.
ML engineers working on reinforcement learning for large language models should care, as this paper identifies and addresses a critical, overlooked failure mode in PPO critic training.
8/10
