Relative Surprisal Index (RSI)
- RSI is a token-level metric that quantifies surprise by comparing the selected token's log-probability to the average, normalized by the predictive entropy.
- It captures both uncertainty and token rarity, reconciling high-entropy signals with low-probability token drawbacks in reinforcement learning with verifiable rewards.
- RSI selection (RSI-S) applies asymmetric bounds to filter tokens adaptively, leading to improved model performance and reduced response length on math benchmarks.
Searching arXiv for the specified paper to ground the article in the cited source. “Reverse Surprise Index” is a misnomer. In the RLVR literature, the correct term is Relative Surprisal Index (RSI), introduced in “Which Tokens Matter? Adaptive Token Selection for RLVR with the Relative Surprisal Index” (Lv et al., 30 Jun 2026). RSI is a token-level score for reinforcement learning with verifiable rewards that couples two signals previously used separately in token selection and weighting: predictive entropy at a token position and the probability of the specific token that was actually sampled. The stated motivation is that evaluating sampled-token probability or entropy in isolation is insufficient to capture policy optimization dynamics. RSI therefore measures the selected token’s log-probability relative to the distribution’s average log-probability, normalized by entropy, and serves as the basis for RSI Selection (RSI-S), an entropy-adaptive token filtering method for GRPO-style RLVR (Lv et al., 30 Jun 2026).
1. Conceptual definition and nomenclature
The paper explicitly clarifies that Relative Surprisal Index is the correct expansion of RSI and that “Reverse Surprise Index” should not be used (Lv et al., 30 Jun 2026). In context, RSI is designed for RL with Verifiable Rewards, a setting in which token-level training decisions affect policy updates during reasoning-oriented optimization.
RSI is positioned as a response to two empirical paradigms that appear contradictory when stated independently. One line of inquiry emphasizes high-entropy token positions during training, whereas another warns against allowing low-probability tokens to dominate gradient updates. The paper argues that these positions are only superficially inconsistent, because high entropy and low sampled-token probability are correlated but not interchangeable descriptors of the optimization state. This suggests that the operative quantity is not uncertainty alone, nor improbability alone, but the relationship between the realized token and the uncertainty scale of the whole predictive distribution (Lv et al., 30 Jun 2026).
The paper’s core intuition is threefold. Entropy alone indicates where the model is uncertain but does not indicate whether the realized token was typical at that uncertainty level. Probability alone indicates how likely the realized token was, but not whether low probability reflects informative exploration or an unstable tail event. RSI combines both by comparing the selected token’s log-probability to the distribution’s average log-probability and normalizing by entropy. Positive RSI indicates “lower-than-average surprisal,” negative RSI indicates “higher-than-average surprisal,” and moderate magnitudes around zero correspond to tokens aligned with the distribution’s uncertainty scale (Lv et al., 30 Jun 2026).
2. Formal definition and basic properties
At step , the language-model policy is defined as the softmax over logits , where the vocabulary size is and
The predictive entropy at position is
The distributional average log-probability is
If the sampled token is with probability , then the deviation of the selected log-probability from the distributional average is
The Relative Surprisal Index is then defined as
0
It is defined for 1. For degenerate distributions with 2, the paper states that RSI is defined by continuity, with
3
(Appendix C) (Lv et al., 30 Jun 2026).
Two normalization properties are emphasized. First, the expectation under the policy distribution is
4
Second, the range is
5
with heavy empirical mass near 6 according to Figure 1. These properties matter because they establish RSI as a centered, bounded-above token statistic whose sign and magnitude are directly interpretable in terms of relative surprisal rather than absolute surprisal (Lv et al., 30 Jun 2026).
3. Theoretical interpretation via local sensitivity
The theoretical contribution is framed through a perturbation of the selected logit,
7
and through the gradient norm of the selected-token log-probability with respect to logits. The gradient is
8
where 9 is the one-hot basis vector and 0 is the probability vector. Its 1 norm is
2
Under mild conditions, specifically 3 and 4, Theorem 1 states that the ratio of first-order variations of 5 and 6 with respect to the selected-logit perturbation equals the reciprocal of RSI:
7
The proof sketch proceeds from the softmax derivative
8
the entropy derivative
9
and the gradient-norm derivative
0
which combine algebraically into the stated ratio (Lv et al., 30 Jun 2026).
The interpretation given in the paper is that RSI characterizes the local sensitivity coupling between changes in entropy and changes in the logit-gradient norm when nudging the selected logit. Smaller 1 implies stronger coupling, since the magnitude of the ratio becomes larger, while extreme negative RSI is associated with instability from tail tokens. A plausible implication is that RSI is not merely a descriptive score for filtering but also a diagnostic of how token-level perturbations map into optimization-relevant quantities in RLVR (Lv et al., 30 Jun 2026).
4. RSI Selection as entropy-adaptive token filtering
Building on RSI, the paper proposes RSI Selection (RSI-S), which retains tokens whose RSI lies within a stable interval. The method chooses asymmetric bounds 2 with larger magnitude on the negative side, 3, and keeps token 4 iff
5
This admits an equivalent entropy-adaptive probability form:
6
The paper emphasizes that this differs from static probability thresholds because the gate tightens in low-entropy regimes and relaxes in high-entropy regimes (Lv et al., 30 Jun 2026).
The intended effect is to filter both ends of the token spectrum. Redundant low-surprisal tokens, particularly high-probability tokens in low-entropy regimes, are limited through the upper bound 7. Unstable high-surprisal tail tokens, corresponding to very low-probability realizations, are limited through the lower bound 8. The paper presents this as the mechanism by which RSI-S reconciles the empirical preference for high-entropy positions with the caution against tail-token dominance. This suggests that the “middle band” of RSI values is treated as the region in which tokens are simultaneously informative and comparatively stable for policy optimization (Lv et al., 30 Jun 2026).
The paper also includes concrete examples. For a 3-token distribution 9, the entropy is approximately 0. If the selected token has probability 1, then 2 and 3, which is retained under bounds 4. If the selected token has probability 5, then 6 and 7; this remains within 8 but would be excluded under a tighter lower bound such as 9. In a low-surprisal regime with 0 and 1, selecting the 2 token yields 3, so lowering 4 can filter overly predictable positions (Lv et al., 30 Jun 2026).
5. Integration into GRPO-style RLVR
The integration target is GRPO, described in the paper as group relative policy optimization with a KL term to a reference policy. GRPO’s per-token term is written as 5 and includes PPO-style clipping and a KL penalty. The sequence-level advantage is
6
and the policy ratio is
7
The per-token objective term is
8
RSI-S introduces an indicator mask 9 and the kept-token count
0
The resulting masked objective is
1
If 2, the sequence is dropped from the update to avoid ill-defined normalization (Lv et al., 30 Jun 2026).
The algorithmic steps are given explicitly. For each batch of queries, the system generates 3 sequences with the old policy 4. For each sequence and each position, it computes current-policy logits, probabilities, entropy, and RSI. Tokens inside 5 are included and the remainder are masked. The method then computes 6 for included tokens, normalizes by the number of kept tokens, aggregates across tokens and sequences, and updates 7 (Lv et al., 30 Jun 2026).
The practical computation is lightweight. Per-token RSI requires softmax probabilities, entropy 8, and the selected-token log-probability. The paper states that entropy computation is a vector dot product of complexity 9 and that the overall overhead is negligible relative to forward and backward passes. It further recommends log-softmax for numerical stability, notes that no second-order method or value network is required, and comments that temperature or logit scaling affects entropy and selected-token log-probability in a manner that makes selection more permissive in high-uncertainty regimes and stricter in low-uncertainty regimes (Lv et al., 30 Jun 2026).
6. Empirical behavior, baselines, and ablations
The empirical study uses Qwen2.5-1.5B, 3B, and 7B base models without math-specific SFT, trained on DAPO-MATH-17K and evaluated by avg@32 accuracy on AIME (2024–2026) and AMC (2022–2024). The headline result is that RSI-S improves avg@32 accuracy by 2–3 percentage points over GRPO across model scales and shortens responses (Lv et al., 30 Jun 2026).
| Model | Average avg@32: GRPO → RSI-S | Length: GRPO → RSI-S |
|---|---|---|
| Qwen2.5-1.5B | 12.15 → 14.25 | 1109.03 → 1000.33 |
| Qwen2.5-3B | 18.53 → 21.83 | 1204.98 → 1022.12 |
| Qwen2.5-7B | 29.05 → 31.24 | 1167.11 → 901.16 |
The paper also reports benchmark-specific changes. For Qwen2.5-1.5B, AIME changes from 2.36 to 2.47 and AMC from 21.95 to 26.04. For Qwen2.5-3B, AIME changes from 5.07 to 5.73 and AMC from 31.98 to 37.93. For Qwen2.5-7B, AIME changes from 10.69 to 11.25 and AMC from 47.41 to 51.24 (Lv et al., 30 Jun 2026).
The baseline set consists of GRPO, EB, and PB. EB is described as an entropy baseline that keeps the top 20% highest-entropy tokens. PB is described as the probability baseline of “Do Not Let Low-Probability Tokens Over-Dominate,” which reweights tokens to avoid dominance by low-probability tokens. RSI-S is reported to consistently outperform GRPO, EB, and PB across scales (Lv et al., 30 Jun 2026).
The ablations are central to the paper’s argument. Static probability masks fail: for Qwen2.5-3B, fixed ranges P1 0, P2 1, and P3 2 underperform GRPO, whereas RSI-S improves by 3 percentage points. Balanced bounds are also necessary: removing the upper bound (RSI-SU) or lower bound (RSI-SL) degrades performance relative to the balanced interval. Interval sensitivity varies by scale, with best 4 values shifting downward as models grow larger: 5 for 1.5B, 6 for 3B, and 7 for 7B. Finally, Figure 2 shows that RSI covers more high Jensen–Shannon divergence tokens than entropy selection and is less sensitive to threshold choice; the normalized variance of coverage is reported as 8, 9, and 0 the sensitivity of entropy at 1.5B, 3B, and 7B respectively (Lv et al., 30 Jun 2026).
These results support the paper’s claim that entropy-adaptive selection, rather than static probability gating, is the relevant mechanism. A plausible implication is that RSI-S is acting less as a coarse sparsification heuristic than as a token-level variance-control device tuned to the local uncertainty geometry of the policy.
7. Scope, implementation details, and open questions
The implementation described in the paper uses EasyR1 on top of verl with Qwen2.5-1.5B, 3B, and 7B base models. Training uses a unified prompt template and samples responses with temperature 1 and top-2 3. GRPO specifics include group size 4, learning rate 5, batch size 6, mini-batch size 7, ten epochs corresponding to 8 outer steps and 9 mini-batch updates, and a KL term to a reference policy with coefficient 0 together with PPO-style clipping 1. The reported RSI-S intervals are 2 for 1.5B, 3 for 3B, and 4 for 7B. Evaluation uses avg@32 with temperature 5 and top-6 7 on AIME 2024–2026 and AMC 2022–2024 (Lv et al., 30 Jun 2026).
The paper states several limitations. The theoretical link depends on first-order perturbations of the selected logit and requires 8 and 9. In highly degenerate distributions, 00 by continuity, and selection effectively reduces to trimming extreme negatives via 01. Overly aggressive bounds can prune too many tokens, especially for smaller models, whereas overly permissive bounds may fail to reduce variance. The reported results are restricted to math benchmarks with base models, so gains may differ on other domains or under data contamination. The paper also states that RSI-S is complementary to other signals, such as future KL and polarity-aware weighting, but that the interactions require study (Lv et al., 30 Jun 2026).
The future directions outlined are specific. They include replacing hard masks with soft weighting, potentially using a bell-shaped function around zero; learning 02 online or tying bounds to batch-level entropy statistics or reward variance; combining RSI with future-KL, polarity-aware advantages, or sequence-level divergence; exploring applications beyond RLVR for math in instruction-following, coding, or dialogue; and extending the theory to multi-logit perturbations, second-order terms, value-regularized objectives, and explicit variance-reduction analysis (Lv et al., 30 Jun 2026).
Within the scope of the paper, the significance of RSI is therefore narrowly defined but technically precise. It is a token-level, entropy-normalized relative surprisal measure with a local sensitivity interpretation and a practical role as the gating criterion in RSI-S. The terminology “Reverse Surprise Index” obscures that role; the established term in the source is Relative Surprisal Index (Lv et al., 30 Jun 2026).