RegressionFSII

Faithful Shapley Interaction Index via regression using RegressionFSII Fumagalli et al.[1].

from __future__ import annotations

import numpy as np

import shapiq

N_PLAYERS = 8
BUDGET = 200
feature_names = [f"x{i}" for i in range(N_PLAYERS)]

weights = np.array([0.4, 0.3, 0.2, 0.1, 0.05, -0.1, -0.2, -0.3])


def game_fun(coalitions: np.ndarray) -> np.ndarray:
    coalitions = np.atleast_2d(coalitions)
    return (coalitions @ weights) + 0.5 * coalitions[:, 0] * coalitions[:, 1]

Approximate FSII values

approximator = shapiq.RegressionFSII(n=N_PLAYERS, max_order=2, random_state=42)
iv = approximator.approximate(BUDGET, game_fun)
print(iv)

InteractionValues(
    index=FSII, max_order=2, min_order=0, estimated=True, estimation_budget=200,
    n_players=8, baseline_value=0.0,
    Top 10 interactions:
        (0, 1): 0.49999985919524687
        (0,): 0.39999993296127234
        (1,): 0.29999983738348335
        (2,): 0.20000000007673918
        (3,): 0.10000006674255016
        (4,): 0.05000009032363183
        (2, 6): 2.261603439959434e-07
        (5,): -0.09999998225449135
        (6,): -0.2000000353870478
        (7,): -0.30000008137332523
)

Force plot

iv.plot_force(feature_names=feature_names)

Network plot

iv.plot_network(feature_names=feature_names)

References

[1]

Fabian Fumagalli, Maximilian Muschalik, Patrick Kolpaczki, Eyke Hüllermeier, and Barbara Hammer. SHAP-IQ: unified approximation of any-order shapley interactions. In Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, (NeurIPS 2023). 2023. URL: http://papers.nips.cc/paper\_files/paper/2023/hash/264f2e10479c9370972847e96107db7f-Abstract-Conference.html.