arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.14528stat.MLcs.LG

Surrogate Locally-Interpretable Models with Supervised Machine Learning Algorithms

Corporate Model Risk, Wells Fargo(富国银行企业模型风险部门)

Linwei Hu, Jie Chen, Vijayan N. Nair, Agus Sudjianto

英文摘要:

Supervised Machine Learning (SML) algorithms, such as Gradient Boosting, Random Forest, and Neural Networks, have become popular in recent years due to their superior predictive performance over traditional statistical methods. However, their complexity makes the results hard to interpret without additional tools. There has been a lot of recent work in developing global and local diagnostics for interpreting SML models. In this paper, we propose a locally-interpretable model that takes the fitted ML response surface, partitions the predictor space using model-based regression trees, and fits interpretable main-effects models at each of the nodes. We adapt the algorithm to be efficient in dealing with high-dimensional predictors. While the main focus is on interpretability, the resulting surrogate model also has reasonably good predictive performance.

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

↑