Papers
Topics
Authors
Recent
Search
2000 character limit reached

Tail-Likelihood Reinforcement Learning (TailRL)

Updated 7 September 2026
  • TailRL is a reinforcement-learning objective that enhances the likelihood of achieving sufficiently high, rare reward outcomes. By optimizing the expected log tail probability, it efficiently addresses scenarios where deployment generates multiple samples, focusing on upper reward distribution benefits of adversarial behaviors.
  • TailRL treats the high-reward outcomes optimization objective in policy learning, focusing on rare positive outcomes, improving decision-making in adversarial environments when expected rewards are less effective. This method provides greater marginal weight to rare outcomes rather than predicted averages.
  • TailRL improves outcomes where multiple samples are generated and the best sample is taken, such as object localization, text-maze navigation, GUI grounding, and code runtime optimization, offering benefits in environments with high-reward rare tasks that aren't as impressive for average outcomes in the tasks.

Tail-Likelihood Reinforcement Learning (TailRL) is a reinforcement-learning objective for stochastic generative policies that optimizes the probability of exceeding reward thresholds across the entire upper reward distribution, rather than optimizing expected reward alone. For an input xx, rollout z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x), and normalized reward r(x,z)∈[0,1]r(x,z)\in[0,1], TailRL defines the objective as the expected log tail probability

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,

where

pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).

The objective gives greater marginal weight to rare threshold exceedances and is equivalent to a harmonic mixture of Best-of-kk objectives over all sampling budgets. Its practical estimator requires only group rollouts, reward sorting, and replacement of the conventional advantage function. TailRL is introduced in “Tail-Likelihood Reinforcement Learning” (Ramasubramanian et al., 2 Sep 2026).

1. Motivation and conceptual basis

Standard policy-gradient reinforcement learning optimizes the expected reward

JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].

Its gradient is

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].

Expected reward is appropriate when a single rollout is used or when the mean is the desired statistic. It is less suitable when deployment can generate multiple samples and select the best one. Two policies can have similar means while differing substantially in the probability of producing rare, high-reward outcomes. A policy that usually produces moderately good results may therefore be inferior under repeated sampling to a policy that occasionally produces an exceptionally high-reward result.

For binary rewards, the relevant repeated-sampling metric is

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.

For continuous rewards, the corresponding quantity is

Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].

At z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)0, both reduce to mean reward. As z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)1 increases, their values depend increasingly on the upper tail of the reward distribution.

TailRL addresses the possibility that expected-reward optimization removes probability mass from rare but valuable behaviors. Once such trajectories become sufficiently unlikely, they are sampled infrequently, reducing the data available for further improvement. TailRL instead seeks to increase the likelihood of exceeding every reward threshold, with particular emphasis on thresholds that are currently cleared rarely.

The method is distinct from several other uses of “tail” in reinforcement learning. Heavy-tailed reward methods treat extreme observations as a statistical nuisance while continuing to optimize expected reward (Zhuang et al., 2021, Huang et al., 2023). Heavy-tailed policy search uses heavy-tailed action distributions to improve nonlocal exploration and basin escape, while also optimizing expected return (Bedi et al., 2021). Distributional RL models the return distribution and may derive risk measures from it, but does not necessarily optimize tail probabilities directly (Özsoy, 16 Jul 2025). TailRL specifically changes the policy objective toward upper-tail coverage.

2. Tail probabilities and the logarithmic objective

For each threshold z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)2, define the binary event

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)3

and its probability

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)4

The expected reward admits the layer-cake representation

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)5

Expected-reward RL therefore aggregates tail probabilities arithmetically. TailRL instead aggregates their logarithms:

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)6

The logarithm produces inverse-probability weighting. Since

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)7

an absolute increase in a rare tail probability has greater marginal value than the same increase in a commonly achieved event. The general family

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)8

has gradient

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)9

TailRL corresponds to r(x,z)∈[0,1]r(x,z)\in[0,1]0.

Using the score-function identity,

r(x,z)∈[0,1]r(x,z)\in[0,1]1

the TailRL gradient becomes

r(x,z)∈[0,1]r(x,z)\in[0,1]2

A rollout with reward r(x,z)∈[0,1]r(x,z)\in[0,1]3 exceeds every threshold below r(x,z)∈[0,1]r(x,z)\in[0,1]4. The integral therefore accumulates inverse-tail-probability weights over the thresholds cleared by that rollout. High-reward rollouts receive more credit when the thresholds they clear are rarely achieved by the current policy.

For a general bounded reward r(x,z)∈[0,1]r(x,z)\in[0,1]5, TailRL applies to the affine-normalized reward:

r(x,z)∈[0,1]r(x,z)\in[0,1]6

3. Likelihood interpretation and relation to MaxRL

TailRL has an audit interpretation based on independent threshold assignments. Select r(x,z)∈[0,1]r(x,z)\in[0,1]7 thresholds

r(x,z)∈[0,1]r(x,z)\in[0,1]8

and independently sample rollouts

r(x,z)∈[0,1]r(x,z)\in[0,1]9

Define

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,0

Independence gives

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,1

Consequently,

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,2

As JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,3 tends to infinity, the Riemann sum converges to

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,4

This interpretation requires independent rollouts. Reusing a single rollout for every threshold would collapse the nested events to the hardest threshold rather than produce a product of tail probabilities.

For binary rewards, let

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,5

For every JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,6,

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,7

and therefore

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,8

This recovers the binary MaxRL objective:

JTailRL(θ;x)=∫01log⁡pθ(x,τ) dτ,J_{\mathrm{TailRL}}(\theta;x) = \int_0^1 \log p_\theta(x,\tau)\,d\tau,9

with gradient

pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).0

The factor pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).1 emphasizes rare successful rollouts. TailRL generalizes this binary principle to the complete continuum of reward thresholds.

TailRL should not be confused with direct tail-likelihood estimation in the statistical-density sense. It does not require a parametric reward density, a learned conditional density, or a likelihood-ratio model. Its “likelihood” is the probability of threshold exceedance under the policy, and the logarithm of that probability is used as the optimization objective.

4. Best-of-pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).2 decomposition

Let pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).3 be independent rollouts and define

pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).4

The layer-cake identity gives

pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).5

Because all pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).6 rollouts fail to exceed pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).7 with probability pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).8,

pθ(x,τ)=Pr⁡z∼πθ(⋅∣x)(r(x,z)>τ).p_\theta(x,\tau) = \Pr_{z\sim\pi_\theta(\cdot\mid x)} \left(r(x,z)>\tau\right).9

Using

kk0

TailRL admits the decomposition

kk1

Differentiation yields

kk2

TailRL is therefore a harmonic mixture of Best-of-kk3 gradients over all sampling budgets. It is not tuned to a single deployment value of kk4. The harmonic weighting gives nonzero importance to every finite sampling budget while emphasizing lower orders more strongly.

A finite-order TailRL objective is

kk5

Its gradient can be written as

kk6

The threshold multiplier is

kk7

At kk8,

kk9

so the finite-order objective reduces to Best-of-JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].0 and therefore to expected reward up to an additive constant. As JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].1 increases, the objective incorporates increasingly higher-order Best-of-JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].2 signals. In the limit,

JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].3

recovering population TailRL.

The rollout group size is thus not merely a variance-reduction parameter. It selects the order of the finite-sample objective being estimated:

  • JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].4: expected reward;
  • larger JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].5: increasingly higher-order Best-of-JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].6 signals;
  • JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].7: population TailRL.

These identities require independent rollouts, bounded rewards in the stated range, fixed policy support, differentiable policy probabilities, summable parameter gradients, and sufficient integrability to exchange differentiation, integration, and expectation.

5. Finite-rollout estimator and algorithm

In practice, JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].8 is unknown. TailRL draws a group of JRL(θ;x)=Ez∼πθ(⋅∣x)[r(x,z)].J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta(\cdot\mid x)} [r(x,z)].9 independent rollouts

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].0

with rewards

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].1

For rollout ∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].2, define

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].3

At threshold ∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].4, the credit is divided equally among the rollouts that exceed it. A threshold exceeded by only one rollout gives that rollout all the threshold credit; a threshold exceeded by all ∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].5 rollouts gives each rollout credit ∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].6.

The uncentered estimator is

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].7

The paper establishes

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].8

The weights are computed by sorting rewards. Let

∇θJRL(θ;x)=Ez∼πθ[r(x,z)∇θlog⁡πθ(z∣x)].\nabla_\theta J_{\mathrm{RL}}(\theta;x) = \mathbb{E}_{z\sim\pi_\theta} \left[ r(x,z)\nabla_\theta\log\pi_\theta(z\mid x) \right].9

Then

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.0

or equivalently

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.1

The computation requires sorting and a cumulative calculation. A useful identity is

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.2

For variance reduction, the practical implementation uses the group baseline

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.3

and the centered TailRL advantage

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.4

The policy-gradient update then has the standard form

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.5

Centering introduces a finite-order subtlety: the centered estimator targets the order-Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.6 objective rather than the order-Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.7 gradient. Mean-reduced implementations also require attention to whether coefficients are multiplied by Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.8.

For comparison, RLOO uses

Pass@k(θ;x)=1−(Pr⁡z∼πθ[r(x,z)=0])k.\mathrm{Pass@}k(\theta;x) = 1- \left( \Pr_{z\sim\pi_\theta}[r(x,z)=0] \right)^k.9

while GRPO uses

Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].0

Both retain the expected-reward target. TailRL changes the relative weighting of reward levels: high thresholds cleared by few rollouts receive disproportionately greater credit.

For binary rewards, if Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].1 of the Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].2 rollouts succeed,

Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].3

and hence

Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].4

which is the MaxRL estimator.

A practical TailRL iteration consists of:

  1. Sampling a group of Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].5 rollouts for each input.
  2. Computing the reward of every rollout.
  3. Sorting the rewards.
  4. Computing the Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].6 values.
  5. Optionally centering them to obtain advantages.
  6. Substituting the resulting advantages into an existing policy-gradient loss.
  7. Performing an on-policy update.

TailRL is compatible with group-based systems such as GRPO and RLOO. It does not require a learned critic, an explicit numerical integral over thresholds, or changes to the policy architecture, reward function, rollout engine, or token-level log-probability calculation.

6. Empirical results, scope, and limitations

TailRL was evaluated on object localization, text-maze navigation, GUI grounding, and code runtime optimization (Ramasubramanian et al., 2 Sep 2026). Across these settings, the reported pattern is that TailRL is most useful when high-reward outcomes are attainable but rare, and when expected-reward methods tend to collapse onto reliable moderate-reward behaviors.

Object localization

The ImageNet object-localization task assigns an IoU reward in Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].7 to sampled bounding boxes. A pretrained ResNet-50 backbone and four categorical policy heads produce Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].8 bins per coordinate, yielding Best-of-k(θ;x)=Ez1,…,zk∼πθ[max⁡1≤i≤kr(x,zi)].\mathrm{Best\text{-}of\text{-}k}(\theta;x) = \mathbb{E}_{z_1,\ldots,z_k\sim\pi_\theta} \left[ \max_{1\leq i\leq k}r(x,z_i) \right].9 possible boxes. Because the output space is finite, population tail probabilities can be computed exactly.

TailRL was compared with GRPO, RLOO, PKPO, and supervised objectives including MSE, z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)00, GIoU, and z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)01. Rollout groups had sizes z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)02. The population TailRL objective matched or exceeded direct-coordinate supervision on CorLoc and mean IoU while improving Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)03 IoU. At matched z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)04, TailRL outperformed GRPO and RLOO on CorLoc, mean IoU, and Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)05 IoU. At z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)06, it exceeded GRPO and RLOO trained with z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)07 on the reported metrics.

The advantage was larger on difficult images. The reported TailRL improvement over the stronger expected-reward baseline was approximately z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)08 IoU on easy images, z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)09 on medium images, and z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)10 on hard images. On hard images, expected-reward baselines were near z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)11 IoU, while TailRL reached approximately z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)12.

Text-maze navigation

In the text-maze task, a decoder-only transformer generates movement sequences in a z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)13 maze. Rewards combine progress toward the goal and path efficiency, with reward z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)14 corresponding to a shortest path.

The experiment used z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)15 rollouts per prompt, with additional sweeps over z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)16. Initial shortest-path success rates ranged from approximately z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)17 to z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)18.

When initial success was relatively common, TailRL, GRPO, and RLOO behaved similarly. When successful paths were rare, expected-reward methods failed to improve reliably, whereas TailRL continued to improve. With z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)19, TailRL remained near the baseline floor; with z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)20, it improved sharply because successful paths appeared often enough to be exploited. Larger groups produced smaller additional gains.

At inference time, TailRL’s Pass@z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)21 and Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)22 reward increased more rapidly with z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)23. The advantage was not always large at z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)24, but became increasingly visible with additional samples. Training dynamics showed that GRPO and RLOO lost entropy rapidly, while TailRL retained higher entropy and generated progressively longer paths.

GUI grounding

The GUI-grounding task asks a vision-LLM to generate click coordinates from screenshots and instructions. Qwen2.5-VL-3B and Qwen2.5-VL-7B models were evaluated on ScreenSpot-Pro.

At z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)25, TailRL and RLOO had similar Pass@1. As z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)26 increased, TailRL’s performance continued to improve while RLOO plateaued earlier. TailRL matched RLOO’s Pass@1024 using eight samples for the 3B model and four samples for the 7B model. Category-wise evaluation showed TailRL leading the stronger baseline in 11 of 12 model-scale/category combinations at z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)27, with gains as large as 12.0 percentage points on Dev at 3B and 11.5 points on CAD at 7B.

Code runtime optimization

The code-optimization task rewards functionally equivalent C++ programs according to execution speed. Copying the source program is always correct and receives reward exactly z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)28, creating a reliable but suboptimal mode. Initially, z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)29 of rollouts were incorrect, z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)30 were correct but no faster than the input, and z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)31 were correct and faster.

GRPO and RLOO rapidly collapsed toward copying, with correctness above z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)32, mean reward below z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)33, and Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)34 speedups of z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)35 and z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)36, respectively. TailRL maintained higher entropy and continued producing risky rewrites. Its mean training reward reached z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)37, nearly three times the copy baseline. On the held-out test set, TailRL achieved a Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)38 speedup of z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)39, compared with z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)40 for GRPO and z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)41 for RLOO.

Theoretical and practical limitations

TailRL does not guarantee discovery of an outcome whose policy probability is zero. Rare-event learning still requires at least occasional successful or high-reward rollouts. The maze experiment demonstrates that a group size of z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)42 can be insufficient when successful trajectories are extremely rare.

Inverse-tail weighting can produce large gradients when only one rollout clears a high threshold. Gradient clipping or careful optimization may therefore be needed. The method also depends on the reward function: if the reward is misspecified, aggressively optimizing rare high rewards can amplify reward hacking or other undesirable behaviors.

The population objective is generally unavailable in practice. Finite groups estimate an order-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)43 objective, and centering changes the exact finite-order target. Larger groups expose rarer thresholds but increase sampling cost. TailRL also assumes reliable reward evaluation and independent rollouts for its Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)44 interpretation.

The method does not establish global convergence or global optimality for neural-network policies. Its theoretical results concern gradient identities, finite-order estimators, and Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)45 decompositions under support, smoothness, boundedness, and integrability assumptions. The empirical results are limited to the reported localization, navigation, GUI, and code-optimization settings.

TailRL is also distinct from risk-averse tail optimization. It emphasizes the upper tail of reward outcomes and seeks rare high-quality successes. It does not optimize lower-tail safety criteria such as CVaR or expected shortfall. In contrast, IQN–CVaR methods estimate a return distribution and optimize lower-tail loss severity (Zhang, 6 Oct 2025), while SteinGate compares rollout-cost distributions with a safe reference distribution rather than directly optimizing a reward upper tail (Chemingui et al., 14 Jul 2026).

The central distinction is therefore:

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)46

whereas

z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)47

Its finite-sample implementation is a sorted, tail-aware replacement for the ordinary group advantage, while its population interpretation is a harmonic combination of Best-of-z∼πθ(⋅∣x)z\sim\pi_\theta(\cdot\mid x)48 objectives aligned with inference-time sampling.

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Tail-Likelihood Reinforcement Learning (TailRL).