ExplainabilityTransparencyAccountabilitycs.LG
Explainable AI for Trees: From Local Explanations to Global Understanding
Scott M. Lundberg, Gabriel Erion, Hugh Chen, Alex DeGrave, Jordan M. Prutkin, Bala Nair, Ronit Katz, Jonathan Himmelfarb, Nisha Bansal, Su-In Lee
May 11, 2019
arXiv.org
405 citations
31 influential
Key Takeaways
-
Game-theoretic explanations make tree models more trustworthy: The paper introduces the first polynomial-time algorithm for computing optimal explanations of tree-based model predictions using game theory principles, making it feasible to understand why these widely-used models make specific decisions in real time.
-
Feature interactions reveal hidden risk factors: The research identifies previously overlooked non-linear interactions between risk factors—for example, discovering that certain combinations of health factors create disproportionately high mortality risks that wouldn't be apparent from examining individual factors alone, with direct applications to identifying vulnerable population subgroups.
-
Local explanations can scale to reveal global patterns: By aggregating many individual prediction explanations, the authors demonstrate that organizations can understand overall model behavior and structure while maintaining accuracy at the individual case level—enabling hospitals to monitor whether deployed models are degrading over time as patient populations change.
-
Practical medical applications demonstrate real-world value: The paper demonstrates concrete use cases including identifying mortality risk factors in the US population, detecting distinct patient subgroups with shared characteristics, and flagging which features are causing performance problems in deployed hospital systems—showing that better interpretability directly supports clinical decision-making.
-
Addresses a critical gap in AI transparency: Despite tree-based models being the most commonly used machine learning approach in practice, they have received far less attention for explainability compared to other model types, making this work particularly important for responsible AI deployment across industries.