Addressing Performance Saturation for LLM RL via Precise Entropy Curve Control
Bolian Li, Yifan Wang, Yi Ding, Anamika Lochab, Ananth Grama + 1 more
TLDR
Entrocraft, a new rejection-sampling method, precisely controls entropy in LLM RL, preventing performance saturation and significantly boosting training gains.
Key contributions
- Introduces Entrocraft, a rejection-sampling method for precise entropy control in LLM RL.
- Realizes user-customized entropy schedules by biasing advantage distributions, requiring no objective regularization.
- Theoretically explains per-step entropy change and finds linear annealing entropy schedules optimal.
- Addresses performance saturation, enabling a 4B model to outperform an 8B baseline and extend training gains.
Why it matters
Reinforcement learning for LLMs often hits performance limits due to entropy collapse. Entrocraft offers a novel, stable way to manage entropy, preventing saturation. This breakthrough allows LLMs to train longer and achieve significantly better results, even with smaller models.
Original Abstract
Reinforcement learning (RL) has unlocked complex reasoning abilities in large language models (LLMs). However, most RL algorithms suffer from performance saturation, preventing further gains as RL training scales. This problem can be characterized by the collapse of entropy, a key diagnostic for exploration in RL. Existing attempts have tried to prevent entropy collapse through regularization or clipping, but their resulting entropy curves often exhibit instability in the long term, which hinders performance gains. In this paper, we introduce Entrocraft, a simple rejection-sampling approach that realizes any user-customized entropy schedule by biasing the advantage distributions. Entrocraft requires no objective regularization and is advantage-estimator-agnostic. Theoretically, we relate per-step entropy change to the advantage distribution under minimal assumptions, which explains the behavior of existing RL and entropy-preserving methods. Entrocraft also enables a systematic study of entropy schedules, where we find that linear annealing, which starts high and decays to a slightly lower target, performs best. Empirically, Entrocraft addresses performance saturation, significantly improving generalization, output diversity, and long-term training. It enables a 4B model to outperform an 8B baseline, sustains improvement for up to 4x longer before plateauing, and raises pass@K by 50% over the baseline.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.