---
title: Relative Surprisal Index (RSI)
url: https://www.emergentmind.com/topics/reverse-surprise-index-rsi
type: topic
---

# Relative Surprisal Index (RSI)

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” [2606.31575]. 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 [2606.31575].

## 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 [2606.31575]. 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 [2606.31575].

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 [2606.31575].

## 2. Formal definition and basic properties

At step $t$, the language-model policy is defined as the softmax over logits $l_t \in \mathbb{R}^V$, where the vocabulary size is $V$ and
$$
p_{t,i} = \mathrm{Softmax}(l_t)_i = \frac{\exp(l_{t,i})}{\sum_{j=1}^V \exp(l_{t,j})}.
$$
The predictive entropy at position $t$ is
$$
\mathcal{H}_t = -\sum_{s=1}^V p_{t,s}\,\log p_{t,s}.
$$
The distributional average log-probability is
$$
\mathbb{E}_{p}[\log p] = \sum_s p_{t,s}\log p_{t,s} = -\mathcal{H}_t.
$$
If the sampled token is $o_t$ with probability $p_{t,n} = \pi_\theta(o_t \mid q, o_{<t})$, then the deviation of the selected log-probability from the distributional average is
$$
\delta_t = \log p_{t,n} - \mathbb{E}_{p}[\log p] = \log p_{t,n} + \mathcal{H}_t.
$$
The Relative Surprisal Index is then defined as
$$
\mathrm{RSI}_t = \frac{\delta_t}{\mathcal{H}_t} = 1 + \frac{\log \pi_\theta(o_t \mid q, o_{<t})}{\mathcal{H}_t}.
$$
It is defined for $\mathcal{H}_t > 0$. For degenerate distributions with $\mathcal{H}_t \to 0$, the paper states that RSI is defined by continuity, with
$$
\lim_{\mathcal{H}_t\to 0}\mathrm{RSI}_t = 1
$$
(Appendix C) [2606.31575].

Two normalization properties are emphasized. First, the expectation under the policy distribution is
$$
\mathbb{E}_{o_t\sim \pi_\theta}[\mathrm{RSI}_t] = 0.
$$
Second, the range is
$$
\mathrm{RSI}_t \in (-\infty, 1],
$$
with heavy empirical mass near $1$ 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 [2606.31575].

## 3. Theoretical interpretation via local sensitivity

The theoretical contribution is framed through a perturbation of the selected logit,
$$
l_{t,n} \mapsto l_{t,n} + \Delta l_{t,n},
$$
and through the gradient norm of the selected-token log-probability with respect to logits. The gradient is
$$
\nabla_{l_t} \log p_{t,n} = \mathbf{e}_n - \mathbf{p}_t,
$$
where $\mathbf{e}_n$ is the one-hot basis vector and $\mathbf{p}_t$ is the probability vector. Its $\ell_2$ norm is
$$
\mathcal{T}_t = \bigl\lVert \nabla_{l_t} \log p_{t,n}\bigr\rVert_2 = \sqrt{(1-p_{t,n})^2 + \sum_{j\neq n}p_{t,j}^2}.
$$
Under mild conditions, specifically $\mathcal{H}_t > 0$ and $\mathrm{RSI}_t \neq 0$, Theorem 1 states that the ratio of first-order variations of $\log \mathcal{T}_t$ and $\log \mathcal{H}_t$ with respect to the selected-logit perturbation equals the reciprocal of RSI:
$$
\frac{d \log \mathcal{T}_t}{d \log \mathcal{H}_t} = \frac{1}{\mathrm{RSI}_t}.
$$
The proof sketch proceeds from the softmax derivative
$$
\frac{\partial p_{t,i}}{\partial l_{t,j}} = p_{t,i}(\delta_{ij} - p_{t,j}),
$$
the entropy derivative
$$
\frac{\partial \mathcal{H}_t}{\partial l_{t,i}} = -p_{t,i}\bigl(\log p_{t,i} + \mathcal{H}_t\bigr),
$$
and the gradient-norm derivative
$$
\frac{d \log \mathcal{T}_t}{d l_{t,n}} = -p_{t,n},
$$
which combine algebraically into the stated ratio [2606.31575].

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 $|\mathrm{RSI}|$ 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 [2606.31575].

## 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 $a<0<b$ with larger magnitude on the negative side, $|a| > b$, and keeps token $o_t$ iff
$$
\mathrm{RSI}_t \in [a,b].
$$
This admits an equivalent entropy-adaptive probability form:
$$
e^{(a-1)\mathcal{H}_t} \leq \pi_\theta(o_t \mid q, o_{<t}) \leq e^{(b-1)\mathcal{H}_t}.
$$
The paper emphasizes that this differs from static probability thresholds because the gate tightens in low-entropy regimes and relaxes in high-entropy regimes [2606.31575].

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 $b$. Unstable high-surprisal tail tokens, corresponding to very low-probability realizations, are limited through the lower bound $a$. 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 [2606.31575].

The paper also includes concrete examples. For a 3-token distribution $p = [0.7, 0.2, 0.1]$, the entropy is approximately $0.802$. If the selected token has probability $0.7$, then $\log p \approx -0.357$ and $\mathrm{RSI} \approx 0.555$, which is retained under bounds $[-6, 0.9]$. If the selected token has probability $0.1$, then $\log p \approx -2.303$ and $\mathrm{RSI} \approx -1.872$; this remains within $[-6, 0.9]$ but would be excluded under a tighter lower bound such as $[-1.5, 0.9]$. In a low-surprisal regime with $p = [0.95, 0.03, 0.02]$ and $\mathcal{H}_t \approx 0.232$, selecting the $0.95$ token yields $\mathrm{RSI} \approx 0.779$, so lowering $b$ can filter overly predictable positions [2606.31575].

## 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 $\eta_{i,t}(\theta)$ and includes PPO-style clipping and a KL penalty. The sequence-level advantage is
$$
A_i = \frac{R_i - \mathrm{mean}(R)}{\mathrm{std}(R)},
$$
and the policy ratio is
$$
r_{i,t}(\theta) = \frac{\pi_\theta(o_{i,t}\mid q, o_{i,<t})}{\pi_{\text{old}}(o_{i,t}\mid q, o_{i,<t})}.
$$
The per-token objective term is
$$
\eta_{i,t}(\theta)
=
\min\!\bigl(r_{i,t}(\theta)A_i,\,
\mathrm{clip}(r_{i,t}(\theta), 1-\epsilon_1, 1+\epsilon_2)A_i\bigr)
- \beta\,\mathbb{D}_{\mathrm{KL}}\!\bigl(\pi_\theta(\cdot\mid q,o_{i,<t}) \,\Vert\, \pi_{\mathrm{ref}}(\cdot\mid q,o_{i,<t})\bigr).
$$
RSI-S introduces an indicator mask $\mathbb{I}[\mathrm{RSI}_{i,t}\in[a,b]]$ and the kept-token count
$$
|\kappa_i^{a,b}|=\#\{t:\mathrm{RSI}_{i,t}\in[a,b]\}.
$$
The resulting masked objective is
$$
J^{\mathrm{RSI}}(\theta)
=
\mathbb{E}\Biggl[
\frac{1}{G}\sum_{i=1}^G
\frac{1}{|\kappa_i^{a,b}|}
\sum_t
\underbrace{\mathbb{I}[\mathrm{RSI}_{i,t}\in[a,b]]\,\eta_{i,t}(\theta)}_{\gamma_{i,t}(\theta)}
\Biggr].
$$
If $|\kappa_i^{a,b}|=0$, the sequence is dropped from the update to avoid ill-defined normalization [2606.31575].

The algorithmic steps are given explicitly. For each batch of queries, the system generates $G$ sequences with the old policy $\pi_{\text{old}}$. For each sequence and each position, it computes current-policy logits, probabilities, entropy, and RSI. Tokens inside $[a,b]$ are included and the remainder are masked. The method then computes $\eta_{i,t}(\theta)$ for included tokens, normalizes by the number of kept tokens, aggregates across tokens and sequences, and updates $\theta$ [2606.31575].

The practical computation is lightweight. Per-token RSI requires softmax probabilities, entropy $\mathcal{H}_t = -\sum p\log p$, and the selected-token log-probability. The paper states that entropy computation is a vector dot product of complexity $O(V)$ 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 [2606.31575].

## 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 [2606.31575].

| 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 [2606.31575].

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 [2606.31575].

The ablations are central to the paper’s argument. Static probability masks fail: for Qwen2.5-3B, fixed ranges P1 $[0.1,0.9]$, P2 $[0.2,0.8]$, and P3 $[0.3,0.7]$ underperform GRPO, whereas RSI-S improves by $+3.30$ 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 $b$ values shifting downward as models grow larger: $0.95$ for 1.5B, $0.90$ for 3B, and $0.80$ for 7B. Finally, Figure 4 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 $0.50\times$, $0.11\times$, and $0.22\times$ the sensitivity of entropy at 1.5B, 3B, and 7B respectively [2606.31575].

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.0$ and top-$p$ $1.0$. GRPO specifics include group size $G=16$, learning rate $1\mathrm{e}{-6}$, batch size $1024$, mini-batch size $128$, ten epochs corresponding to $160$ outer steps and $1280$ mini-batch updates, and a KL term to a reference policy with coefficient $\beta$ together with PPO-style clipping $\epsilon_1,\epsilon_2$. The reported RSI-S intervals are $[-6,0.95]$ for 1.5B, $[-6,0.90]$ for 3B, and $[-6,0.80]$ for 7B. Evaluation uses avg@32 with temperature $0.6$ and top-$p$ $0.95$ on AIME 2024–2026 and AMC 2022–2024 [2606.31575].

The paper states several limitations. The theoretical link depends on first-order perturbations of the selected logit and requires $\mathcal{H}_t>0$ and $\mathrm{RSI}_t\neq 0$. In highly degenerate distributions, $\mathrm{RSI}\to 1$ by continuity, and selection effectively reduces to trimming extreme negatives via $a$. 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 [2606.31575].

The future directions outlined are specific. They include replacing hard masks with soft weighting, potentially using a bell-shaped function around zero; learning $[a,b]$ 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 [2606.31575].

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** [2606.31575].

Source: https://www.emergentmind.com/topics/reverse-surprise-index-rsi