π€ AI Summary
This work addresses the limitations of traditional discriminative value models in reinforcement learning with large language models, which often struggle with credit assignment due to restricted representational capacity. To overcome this, the authors propose the Generative Actor-Critic (GenAC) framework, replacing single-step scalar value prediction with a generative critic endowed with chain-of-thought reasoning capabilities. A context-conditioning mechanism is introduced to align the criticβs evaluations with the current policy. By reframing value modeling as a generative, sequential reasoning process rather than a discriminative prediction task, GenAC substantially enhances the expressiveness of the value function, improves ranking reliability, and boosts out-of-distribution generalization. Empirical results demonstrate consistent and significant improvements over both value-based and value-free baselines across downstream reinforcement learning tasks.
π Abstract
Credit assignment is a central challenge in reinforcement learning (RL). Classical actor-critic methods address this challenge through fine-grained advantage estimation based on a learned value function. However, learned value models are often avoided in modern large language model (LLM) RL because conventional discriminative critics are difficult to train reliably. We revisit value modeling and argue that this difficulty is partly due to limited expressiveness. In particular, representation complexity theory suggests that value functions can be hard to approximate under the one-shot prediction paradigm used by existing value models, and our scaling experiments show that such critics do not improve reliably with scale. Motivated by this observation, we propose Generative Actor-Critic (GenAC), which replaces one-shot scalar value prediction with a generative critic that performs chain-of-thought reasoning before producing a value estimate. We further introduce In-Context Conditioning, which helps the critic remain calibrated to the current actor throughout training. GenAC improves value approximation, ranking reliability, and out-of-distribution generalization, and these gains translate into stronger downstream RL performance than both value-based and value-free baselines. Overall, our results suggest that stronger value modeling is a promising direction for improving credit assignment in LLM reinforcement learning.