Breaking the Reinforcement Learning Reward Bottleneck: A New Approach
Reinforcement learning hits a snag with reward signals. Introducing Cross-Model Entropy, a game-changing method for training large language models.
reinforcement learning, the reward signal has long been a stumbling block for effectively post-training large language models. Traditional methods rely either on verifiable ground-truths, limiting their application to well-defined domains like mathematics and code execution, or on human preference labels, which come with their own set of challenges, including high costs and vulnerability to reward hacking.
A New Contender: Cross-Model Entropy
Enter Cross-Model Entropy (CME), a fresh approach that seeks to dismantle these barriers. CME uses the mean log-likelihood of a generator's response, as evaluated by a separate, independent verifier model. This method offers a continuous, training-free reward signal, rooted in the idea that responses deemed unsurprising by an independent verifier are likely accurate or of high quality. The independence of the verifier from the generator ensures that this feedback loop can't be gamed through self-consistency, a common pitfall in self-referential methods.
Application and Results
Integrating CME into the Generalized Reinforcement Policy Optimization (GRPO) framework, without altering the training loop, extends the application of label-free reinforcement learning to more complex environments, such as open-ended instruction following. This is particularly critical in scenarios where self-referential signals falter.
In practical terms, testing on UltraFeedback prompts through AlpacaEval 2.0 revealed that CME-based rewards outshined the untrained base in head-to-head LLM-as-Judge comparisons. This was consistent across diverse model families, including Qwen, Llama, Gemma, and OLMo, as well as different training regimes, such as pretrained, SFT, and instruction-tuned, with tie-adjusted win rates spanning from 52.5% to 71.4%. Simply put, the numbers speak for themselves.
Why It Matters
Let's apply some rigor here. This development could mark a significant step forward for reinforcement learning, particularly in complex, less structured environments. By providing a reliable, non-gamable reward signal, CME might just be the breakthrough needed to train large language models more effectively and broadly.
But here's the catch: while the initial results are promising, will this method hold up under broader scrutiny? The research community, often enamored with its own innovations, would do well to remain vigilant against overfitting and ensure reproducibility across varied contexts. Only time will reveal CME's true impact, but for now, it seems to offer a refreshing alternative to traditional reward signal methodologies.
Get AI news in your inbox
Daily digest of what matters in AI.