arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2105.02410cs.LGcs.AI
收藏

Partially Interpretable Estimators (PIE): Black-Box-Refined Interpretable Machine Learning

University of Iowa(爱荷华大学);University of Texas at Austin(德克萨斯大学奥斯汀分校)

Tong Wang, Jingyi Yang, Yunyi Li, Boxiang Wang

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑