← Benchmark resultsMarkdown source

About the benchmark

22 estimators · 24 datasets · 2,048 games. We compare contribution estimates for predictions, training data and feature selection.

Games, Shapley values and estimation error

\[\phi_i = \sum_{S \subseteq N \setminus \{i\}} \frac{|S|!\,(d - |S| - 1)!}{d!}\left[f(S \cup \{i\}) - f(S)\right].\]
\[\mathbf{nMSE} = \frac{\lVert \hat{\boldsymbol{\phi}} - \boldsymbol{\phi} \rVert_2^2}{\lVert \boldsymbol{\phi} \rVert_2^2}.\]

Lower is better: zero is exact; one is the error of estimating every contribution as zero. Zero or negligible reference vectors are excluded. For interaction targets, the vectors contain the requested interaction coefficients; the empty-coalition baseline is excluded. Main effects and pairs can also be viewed separately.

Explaining an individual prediction

Players are input features. Fit a predictor \(h\) once and explain one held-out point \(x\). For the small games, keep \(x\)'s selected features and replace the others with fixed training-background means \(b\):

\[f(S) = h(z(S)) - h(b), \qquad z_i(S) = \begin{cases}x_i & i \in S,\\ b_i & i \notin S.\end{cases}\]

The output is a fixed class probability or a regression prediction. Exact reference: evaluate all \(2^d\) coalitions.

Tree games extend this to larger player sets. Interventional games average predictions with missing features filled from fixed background rows. Path-dependent games average missing branches using training-path weights. Matching exact tree solvers supply the references. Both tree constructions use random forests only.

Models: linear/logistic regression and random forest for baseline games, with nested feature sets of at most 12 players. RF tree games use 8, 16 and 32 features where available. A dataset's full width below 12 is also included in the baseline grid. Tables show selected player counts; “—” means no selected recipe.

Dataset coverage
Dataset Baseline features (linear / RF) RF tree features (path / interventional)
Adult Census 4, 8, 12 / 4, 8, 12 8 / 8
Mushroom 4, 8, 12 / 4, 8, 12 8, 16 / 8, 16
Ionosphere 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
Bioresponse — / 4, 8, 12 8, 16, 32 / 8, 16, 32
Breast Cancer 4, 8, 12 / 4, 8, 12 8, 16 / 8, 16
Digits 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
Wine (classification) 4, 8, 12 / 4, 8, 12 8 / 8
Amazon Employee Access 8, 9 / 8, 9 8 / 8
APS Failure 4, 8, 12 / 4, 12 16, 32 / 16, 32
Anneal 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
Splice 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
Credit Card Default 4, 8, 12 / 4, 8, 12 8, 16 / 8, 16
Wine Quality 4, 8, 12 / 4, 8, 12 8 / 8
Communities and Crime 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
NHANES I — / 8, 12 8, 32 / 8, 32
QSAR-TID11 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
Miami Housing 4, 8, 12 / 4, 8, 12 8 / 8
Superconductivity 4, 8, 12 / 4, 8, 12 8, 16, 32 / 8, 16, 32
California Housing 4 / 4, 8 8 / 8
Diabetes 4, 8, 10 / 4, 8, 10 8 / 8
Bike Sharing 4, 8, 12 / 4, 8, 12 8 / 8
Airfoil Self Noise 4, 5 / 4, 5 — / —
Concrete Strength 4, 8 / 4, 8 8 / 8
Protein 4, 8, 9 / 4, 8, 9 8 / 8

Valuing training data

Players are groups of training rows. For a coalition \(S\), fit \(h_S\) on the union of its groups and evaluate on a fixed held-out set \(T\):

\[f(S) = \begin{cases}\operatorname{accuracy}(h_S, T) & \text{classification},\\ -\operatorname{MSE}(h_S, T) & \text{regression},\end{cases} \qquad f(\varnothing) = 0.\]

Exact reference: refit and score every coalition of 4, 8 or 12 groups. Both linear/logistic regression and random forest use exactly \(\min(12, \text{dataset width})\) input features. There is no input-width sweep; input width is separate from the number of players.

Neighbor games use individual training rows as players for a fixed test example. KNN utility counts matching labels among the coalition's nearest \(\min(k, |S|)\) rows, divided by fixed \(k\). TNN utility is the matching-label fraction within a fixed radius, or \(1 / \text{number of classes}\) when no rows are present. Selected classification games use 32 or 64 players, with exact native Shapley references. All input selections are fixed before estimator evaluation.

Dataset coverage
Dataset Training-row groups (linear / RF) Neighbor players (KNN / TNN) Input features
Adult Census 4, 8, 12 / 4, 8, 12 64 / 64 12
Mushroom 4, 8, 12 / 4, 8, 12 32 / 32, 64 12
Ionosphere 4, 8, 12 / 4, 8, 12 32, 64 / 64 12
Bioresponse — / 4, 8, 12 32, 64 / 32 12
Breast Cancer 4, 8, 12 / 4, 8, 12 64 / 32, 64 12
Digits 4, 8, 12 / 4, 8, 12 32 / 32, 64 12
Wine (classification) 4, 8, 12 / 4, 8, 12 32, 64 / 64 12
Amazon Employee Access 4, 8, 12 / 4, 8, 12 32, 64 / 32 9
APS Failure 4, 8, 12 / 4, 8, 12 64 / 32, 64 12
Anneal 4, 8, 12 / 4, 8, 12 32 / 32, 64 12
Splice 4, 8, 12 / 4, 8, 12 32, 64 / 64 12
Credit Card Default 4, 8, 12 / 4, 8, 12 32, 64 / 32 12
Wine Quality 4, 8, 12 / 4, 8, 12 — / — 12
Communities and Crime 4, 8, 12 / 4, 8, 12 — / — 12
NHANES I — / 4, 8, 12 — / — 12
QSAR-TID11 4, 8, 12 / 4, 8, 12 — / — 12
Miami Housing 4, 8, 12 / 4, 8, 12 — / — 12
Superconductivity 4, 8, 12 / 4, 8, 12 — / — 12
California Housing — / 4, 8, 12 — / — 8
Diabetes 4, 8, 12 / 4, 8, 12 — / — 10
Bike Sharing 4, 8, 12 / 4, 8, 12 — / — 12
Airfoil Self Noise 4, 8, 12 / 4, 8, 12 — / — 5
Concrete Strength 4, 8, 12 / 4, 8, 12 — / — 8
Protein 4, 8, 12 / 4, 8, 12 — / — 9

Selecting useful features

Players are input features. For every coalition \(S\), fit a fresh predictor \(h_S\) using only those columns, keeping the training and held-out rows fixed:

\[f(S) = \begin{cases}\operatorname{accuracy}(h_S, T_S) & \text{classification},\\ -\operatorname{MSE}(h_S, T_S) & \text{regression},\end{cases} \qquad f(\varnothing) = 0.\]

Exact reference: fit and score all \(2^d\) coalitions. Unlike an individual-prediction game, this measures how well a model can learn from the chosen features.

Models: linear/logistic regression and random forest, with the same nested feature grids of at most 12 players used by the baseline prediction games.

Dataset coverage
Dataset Linear features RF features
Adult Census 4, 8, 12 4, 8, 12
Mushroom 4, 8, 12 4, 8, 12
Ionosphere 4, 8, 12 4, 8, 12
Bioresponse — 4, 8, 12
Breast Cancer 4, 8, 12 4, 8, 12
Digits 4, 8, 12 4, 8, 12
Wine (classification) 4, 8, 12 4, 8, 12
Amazon Employee Access 8, 9 8, 9
APS Failure 4, 8, 12 4, 12
Anneal 4, 8, 12 4, 8, 12
Splice 4, 8, 12 4, 8, 12
Credit Card Default 4, 8, 12 4, 8, 12
Wine Quality 4, 8, 12 4, 8, 12
Communities and Crime 4, 8, 12 4, 8, 12
NHANES I — 8, 12
QSAR-TID11 4, 8, 12 4, 8, 12
Miami Housing 4, 8, 12 4, 8, 12
Superconductivity 4, 8, 12 4, 8, 12
California Housing 4 4, 8
Diabetes 4, 8, 10 4, 8, 10
Bike Sharing 4, 8, 12 4, 8, 12
Airfoil Self Noise 4, 5 4, 5
Concrete Strength 4, 8 4, 8
Protein 4, 8, 9 4, 8, 9

Comparing and reproducing results

Estimator combinations

For each target and budget, we fit signed ridge regression with no intercept:

\[\hat{\boldsymbol{\phi}}_{\alpha} = \sum_{m=1}^{M} \alpha_m \hat{\boldsymbol{\phi}}^{(m)}.\]