arXiv cs.CL· Shourov Joarder, Diganta Sikdar, Ahsan Habib Akash, Binod Bhattarai, Prashnna Gyawali·· 12 小时前AI 评分32
双奖励优于单奖励:一种无坍缩的多奖励 RLIF 训练框架
Two is better than one: A Collapse-free Multi-Reward RLIF Training Framework
AI 导读
研究提出一种多奖励 RLIF 框架,将训练信号拆解为基于聚类投票的答案级奖励与基于 token 级自确定度的补全级奖励,并用 GDPO 归一化缓解奖励尺度失衡。该方法还引入 KL-Cov 正则化,针对导致熵过度下降的低熵 token 分布,以保持探索、防止后期坍缩。在数学推理与代码生成基准上,其稳定性和鲁棒性优于此前无监督 RL 方法,性能接近有监督 RLVR。
正文 · 原文
Abstract:Reinforcement learning with verifiable rewards (RLVR) has substantially improved the reasoning ability of LLMs, but often depends on external supervision from human annotations or gold-standard solutions. Reinforcement learning from internal feedback (RLIF) has recently emerged as a scalable unsupervised alternative, using signals extracted from the model itself. However, existing RLIF methods typically rely on a single internal reward, which can lead to reward hacking, entropy collapse, and degraded reasoning structure. We propose a multi-reward RLIF framework that decomposes the training signal into two complementary components: an answer-level reward based on cluster voting and a completion-level reward based on token-wise self-certainty. To combine these signals robustly, we apply GDPO-based normalization to reduce reward-scale imbalance. We further introduce KL-Cov regularization, which targets low-entropy token distributions responsible for disproportionate entropy reduction, preserving exploration and preventing late-stage collapse. Across mathematical reasoning and code-generation benchmarks, our method improves stability and robustness over prior unsupervised RL approaches, while achieving performance close to supervised RLVR methods. These results show that complementary internal rewards, combined with targeted regularization, can support stable long-horizon reasoning without relying on external ground-truth supervision. Code will be released soon.
| Subjects: | Machine Learning (cs.LG); Computation and Language (cs.CL) |
| Cite as: | arXiv:2605.22620 [cs.LG] |
| (or arXiv:2605.22620v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2605.22620 arXiv-issued DOI via DataCite |
Submission history
From: Prashnna Gyawali [view email]
[v1]
Thu, 21 May 2026 15:30:47 UTC (237 KB)
[v2]
Sat, 31 Oct 2026 14:42:51 UTC (6,091 KB)
来源:arXiv cs.CL · arxiv.org