Shapley Information Gain (SIG) Overview
- 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-. 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-, fact-level answer utility, or posterior uncertainty about Shapley values.
| Setting | Characteristic quantity | SIG object |
|---|---|---|
| Predictive uncertainty | Feature attribution to uncertainty | |
| Generalized linear models | Covariate importance | |
| ProMed | Reward for asking questions | |
| ShaplEIG | Acquisition criterion for coalition evaluation |
This plurality matters technically. In the uncertainty setting, SIG explains why a model is uncertain at a specific . 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 be the feature vector, the model output whose uncertainty is to be explained, and fix a test point . For any subset 0, the payoff is defined as
1
where
2
Equivalent formulations use 3, or the KL- or cross-entropy games 4; these differ by 5-independent constants and yield identical Shapley values (Watson et al., 2023).
With 6, the attribution assigned to feature 7 at 8 is
9
Under 0, each marginal term is an entropy reduction: 1 Because 2, each marginal Shapley increment equals
3
Thus the local attribution is an average conditional mutual information over coalitions.
Theoretical properties are explicit. Efficiency yields
4
and under 5,
6
the local mutual information. The conditional-independence characterization states that for each 7 and any 8, 9 if and only if 0. Context-specific independence implies 1 at that 2, and the set of distributions for which 3 yet 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 5 i.i.d. samples, 6 is used for model fitting and 7 of size 8 for attribution. For target level 9, let 0 be the 1-th order statistic of 2 and 3 the 4-th. Then for a new 5,
6
with exactness up to 7 when the joint of 8's is continuous. This provides finite-sample, marginal coverage for testing whether 9 differs from zero.
Practical approximation replaces the 0 subset sum by coalition sampling. The method chooses an entropy or variance estimator 1, samples out-of-coalition features 2 via a model-agnostic kernel, and samples 3 coalitions from the Shapley weight 4, often with paired sampling and linear-model variance reduction as in KernelSHAP. Williamson & Feng (2020) show that 5 coalitions suffices for asymptotically optimal inference. The overall complexity is
6
and in practice 7 yields accurate attributions for 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 9 yields 0 even with 1 missingness. Taken together, the method explains not just point predictions but full predictive uncertainty.
3. KL-2 SIG in generalized linear models
In generalized linear models, SIG is defined by decomposing a Kullback–Leibler-based pseudo-3 with Shapley values (Acemoglu et al., 2 Jan 2026). Let 4 be observed responses and 5 a regular one-parameter exponential family density,
6
The per-observation Kullback–Leibler divergence between 7 and 8 is
9
With a constant-only null model fitted mean 0 and a model fitted mean 1, the KL 2 is
3
Key special cases are explicit: linear regression gives the classical 4, binary logit gives McFadden’s 5, and Poisson gives deviance-based 6.
Let 7 be the covariate indices and 8 a coalition. The characteristic function is
9
The resulting SIG for covariate 0 is
1
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 2, and the upper bound 3 for the saturated model 4. Because the KL 5 satisfies these, 6 and 7. Relative importance is
8
and absolute importance is 9.
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 0. In Poisson regression the unit deviance is
1
and 2 is the explained fraction of total deviance relative to the null model.
Exact computation requires summing over all 3 subsets for each 4. The paper gives two practical strategies: brute-force for 5, and permutation approximation for larger 6 by averaging marginal contributions along 7 random forward-selection paths. The reported examples include a Poisson doctor-visits model with full-model 8, where 9; a Poisson-hurdle insurance-claims model with separate SIG decompositions for the two parts; and a geometric-regression doctor-visits model with full 00, where 01. 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 02, after question 03 and response 04, the model’s understanding is 05. Let 06 be the full set of atomic facts. The raw information gain of 07 is
08
Fact importance is then defined with a Shapley value over atomic facts. For any subset 09,
10
and the Shapley value of fact 11 is 12. After softmax normalization,
13
the SIG reward becomes
14
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 15 uses a Monte Carlo approximation over random permutations, with maximum iterations 16 and tolerance 17. After each question, the system elicits current understanding 18 via a “doctor understanding” prompt, runs a fact-checker on each 19, and computes 20.
ProMed integrates SIG into a two-stage training pipeline. Stage 1, SIG-Guided Model Initialization, runs Monte Carlo Tree Search from partial input 21, scores question nodes with 22, retains the best answer-correct trajectory per case, and fine-tunes via supervised loss. A complete trajectory
23
receives
24
Stage 2, SIG-Augmented Policy Optimization, builds on Group Relative Policy Optimization and decomposes trajectory reward into action-level signals. Question 25 receives
26
while the final answer receives
27
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 28 newly appears in 29, and if 30 and no other facts are gained, then 31. Experimentally, across three LLMs and two benchmarks, ProMed(Stage 1+2) achieves up to 32 vs. second-best 33 on MedQA and 34 vs. 35 on CMB, with average relative improvement over second-best 36 and a 37 gain over the direct baseline. Ablations that remove SIG or its components degrade performance by up to 38 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 39 be the player set, 40 the expensive value of coalition 41, and 42 the vector of Shapley values for 43. At iteration 44, with data
45
where each 46 encodes a coalition and 47, 48, the next coalition is chosen to maximize
49
Here the quantity is the expected reduction in differential entropy about the Shapley vector after observing 50.
A Gaussian-process prior is placed on 51 over 52, with zero mean and a Hamming-distance kernel. Writing 53 for all coalitions, the posterior is
54
By linearity of Shapley values,
55
where 56 is the fixed matrix of Shapley weights. Hence
57
Under this linear-Gaussian structure, mutual information is available in closed form. For a one-point design 58 with observation 59, SIG simplifies to
60
where 61 and 62. The paper states that 63 is the marginal posterior variance at 64, and 65 reflects the variance after conditioning on 66.
Naïve evaluation is exponential in 67, because it would require manipulating 68 matrices. ShaplEIG reduces this to polynomial complexity by exploiting the product structure of the Hamming kernel and the combinatorial structure of 69. Elementary symmetric polynomials are used for the linear term 70, and bivariate generating polynomials with pre- and suffix table convolutions are used for the quadratic term 71. The resulting complexity is 72 for a single candidate and 73 for a batch of 74 candidates.
The practical loop alternates GP hyperparameter fitting, SIG evaluation over candidate coalitions, and selection of 75. After 76 costly evaluations, the posterior mean 77 yields consistent Shapley-value estimates. Empirically, across nine real-world costly games with 78 players, ShaplEIG outperforms or matches stochastic, surrogate-based, and fixed-design Bayesian experimental-design baselines, with gains largest in the low-budget regime 79.
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 80, a goodness-of-fit quantity normalized to 81 (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 82, 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.