arXiv · 2105.02410
Partially Interpretable Estimators (PIE): Black-Box-Refined Interpretable Machine Learning
Abstract
We propose Partially Interpretable Estimators (PIE) which attribute a prediction to individual features via an interpretable model, while a (possibly) small part of the PIE prediction is attributed to the interaction of features via a black-box model, with the goal to boost the predictive performance while maintaining interpretability. As such, the interpretable model captures the main contributions of features, and the black-box model attempts to complement the interpretable piece by capturing the "nuances" of feature interactions as a refinement. We design an iterative training algorithm to jointly train the two types of models. Experimental results show that PIE is highly competitive to black-box models while outperforming interpretable baselines. In addition, the understandability of PIE is comparable to simple linear models as validated via a human evaluation.
Explore related subjects
Keep this discovery
Tong Wang, Jingyi Yang, Yunyi Li, Boxiang Wang. 2021-05-06. Partially Interpretable Estimators (PIE): Black-Box-Refined Interpretable Machine Learning. https://arxiv.org/abs/2105.02410
Cite the original work for its findings. Save a collection to share your selection of sources.