Papers
Topics
Authors
Recent
Search
2000 character limit reached

Shapley Information Gain (SIG) Overview

Updated 9 July 2026
  • Shapley Information Gain (SIG) is a family of methods that quantify the marginal information contribution of features or coalitions using information-centric objectives such as predictive entropy, pseudo-R², fact utility, or posterior uncertainty.
  • SIG applications span uncertainty explanation in models, regression interpretability through Kullback–Leibler-based pseudo-R², reinforcement learning in proactive healthcare, and Bayesian experimental design for adaptive coalition evaluation.
  • Practical implementations of SIG leverage techniques like coalition sampling, brute-force enumeration, and Gaussian process surrogates to overcome combinatorial challenges and provide efficient, interpretable attributions.

Shapley Information Gain (SIG) denotes a family of Shapley-value-based constructions in which marginal contributions are expressed through explicitly information-centered objectives. In "Explaining Predictive Uncertainty with Information Theoretic Shapley Values" (Watson et al., 2023), SIG extends the Shapley framework to explain predictive uncertainty by quantifying each feature's contribution to the conditional entropy of individual model outputs. In "Variable Importance in Generalized Linear Models -- A Unifying View Using Shapley Values" (Acemoglu et al., 2 Jan 2026), SIG is the Shapley decomposition of a Kullback–Leibler-based pseudo-R2R^2. In "ProMed: Shapley Information Gain Guided Reinforcement Learning for Proactive Medical LLMs" (Ding et al., 19 Aug 2025), SIG is an importance-weighted gain in fact coverage used as a reinforcement-learning reward. In "ShaplEIG: Bayesian Experimental Design for Shapley Value Estimation" (Rundel et al., 1 Jun 2026), SIG is the expected information gain about the Shapley vector itself under a Gaussian-process surrogate. The shared structure is coalition-based attribution, but the object being valued differs substantially across these settings.

1. Conceptual scope

The literature uses the label “Shapley Information Gain” for several related but non-identical constructions. Each construction defines a characteristic function over subsets and then applies Shapley averaging, but the underlying target may be predictive entropy, pseudo-R2R^2, fact-level answer utility, or posterior uncertainty about Shapley values.

Setting Characteristic quantity SIG object
Predictive uncertainty vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S) Feature attribution to uncertainty
Generalized linear models v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)} Covariate importance
ProMed v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S) Reward for asking questions
ShaplEIG I(ϕ;yzDt)I(\phi;y_z\mid D_t) Acquisition criterion for coalition evaluation

This plurality matters technically. In the uncertainty setting, SIG explains why a model is uncertain at a specific xx. In the GLM setting, it decomposes model fit into relative and absolute importance. In ProMed, it weights newly acquired atomic facts by their contextual importance. In ShaplEIG, it is not a post hoc attribution of an existing prediction, but a Bayesian experimental-design criterion for deciding which coalition to evaluate next. A common misconception is therefore to treat SIG as a single universally fixed formula. The published usages instead define a family of Shapley-based information-gain functionals whose semantics depend on the value function.

2. Predictive-uncertainty SIG

For predictive uncertainty, let XRdX\in\mathbb{R}^d be the feature vector, YY the model output whose uncertainty is to be explained, and fix a test point xx. For any subset R2R^20, the payoff is defined as

R2R^21

where

R2R^22

Equivalent formulations use R2R^23, or the KL- or cross-entropy games R2R^24; these differ by R2R^25-independent constants and yield identical Shapley values (Watson et al., 2023).

With R2R^26, the attribution assigned to feature R2R^27 at R2R^28 is

R2R^29

Under vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)0, each marginal term is an entropy reduction: vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)1 Because vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)2, each marginal Shapley increment equals

vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)3

Thus the local attribution is an average conditional mutual information over coalitions.

Theoretical properties are explicit. Efficiency yields

vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)4

and under vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)5,

vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)6

the local mutual information. The conditional-independence characterization states that for each vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)7 and any vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)8, vIG(S,x):=H(YXS=xS)v_{IG}(S,x):=-H(Y\mid X_S=x_S)9 if and only if v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}0. Context-specific independence implies v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}1 at that v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}2, and the set of distributions for which v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}3 yet v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}4 fails has Lebesgue measure zero; the paper describes these as “conspiratorial” exceptions requiring exact cancellation of positive and negative log-ratios. SIG-Shapley also satisfies efficiency, symmetry, sensitivity, and linearity.

The finite-sample inference result uses split conformal bounds. With v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}5 i.i.d. samples, v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}6 is used for model fitting and v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}7 of size v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}8 for attribution. For target level v(S)=RKL(S)2v(S)=R^2_{\mathrm{KL}(S)}9, let v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)0 be the v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)1-th order statistic of v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)2 and v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)3 the v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)4-th. Then for a new v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)5,

v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)6

with exactness up to v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)7 when the joint of v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)8's is continuous. This provides finite-sample, marginal coverage for testing whether v(S)=logPθ(AQ,S)v(S)=\log P_\theta(A^*\mid Q,S)9 differs from zero.

Practical approximation replaces the I(ϕ;yzDt)I(\phi;y_z\mid D_t)0 subset sum by coalition sampling. The method chooses an entropy or variance estimator I(ϕ;yzDt)I(\phi;y_z\mid D_t)1, samples out-of-coalition features I(ϕ;yzDt)I(\phi;y_z\mid D_t)2 via a model-agnostic kernel, and samples I(ϕ;yzDt)I(\phi;y_z\mid D_t)3 coalitions from the Shapley weight I(ϕ;yzDt)I(\phi;y_z\mid D_t)4, often with paired sampling and linear-model variance reduction as in KernelSHAP. Williamson & Feng (2020) show that I(ϕ;yzDt)I(\phi;y_z\mid D_t)5 coalitions suffices for asymptotically optimal inference. The overall complexity is

I(ϕ;yzDt)I(\phi;y_z\mid D_t)6

and in practice I(ϕ;yzDt)I(\phi;y_z\mid D_t)7 yields accurate attributions for I(ϕ;yzDt)I(\phi;y_z\mid D_t)8 up to a few dozen.

The reported experiments span MNIST digit classification, IMDB sentiment analysis, covariate-shift detection on UCI tabular data, and active feature-value acquisition on a modified Friedman benchmark. In the semi-synthetic missing-data protocol, ranking features by I(ϕ;yzDt)I(\phi;y_z\mid D_t)9 yields xx0 even with xx1 missingness. Taken together, the method explains not just point predictions but full predictive uncertainty.

3. KL-xx2 SIG in generalized linear models

In generalized linear models, SIG is defined by decomposing a Kullback–Leibler-based pseudo-xx3 with Shapley values (Acemoglu et al., 2 Jan 2026). Let xx4 be observed responses and xx5 a regular one-parameter exponential family density,

xx6

The per-observation Kullback–Leibler divergence between xx7 and xx8 is

xx9

With a constant-only null model fitted mean XRdX\in\mathbb{R}^d0 and a model fitted mean XRdX\in\mathbb{R}^d1, the KL XRdX\in\mathbb{R}^d2 is

XRdX\in\mathbb{R}^d3

Key special cases are explicit: linear regression gives the classical XRdX\in\mathbb{R}^d4, binary logit gives McFadden’s XRdX\in\mathbb{R}^d5, and Poisson gives deviance-based XRdX\in\mathbb{R}^d6.

Let XRdX\in\mathbb{R}^d7 be the covariate indices and XRdX\in\mathbb{R}^d8 a coalition. The characteristic function is

XRdX\in\mathbb{R}^d9

The resulting SIG for covariate YY0 is

YY1

The paper emphasizes both relative and absolute importance. Classical Shapley axioms give efficiency, symmetry, dummy, and additivity. For regression interpretability, the additional requirements are monotonicity, the lower bound YY2, and the upper bound YY3 for the saturated model YY4. Because the KL YY5 satisfies these, YY6 and YY7. Relative importance is

YY8

and absolute importance is YY9.

Closed-form increments are given for linear, logistic, and Poisson regression. In linear regression this recovers Lindeman–Merenda–Gold. In logistic regression the increment is written in terms of McFadden’s xx0. In Poisson regression the unit deviance is

xx1

and xx2 is the explained fraction of total deviance relative to the null model.

Exact computation requires summing over all xx3 subsets for each xx4. The paper gives two practical strategies: brute-force for xx5, and permutation approximation for larger xx6 by averaging marginal contributions along xx7 random forward-selection paths. The reported examples include a Poisson doctor-visits model with full-model xx8, where xx9; a Poisson-hurdle insurance-claims model with separate SIG decompositions for the two parts; and a geometric-regression doctor-visits model with full R2R^200, where R2R^201. These examples illustrate that the framework is designed to interpret both fitted-model importance and importance relative to the saturated-model benchmark.

4. SIG-guided reinforcement learning in proactive medical LLMs

In ProMed, SIG is a reward for interactive medical questioning rather than a post hoc explanation of a conventional predictor (Ding et al., 19 Aug 2025). At dialogue turn R2R^202, after question R2R^203 and response R2R^204, the model’s understanding is R2R^205. Let R2R^206 be the full set of atomic facts. The raw information gain of R2R^207 is

R2R^208

Fact importance is then defined with a Shapley value over atomic facts. For any subset R2R^209,

R2R^210

and the Shapley value of fact R2R^211 is R2R^212. After softmax normalization,

R2R^213

the SIG reward becomes

R2R^214

The intended decomposition is explicit: quantity is measured by newly entailed facts, while contextual importance is measured by Shapley values that encode standalone utility and synergy or competition with other facts.

The computation of R2R^215 uses a Monte Carlo approximation over random permutations, with maximum iterations R2R^216 and tolerance R2R^217. After each question, the system elicits current understanding R2R^218 via a “doctor understanding” prompt, runs a fact-checker on each R2R^219, and computes R2R^220.

ProMed integrates SIG into a two-stage training pipeline. Stage 1, SIG-Guided Model Initialization, runs Monte Carlo Tree Search from partial input R2R^221, scores question nodes with R2R^222, retains the best answer-correct trajectory per case, and fine-tunes via supervised loss. A complete trajectory

R2R^223

receives

R2R^224

Stage 2, SIG-Augmented Policy Optimization, builds on Group Relative Policy Optimization and decomposes trajectory reward into action-level signals. Question R2R^225 receives

R2R^226

while the final answer receives

R2R^227

The case study in rheumatology centers on the missing key fact nail pitting. After training, the model asks, “Do you have any nail changes (pitting, onycholysis)?”, the fact-checker finds that R2R^228 newly appears in R2R^229, and if R2R^230 and no other facts are gained, then R2R^231. Experimentally, across three LLMs and two benchmarks, ProMed(Stage 1+2) achieves up to R2R^232 vs. second-best R2R^233 on MedQA and R2R^234 vs. R2R^235 on CMB, with average relative improvement over second-best R2R^236 and a R2R^237 gain over the direct baseline. Ablations that remove SIG or its components degrade performance by up to R2R^238 points.

5. SIG as expected information gain for adaptive coalition selection

ShaplEIG uses SIG as an acquisition criterion for estimating Shapley values when value-function evaluations are expensive (Rundel et al., 1 Jun 2026). Let R2R^239 be the player set, R2R^240 the expensive value of coalition R2R^241, and R2R^242 the vector of Shapley values for R2R^243. At iteration R2R^244, with data

R2R^245

where each R2R^246 encodes a coalition and R2R^247, R2R^248, the next coalition is chosen to maximize

R2R^249

Here the quantity is the expected reduction in differential entropy about the Shapley vector after observing R2R^250.

A Gaussian-process prior is placed on R2R^251 over R2R^252, with zero mean and a Hamming-distance kernel. Writing R2R^253 for all coalitions, the posterior is

R2R^254

By linearity of Shapley values,

R2R^255

where R2R^256 is the fixed matrix of Shapley weights. Hence

R2R^257

Under this linear-Gaussian structure, mutual information is available in closed form. For a one-point design R2R^258 with observation R2R^259, SIG simplifies to

R2R^260

where R2R^261 and R2R^262. The paper states that R2R^263 is the marginal posterior variance at R2R^264, and R2R^265 reflects the variance after conditioning on R2R^266.

Naïve evaluation is exponential in R2R^267, because it would require manipulating R2R^268 matrices. ShaplEIG reduces this to polynomial complexity by exploiting the product structure of the Hamming kernel and the combinatorial structure of R2R^269. Elementary symmetric polynomials are used for the linear term R2R^270, and bivariate generating polynomials with pre- and suffix table convolutions are used for the quadratic term R2R^271. The resulting complexity is R2R^272 for a single candidate and R2R^273 for a batch of R2R^274 candidates.

The practical loop alternates GP hyperparameter fitting, SIG evaluation over candidate coalitions, and selection of R2R^275. After R2R^276 costly evaluations, the posterior mean R2R^277 yields consistent Shapley-value estimates. Empirically, across nine real-world costly games with R2R^278 players, ShaplEIG outperforms or matches stochastic, surrogate-based, and fixed-design Bayesian experimental-design baselines, with gains largest in the low-budget regime R2R^279.

6. Common structure, distinctions, and recurrent themes

Across these formulations, SIG is always built from subset-based evaluation and Shapley aggregation, but the meaning of “information gain” changes with the task. In predictive uncertainty, it is a reduction in local conditional entropy and is directly identified with conditional mutual information (Watson et al., 2023). In generalized linear models, it is the Shapley decomposition of increments in R2R^280, a goodness-of-fit quantity normalized to R2R^281 (Acemoglu et al., 2 Jan 2026). In ProMed, it is the importance-weighted gain in entailed atomic facts that guides question asking (Ding et al., 19 Aug 2025). In ShaplEIG, it is mutual information between a future coalition observation and the Shapley vector under a GP posterior (Rundel et al., 1 Jun 2026).

This suggests a useful unifying view: SIG is less a single estimator than a design pattern in which a coalition value function is chosen to represent uncertainty reduction, goodness-of-fit gain, clinical fact utility, or posterior learning about attributions. The design choice that determines the interpretation is the characteristic function R2R^282, not the Shapley operator itself.

Several recurrent technical themes also appear across the literature. Exact computation is combinatorial or exponential unless additional structure is exploited. The uncertainty paper uses coalition sampling from the Shapley weight and model-agnostic reference distributions; the GLM paper uses brute-force or permutation approximation; ProMed uses Monte Carlo permutations over atomic facts; ShaplEIG obtains polynomial-time evaluation through Gaussian-process linearity and elementary symmetric polynomials. Another recurrent theme is that SIG is not restricted to explanation in the narrow sense. The published applications include covariate-shift detection, active learning, feature selection, active feature-value acquisition, proactive medical dialogue, data valuation, hyperparameter importance, and local explanations.

The term therefore carries both continuity and ambiguity. The continuity lies in Shapley-based averaging of marginal gains under a carefully chosen information-centric objective. The ambiguity lies in the fact that different papers use “SIG” for substantively different targets. For technical reading, the decisive question is always which random variable, fit criterion, fact set, or posterior uncertainty the information gain is defined over.

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 Shapley Information Gain (SIG).