Skip to main navigation Skip to search Skip to main content

Feature Selection Using Approximated High-Order Interaction Components of the Shapley Value for Boosted Tree Classifier

  • Carlin Chun Fai Chu
  • , David Po Kin Chan

Research output: Contribution to journal › Article › peer-review

31 Citations (Scopus)

Abstract

The Shapely value originates from the cooperative game theory and has been widely used in solving machine learning problems due to its high interpretability and consistency. The typical Shapley value evaluates the importance score of a feature as its average marginal contribution to a fully parameterized model under all possible feature combinations; as a result, its value includes the influences from both selected features and unselected ones. To better separate the corresponding sources, it is suggested to decompose the Shapley value into high-order interaction effect components such that the influences from each individual feature can be effectively untangled. A feature's contribution is decomposed into distinct interaction components, and each component corresponds to a joint contribution resulting from a particular feature combination. The feature ranking is therefore evaluated with respect to the selected feature subset and calculated as the total incremental contribution by summing up its corresponding decomposed interaction values. In this study, a computationally efficient model-dependent greedy search algorithm on the high-order interaction components is proposed to solve an optimal subset selection problem with hundreds of time-lagged interrelated input features. Our algorithm extends a recently developed low-order polynomial time method for calculating the interaction component values. The empirical analysis demonstrates that the proposed method always outperforms other methods that are based on the typical Shapley value, the gain or the split count criteria, in terms of in-sample representativeness and out-of-sample forecasting performance for handling a problem with hundreds of time-lagged input features.

Original languageEnglish
Article number9117109
Pages (from-to)112742-112750
Number of pages9
JournalIEEE Access
Volume8
DOIs
Publication statusPublished - 2020

Keywords

  • Computational intelligence
  • feature selection
  • machine learning
  • Shapley value

Fingerprint

Dive into the research topics of 'Feature Selection Using Approximated High-Order Interaction Components of the Shapley Value for Boosted Tree Classifier'. Together they form a unique fingerprint.

Cite this