Explanations of Black-Box Models based on Directional Feature Interactions

Aria Masoomi, Davin Hill, Zhonghui Xu, Craig P Hersh, Edwin K. Silverman +3 more
4/16/2023
cs.LG

No official code found. Best implementation: davinhill/BivariateShapley (20 stars, MIT, last commit 2025-05-19).

Abstract

As machine learning algorithms are deployed ubiquitously to a variety of domains, it is imperative to make these often black-box models transparent. Several recent works explain black-box models by capturing the most influential features for prediction per instance; such explanation methods are univariate, as they characterize importance per feature. We extend univariate explanation to a higher-order; this enhances explainability, as bivariate methods can capture feature interactions in black-box models, represented as a directed graph. Analyzing this graph enables us to discover groups of features that are equally important (i.e., interchangeable), while the notion of directionality allows us to identify the most influential features. We apply our bivariate method on Shapley value explanations, and experimentally demonstrate the ability of directional explanations to discover feature interactions. We show the superiority of our method against state-of-the-art on CIFAR10, IMDB, Census, Divorce, Drug, and gene data.

View on arXivView PDF

Code Implementations(1)

MIT

Bivariate Shapley is a Shapley-based method of identifying directional feature interactions and feature redundancy

202PythonMay 31, 20211 years agoMIT

Similar Papers

Cite this paper

@article{masoomi2023explanations,
  title  = {Explanations of Black-Box Models based on Directional Feature Interactions},
  author = {Aria Masoomi and Davin Hill and Zhonghui Xu and Craig P Hersh and Edwin K. Silverman and Peter J. Castaldi and Stratis Ioannidis and Jennifer Dy},
  year   = {2023},
  eprint = {2304.07670},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2304.07670v1},
  journal = {ICLR 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2304.07670.svg)](https://codeofpaper.com/paper/2304.07670)

Discussion