发现强化学习中策略熵坍塌机制,提出可控探索新方法。
The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models

- 揭示熵下降由动作概率与逻辑值变化的协方差驱动,理论推导匹配实证结果。
- 提出Clip-Cov与KL-Cov两种方法,有效抑制熵坍塌,提升探索能力。
- 适用于大规模语言模型强化训练,对算力扩展下的持续探索有关键意义。
本文针对大语言模型强化学习中策略熵坍塌这一核心障碍展开研究。在大量强化训练实验中,未干预时策略熵在训练初期急剧下降,伴随性能饱和。我们建立经验关系式 R = -a·e^H + b,表明性能与熵存在可预测的权衡关系,当熵为零时性能达到上限。理论分析指出,熵的变化由动作概率与逻辑值变化的协方差驱动,该协方差在策略梯度类算法下正比于优势函数。实证验证了协方差项与熵差值完全一致,且其值全程基本为正,解释了熵单调下降的机制。基于此,我们提出两种简单有效的方法:Clip-Cov(限制高协方差词元更新)和KL-Cov(对高协方差词元施加KL惩罚),实验显示二者能有效维持探索能力,避免熵坍塌,提升下游性能。
原文摘要 · Abstract (English)
This paper aims to overcome a major obstacle in scaling RL for reasoning with LLMs, namely the collapse of policy entropy. Such phenomenon is consistently observed across vast RL runs without entropy intervention, where the policy entropy dropped sharply at the early training stage, this diminished exploratory ability is always accompanied with the saturation of policy performance. In practice, we establish a transformation equation R=-a*e^H+b between entropy H and downstream performance R. This empirical law strongly indicates that, the policy performance is traded from policy entropy, thus bottlenecked by its exhaustion, and the ceiling is fully predictable H=0, R=-a+b. Our finding necessitates entropy management for continuous exploration toward scaling compute for RL. To this end, we investigate entropy dynamics both theoretically and empirically. Our derivation highlights that, the change in policy entropy is driven by the covariance between action probability and the change in logits, which is proportional to its advantage when using Policy Gradient-like algorithms. Empirical study shows that, the values of covariance term and entropy differences matched exactly, supporting the theoretical conclusion. Moreover, the covariance term stays mostly positive throughout training, further explaining why policy entropy would decrease monotonically. Through understanding the mechanism behind entropy dynamics, we motivate to control entropy by restricting the update of high-covariance tokens. Specifically, we propose two simple yet effective techniques, namely Clip-Cov and KL-Cov, which clip and apply KL penalty to tokens with high covariances respectively. Experiments show that these methods encourage exploration, thus helping policy escape entropy collapse and achieve better downstream performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。