We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: Interpretable feature subset selection: A Shapley value based approach

Abstract: For feature selection and related problems, we introduce the notion of classification game, a cooperative game, with features as players and hinge loss based characteristic function and relate a feature's contribution to Shapley value based error apportioning (SVEA) of total training error. Our major contribution is ($\star$) to show that for any dataset the threshold 0 on SVEA value identifies feature subset whose joint interactions for label prediction is significant or those features that span a subspace where the data is predominantly lying. In addition, our scheme ($\star$) identifies the features on which Bayes classifier doesn't depend but any surrogate loss function based finite sample classifier does; this contributes to the excess $0$-$1$ risk of such a classifier, ($\star$) estimates unknown true hinge risk of a feature, and ($\star$) relate the stability property of an allocation and negative valued SVEA by designing the analogue of core of classification game. Due to Shapley value's computationally expensive nature, we build on a known Monte Carlo based approximation algorithm that computes characteristic function (Linear Programs) only when needed. We address the potential sample bias problem in feature selection by providing interval estimates for SVEA values obtained from multiple sub-samples. We illustrate all the above aspects on various synthetic and real datasets and show that our scheme achieves better results than existing recursive feature elimination technique and ReliefF in most cases. Our theoretically grounded classification game in terms of well defined characteristic function offers interpretability (which we formalize in terms of final task) and explainability of our framework, including identification of important features.
Comments: A shorter version of this work appeared in a special session titled Explainable AI at IEEE BigData'20 conference. More experiments and a new notion of interpretable FSS introduced in this version. Earlier plots for sample bias robustness are corrected and updated
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:2001.03956 [stat.ML]
  (or arXiv:2001.03956v3 [stat.ML] for this version)

Submission history

From: Sandhya Tripathi [view email]
[v1] Sun, 12 Jan 2020 16:27:08 GMT (489kb,D)
[v2] Sun, 22 Nov 2020 19:28:58 GMT (612kb,D)
[v3] Sun, 25 Apr 2021 19:24:45 GMT (613kb,D)

Link back to: arXiv, form interface, contact.