Tail-Likelihood Reinforcement Learning (TailRL)
- 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 , rollout , and normalized reward , TailRL defines the objective as the expected log tail probability
where
The objective gives greater marginal weight to rare threshold exceedances and is equivalent to a harmonic mixture of Best-of- 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
Its gradient is
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
For continuous rewards, the corresponding quantity is
At 0, both reduce to mean reward. As 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 2, define the binary event
3
and its probability
4
The expected reward admits the layer-cake representation
5
Expected-reward RL therefore aggregates tail probabilities arithmetically. TailRL instead aggregates their logarithms:
6
The logarithm produces inverse-probability weighting. Since
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
8
has gradient
9
TailRL corresponds to 0.
Using the score-function identity,
1
the TailRL gradient becomes
2
A rollout with reward 3 exceeds every threshold below 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 5, TailRL applies to the affine-normalized reward:
6
3. Likelihood interpretation and relation to MaxRL
TailRL has an audit interpretation based on independent threshold assignments. Select 7 thresholds
8
and independently sample rollouts
9
Define
0
Independence gives
1
Consequently,
2
As 3 tends to infinity, the Riemann sum converges to
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
5
For every 6,
7
and therefore
8
This recovers the binary MaxRL objective:
9
with gradient
0
The factor 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-2 decomposition
Let 3 be independent rollouts and define
4
The layer-cake identity gives
5
Because all 6 rollouts fail to exceed 7 with probability 8,
9
Using
0
TailRL admits the decomposition
1
Differentiation yields
2
TailRL is therefore a harmonic mixture of Best-of-3 gradients over all sampling budgets. It is not tuned to a single deployment value of 4. The harmonic weighting gives nonzero importance to every finite sampling budget while emphasizing lower orders more strongly.
A finite-order TailRL objective is
5
Its gradient can be written as
6
The threshold multiplier is
7
At 8,
9
so the finite-order objective reduces to Best-of-0 and therefore to expected reward up to an additive constant. As 1 increases, the objective incorporates increasingly higher-order Best-of-2 signals. In the limit,
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:
- 4: expected reward;
- larger 5: increasingly higher-order Best-of-6 signals;
- 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, 8 is unknown. TailRL draws a group of 9 independent rollouts
0
with rewards
1
For rollout 2, define
3
At threshold 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 5 rollouts gives each rollout credit 6.
The uncentered estimator is
7
The paper establishes
8
The weights are computed by sorting rewards. Let
9
Then
0
or equivalently
1
The computation requires sorting and a cumulative calculation. A useful identity is
2
For variance reduction, the practical implementation uses the group baseline
3
and the centered TailRL advantage
4
The policy-gradient update then has the standard form
5
Centering introduces a finite-order subtlety: the centered estimator targets the order-6 objective rather than the order-7 gradient. Mean-reduced implementations also require attention to whether coefficients are multiplied by 8.
For comparison, RLOO uses
9
while GRPO uses
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 1 of the 2 rollouts succeed,
3
and hence
4
which is the MaxRL estimator.
A practical TailRL iteration consists of:
- Sampling a group of 5 rollouts for each input.
- Computing the reward of every rollout.
- Sorting the rewards.
- Computing the 6 values.
- Optionally centering them to obtain advantages.
- Substituting the resulting advantages into an existing policy-gradient loss.
- 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 7 to sampled bounding boxes. A pretrained ResNet-50 backbone and four categorical policy heads produce 8 bins per coordinate, yielding 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, 00, GIoU, and 01. Rollout groups had sizes 02. The population TailRL objective matched or exceeded direct-coordinate supervision on CorLoc and mean IoU while improving Best-of-03 IoU. At matched 04, TailRL outperformed GRPO and RLOO on CorLoc, mean IoU, and Best-of-05 IoU. At 06, it exceeded GRPO and RLOO trained with 07 on the reported metrics.
The advantage was larger on difficult images. The reported TailRL improvement over the stronger expected-reward baseline was approximately 08 IoU on easy images, 09 on medium images, and 10 on hard images. On hard images, expected-reward baselines were near 11 IoU, while TailRL reached approximately 12.
Text-maze navigation
In the text-maze task, a decoder-only transformer generates movement sequences in a 13 maze. Rewards combine progress toward the goal and path efficiency, with reward 14 corresponding to a shortest path.
The experiment used 15 rollouts per prompt, with additional sweeps over 16. Initial shortest-path success rates ranged from approximately 17 to 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 19, TailRL remained near the baseline floor; with 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@21 and Best-of-22 reward increased more rapidly with 23. The advantage was not always large at 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 25, TailRL and RLOO had similar Pass@1. As 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 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 28, creating a reliable but suboptimal mode. Initially, 29 of rollouts were incorrect, 30 were correct but no faster than the input, and 31 were correct and faster.
GRPO and RLOO rapidly collapsed toward copying, with correctness above 32, mean reward below 33, and Best-of-34 speedups of 35 and 36, respectively. TailRL maintained higher entropy and continued producing risky rewrites. Its mean training reward reached 37, nearly three times the copy baseline. On the held-out test set, TailRL achieved a Best-of-38 speedup of 39, compared with 40 for GRPO and 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 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-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-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-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:
46
whereas
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-48 objectives aligned with inference-time sampling.