NeFut Logo NeFut
EN 管理员登录

[AI学术] 颠覆传统的强化学习:尾部感知信用校准方法 TACO

发布于:2026-07-11 22:00 最后更新:2026-07-13 08:40
#AI #optimization #Reinforcement Learning

在强化学习(RL)领域,尽管取得了显著的成功,提升了大语言模型(LLMs)的推理能力,但现有的无评论员RL方法依赖于均匀的信用分配,这种方法在所有令牌之间广播相同的优势,忽视了它们之间的差异。

我们识别出这一设计的一个关键失败模式,称之为正信用污染:低概率的尾部令牌在上下文中出现错误,却与同一轨迹中的合理令牌获得相同的正信用,从而导致错误推理行为的无差别强化。

为了解决这一问题,我们提出了尾部感知信用校准(TACO)方法,该方法校准均匀的信用分配,以抑制不良的正更新。

TACO 首先计算一个尾部风险评分,该评分结合局部生成上下文来评估每个令牌落入不可靠尾部的风险,区分意外稀有性和不确定性驱动的探索。

然后,TACO 使用该评分来调整风险令牌的正信用,而不完全移除它们的梯度,以便有用的稀有模式能够累积强化,同时逐步抑制偶然噪声。

实验结果显示,在三种 LLM 和八个基准测试中,TACO 始终优于 GRPO 风格的基线。值得注意的是,TACO 提高了训练稳定性,支持长时间强化学习中的持续性能提升。

源代码可在此获取:GitHub

博主点评: TACO 方法通过精细调节风险评分,突出了在强化学习中处理低概率尾部令牌的重要性。这一创新不仅提高了模型的推理能力,也为未来的RL研究提供了新的思路,值得关注和深入研究。

原文链接: https://arxiv.org/abs/2607.07976

[h] 返回首页