Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

All Models are Wrong, but Many are Useful: Learning a Variable's Importance by Studying an Entire Class of Prediction Models Simultaneously

About

Variable importance (VI) tools describe how much covariates contribute to a prediction model's accuracy. However, important variables for one well-performing model (for example, a linear model $f(\mathbf{x})=\mathbf{x}^{T}\beta$ with a fixed coefficient vector $\beta$) may be unimportant for another model. In this paper, we propose model class reliance (MCR) as the range of VI values across all well-performing model in a prespecified class. Thus, MCR gives a more comprehensive description of importance by accounting for the fact that many prediction models, possibly of different parametric forms, may fit the data well. In the process of deriving MCR, we show several informative results for permutation-based VI estimates, based on the VI measures used in Random Forests. Specifically, we derive connections between permutation importance estimates for a single prediction model, U-statistics, conditional variable importance, conditional causal effects, and linear model coefficients. We then give probabilistic bounds for MCR, using a novel, generalizable technique. We apply MCR to a public data set of Broward County criminal records to study the reliance of recidivism prediction models on sex and race. In this application, MCR can be used to help inform VI for unknown, proprietary models.

Aaron Fisher, Cynthia Rudin, Francesca Dominici• 2018

Related benchmarks

TaskDatasetResultRank
Click-Through Rate PredictionAvazu
Logloss0.4006
60
Click-Through Rate PredictionCriteo
AUC77.186
44
Click-Through Rate PredictionICM 500
AUC84.874
30
Click-Through Rate PredictionICM 150
AUC84.64
30
Showing 4 of 4 rows

Other info

Follow for update