Tail-Likelihood Reinforcement Learning

ML RL SRL
强化学习通常以优化平均奖励为目标。然而,对于生成式策略而言,平均值可能掩盖一个关键区别:两种策略虽能达到相同的平均奖励,但在生成稀有但高奖励的轨迹(rollout)方面的概率却可能大相径庭。这一点在训练与推理阶段采样规模不断扩大时尤为关键——因为采样增益能否实现,取决于策略是否仍能为高奖励结果保留足够的概率质量。为此,我们提出直接优化该“覆盖能力”(coverage)。具体而言,我们不再仅关注期望奖励,而是考察其全部上侧尾部(upper tails):对每一个给定的奖励阈值,策略超越该阈值的概率是多少?由此,原本连续的奖励被转化为一族二元成功事件。我们提出了“尾部似然强化学习”(Tail-Likelihood Reinforcement Learning, TailRL),该方法通过最大化策略超越一个随机选取的奖励阈值的对数概率来进行优化。其梯度天然赋予稀有、高奖励轨迹更高权重,并可被解释为一系列“Top-k 采样”(Best-of-k)梯度的混合形式。TailRL 仅需对优势函数(advantage function)作简单修改,因而可无缝嵌入现有强化学习流程。在物体定位、迷宫导航、图形用户界面(GUI)元素接地(grounding)、以及代码优化等任务上,TailRL 能有效利用训练中出现的稀有高奖励样本,规避次优解;同时,所训练出的模型在推理阶段更能从额外采样中获益。
Reinforcement learning typically optimizes average reward. For generative policies, the average can hide an important distinction: two policies can achieve the same mean reward while having very different chances of producing a rare but high-reward rollout. This matters as sampling increases during training and inference, since its benefit depends on retaining probability mass on high-reward outcomes. We propose to optimize this coverage directly. Rather than considering only expected reward, we consider all of its upper tails: for each reward threshold, how likely is the policy to exceed it? This turns a continuous reward into a family of binary success events. We introduce Tail-Likelihood Reinforcement Learning (TailRL), which maximizes the log-probability of exceeding a randomly chosen reward threshold. Its gradient gives more weight to rare, high-reward rollouts and can be interpreted as a mixture of Best-of-(k) gradients. TailRL requires only a simple modification to the advantage function, making it compatible with existing reinforcement learning pipelines. Across object localization, maze navigation, GUI grounding, and code optimization, TailRL leverages rare high-reward training samples to avoid suboptimal solutions and yields models that benefit more from additional samples at inference time.
许愿