Peeking Inside the Black Box: Visualizing Statistical Learning With Plots of Individual Conditional Expectation

By A. Goldstein, Adam Kapelner, Justin Bleich, E. Pitkin. 2013

1,731 citations recorded by the source corpus.

This article presents individual conditional expectation (ICE) plots, a tool for visualizing the model estimated by any supervised learning algorithm. Classical partial dependence plots (PDPs) help visualize the average partial relationship between the predicted response and one or more features. In the presence of substantial interaction effects, the partial response relationship can be heterogeneous. Thus, an average curve, such as the PDP, can obfuscate the complexity of the modeled relationship. Accordingly, ICE plots refine the PDP by graphing the functional relationship between the predicted response and the feature for individual observations. Specifically, ICE plots highlight the variation in the fitted values across the range of a covariate, suggesting where and to what extent heterogeneities might exist. In addition to providing a plotting suite for exploratory analysis, we include a visual test for additive structure in the data-generating model. Through simulated examples and real datasets, we demonstrate how ICE plots can shed light on estimated models in ways PDPs cannot. Procedures outlined are available in the R package ICEbox.

Subfields: Artificial Intelligence, Experimental and Cognitive Psychology, Statistics and Probability, Safety, Robustness & Interpretability

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Longo Luca
  2. Giulia Vilone
  3. Q. Liao
  4. Dino Pedreschi
  5. Riccardo Guidotti
  6. Fosca Giannotti
  7. Marco F. Huber
  8. Anna Monreale

In Experimental and Cognitive Psychology

  1. Przemyslaw Biecek
  2. Vaishak Belle
  3. Q. Liao
  4. Giles Hooker
  5. Nadia Burkart
  6. Bernd Bischl
  7. Francisco Herrera
  8. Kush R. Varshney

In Statistics and Probability

  1. R. Caruana
  2. Lucas K. Mentch
  3. Sarah Tan
  4. Giles Hooker
  5. John Fox
  6. Francisco Herrera
  7. Giulia Vilone
  8. Longo Luca

In Safety, Robustness & Interpretability

  1. Longo Luca
  2. Giulia Vilone
  3. Dino Pedreschi
  4. Riccardo Guidotti
  5. Fosca Giannotti
  6. Mohammed Berrada
  7. Amina Adadi
  8. Anna Monreale

Citations in the corpus