English

Leveraging Black-box Models to Assess Feature Importance in Unconditional Distribution

Machine Learning 2024-12-10 v1 Machine Learning Computation Methodology

Abstract

Understanding how changes in explanatory features affect the unconditional distribution of the outcome is important in many applications. However, existing black-box predictive models are not readily suited for analyzing such questions. In this work, we develop an approximation method to compute the feature importance curves relevant to the unconditional distribution of outcomes, while leveraging the power of pre-trained black-box predictive models. The feature importance curves measure the changes across quantiles of outcome distribution given an external impact of change in the explanatory features. Through extensive numerical experiments and real data examples, we demonstrate that our approximation method produces sparse and faithful results, and is computationally efficient.

Keywords

Cite

@article{arxiv.2412.05759,
  title  = {Leveraging Black-box Models to Assess Feature Importance in Unconditional Distribution},
  author = {Jing Zhou and Chunlin Li},
  journal= {arXiv preprint arXiv:2412.05759},
  year   = {2024}
}