Effective pruning for top-k feature search on the basis of SHAP values - PEPR Santé numérique
Article Dans Une Revue IEEE Access Année : 2024

Effective pruning for top-k feature search on the basis of SHAP values

Résumé

With the increasingly pervasive use of advanced machine learning models comes the need to explain their predictions. The SHAP framework, based on Shapley values, provides explanations to highlight which features could be important for a given prediction. However, its use is hampered by its computational cost, especially in its model-agnostic formulation. Model-specific algorithms offer a restricted solution to this problem, whereas alternative approximation strategies can maintain the model-agnostic property. Here we propose TopShap as an agnostic algorithm that searches the k most important features by interleaving pruning of candidates and refinement of the approximate SHAP values. TopShap is built on three insights: (i) it performs an iterative approximation taking advantage of a previously developed sampling strategy, (ii) it uses confidence interval bounds around approximate SHAP values to determine on-the-fly which features can no longer be part of the top-k, (iii) it stops when these interval bounds are stable. Evaluating TopShap on publicly available datasets shows it performs an effective pruning of the feature search space and leads to an important reduction of the execution cost when compared to the other agnostic approaches. We also apply TopShap to a use case in biology and show that top-k search is meaningful in this context.
Fichier principal
Vignette du fichier
2024_Effective_Pruning_for_Top-k_Feature_Search_on_the_Basis_of_SHAP_Values.pdf (3.2 Mo) Télécharger le fichier
Origine Publication financée par une institution
licence

Dates et versions

hal-04549416 , version 1 (17-04-2024)
hal-04549416 , version 2 (28-11-2024)

Licence

Identifiants

Citer

Lisa Chabrier, Anton Crombach, Sergio Peignier, Christophe Rigotti. Effective pruning for top-k feature search on the basis of SHAP values. IEEE Access, 2024, 12, pp.163079-163092. ⟨10.1109/ACCESS.2024.3489958⟩. ⟨hal-04549416v2⟩
368 Consultations
85 Téléchargements

Altmetric

Partager

More