玩命加载中...
# 利用SHAP解释Xgboost模型 Xgboost相对于线性模型在进行预测时往往有更好的精度,但是同时也失去了线性模型的可解释性。所以Xgboost通常被认为是黑箱模型。 2017年,Lundberg和Lee的[论文](https://arxiv.org/pdf/1705.07874.pdf)提出了SHAP值这一广泛适用的方法用来解释各种模型(分类以及回归),其中最大的受益者莫过于之前难以被理解的黑箱模型,如xgboost和神经网络模型。 本教程中,我们在真实数据集上进行实操,利用SHAP来解释Xgboost模型。 预计学习用时:30分钟。 本教程基于Python 3.6版本、Xgboost 0.82版本以及shap 0.28.5版本。 原创者:**[东布东](http://sofasofa.io/user_profile.php?id=1003672)** | 修改校对:SofaSofa TeamM | ---- ### 1. Feature importance 在SHAP被广泛使用之前,我们通常用feature importance或者partial dependence plot来解释xgboost。 feature importance是用来衡量数据集中每个特征的重要性。 简单来说,每个特征对于提升整个模型的预测能力的贡献程度就是特征的重要性。(拓展阅读:[随机森林、xgboost中feature importance](http://sofasofa.io/forum_main_post.php?postid=1002662),[Partial Dependence Plot是什么意思?](http://sofasofa.io/forum_main_post.php?postid=1003120),[怎么利用permutation importance来解释xgboost模型](http://sofasofa.io/forum_main_post.php?postid=1003937)) Feature importance可以直观地反映出特征的重要性,看出哪些特征对最终的模型影响较大。但是无法判断特征与最终预测结果的关系是如何的。 下面这个例子中,我们用2018年足球球员身价数据(请在SofaSofa数据竞赛页面进行[数据下载](http://sofasofa.io/competition.php?id=7),下载解压后只需要`train.csv`这个文件)来具体阐述。 ```python # 加载模块 import xgboost as xgb import pandas as pd import numpy as np import matplotlib.pyplot as plt; plt.style.use('seaborn') # 读取数据,目标变量y是球员的身价(万欧元) data = pd.read_csv('train.csv') # 获得当时球员年龄 today = pd.to_datetime('2018-01-01') data['birth_date'] = pd.to_datetime(data['birth_date']) data['age'] = np.round((today - data['birth_date']).apply(lambda x: x.days) / 365., 1) # 选择特征,这里只是举例,未必是最佳组合 # 特征依次为身高(厘米)、潜力、速度、射门、传球、带球、防守、体格、国际知名度、年龄 cols = ['height_cm', 'potential', 'pac', 'sho', 'pas', 'dri', 'def', 'phy', 'international_reputation', 'age'] # 训练xgboost回归模型 model = xgb.XGBRegressor(max_depth=4, learning_rate=0.05, n_estimators=150) model.fit(data[cols], data['y'].values) # 获取feature importance plt.figure(figsize=(15, 5)) plt.bar(range(len(cols)), model.feature_importances_) plt.xticks(range(len(cols)), cols, rotation=-45, fontsize=14) plt.title('Feature importance', fontsize=14) plt.show() ``` <img src='feature_imp_1.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> 上图中,我们可以看出国际知名度、潜力和年龄是影响球员身价最重要的三个因素。但是这些因素和身价是正相关、负相关还是其他更复杂的相关性,我们无法从上图得知。我们也无法解读每个特征对每个个体的预测值的影响。 ### 2. SHAP value SHAP的名称来源于**SH**apley **A**dditive ex**P**lanation。 Shapley value起源于合作博弈论。比如说甲乙丙丁四个工人一起打工,甲和乙完成了价值100元的工件,甲、乙、丙完成了价值120元的工件,乙、丙、丁完成了价值150元的工件,甲、丁完成了价值90元的工件,那么该如何**公平、合理**地分配这四个人的工钱呢?Shapley提出了一个合理的计算方法(有兴趣地可以查看[原论文](https://apps.dtic.mil/dtic/tr/fulltext/u2/604084.pdf)),我们称每个参与者分配到的数额为Shapley value。 SHAP是由Shapley value启发的可加性解释模型。对于每个预测样本,模型都产生一个预测值,SHAP value就是该样本中每个特征所分配到的数值。 假设第$i$个样本为$x\_i$,第$i$个样本的第$j$个特征为$x\_{i,j}$,模型对第$i$个样本的预测值为$y\_i$,整个模型的基线(通常是所有样本的目标变量的均值)为$y\_{\text{base}}$,那么SHAP value服从以下等式。 $$y\_i=y\_{\text{base}}+f(x\_{i,1})+f(x\_{i,2})+\cdots+f(x\_{i,k})$$ 其中$f(x\_{i,1})$为$x\_{i,j}$的SHAP值。直观上看,$f(x\_{i,1})$就是第$i$个样本中第1个特征对最终预测值$y\_i$的贡献值,当$f(x\_{i,1})>0$,说明该特征提升了预测值,也正向作用;反之,说明该特征使得预测值降低,有反作用。 很明显可以看出,与上一节中feature importance相比,SHAP value最大的优势是SHAP能对于反映出**每一个样本**中的特征的影响力,而且还表现出影响的正负性。 ### 3. SHAP的Python实现 Python中SHAP值的计算由`shap`这个package实现,可以通过`pip install shap`安装。 下面我们针对第1节中训练出的模型`model`,计算其SHAP值。 引用package并且获得解释器`explainer`。 ```python import shap # model是在第1节中训练的模型 explainer = shap.TreeExplainer(model) ``` 获取训练集`data`各个样本各个特征的SHAP值。 因为`data`中有10441个样本以及10个特征,我们得到的`shap_values`的维度是$10441\times 10$。 ```python shap_values = explainer.shap_values(data[cols]) print(shap_values.shape) ``` (10441, 10) 我们也可以获得在第2节中提到的模型的基线$y\_{\text{base}}$。 通过对比发现,我们可以确认基线值就是训练集的目标变量的拟合值的均值。在这里例子中,目标变量是球员的身价(万欧元),也就是球员的平均身价为229万欧元。 ```python y_base = explainer.expected_value print(y_base) data['pred'] = model.predict(data[cols]) print(data['pred'].mean()) ``` 229.168 229.168 #### 3.1 单个样本的SHAP值 我们可以随机检查其中一位球员身价的预测值以及其特征对预测值的影响。 下面的数据框中第一列是特征名称,第二列是特征的数值,第三列是各个特征在该样本中对应的SHAP值。 ```python # 比如我们挑选数据集中的第30位 j = 30 player_explainer = pd.DataFrame() player_explainer['feature'] = cols player_explainer['feature_value'] = data[cols].iloc[j].values player_explainer['shap_value'] = shap_values[j] player_explainer ``` <div> <style scoped> .dataframe tbody tr th:only-of-type { vertical-align: middle; } .dataframe tbody tr th { vertical-align: top; } .dataframe thead th { text-align: right; } </style> <table border="1" class="dataframe"> <thead> <tr style="text-align: right;"> <th></th> <th>feature</th> <th>feature_value</th> <th>shap_value</th> </tr> </thead> <tbody> <tr> <th>0</th> <td>height_cm</td> <td>185.0</td> <td>0.624574</td> </tr> <tr> <th>1</th> <td>potential</td> <td>86.0</td> <td>1092.649780</td> </tr> <tr> <th>2</th> <td>pac</td> <td>69.0</td> <td>-2.964231</td> </tr> <tr> <th>3</th> <td>sho</td> <td>55.0</td> <td>-27.901392</td> </tr> <tr> <th>4</th> <td>pas</td> <td>68.0</td> <td>-18.346525</td> </tr> <tr> <th>5</th> <td>dri</td> <td>71.0</td> <td>1.230149</td> </tr> <tr> <th>6</th> <td>def</td> <td>76.0</td> <td>110.840019</td> </tr> <tr> <th>7</th> <td>phy</td> <td>84.0</td> <td>31.128113</td> </tr> <tr> <th>8</th> <td>international_reputation</td> <td>2.0</td> <td>97.137123</td> </tr> <tr> <th>9</th> <td>age</td> <td>20.7</td> <td>-180.918320</td> </tr> </tbody> </table> </div> 我们知道一个样本中各特征SHAP值的和加上基线值应该等于该样本的预测值。 我们可以做如下的验证。 ```python print('y_base + sum_of_shap_values: %.2f'%(y_base + player_explainer['shap_value'].sum())) print('y_pred: %.2f'%(data['pred'].iloc[j])) ``` y_base + sum_of_shap_values: 1332.65 y_pred: 1332.65 `shap`还提供极其强大的数据可视化功能。下图是对上面数据框的可视化。 蓝色表示该特征的贡献是负数,红色则表示该特征的贡献是正数。最长的红色条是潜力值,球员的潜力值很高,而他的身价也因此增加了1092万;最长的蓝色条是年龄,这个球员年龄较小才20岁出头,尚未到职业巅峰,未来也有诸多不确定性,身价也因此降低了180万元。 ```python shap.initjs() shap.force_plot(explainer.expected_value, shap_values[j], data[cols].iloc[j]) ``` <img src='example.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> #### 3.2 对特征的总体分析 除了能对单个样本的SHAP值进行可视化之外,还能对特征进行整体的可视化。 下图中每一行代表一个特征,横坐标为SHAP值。一个点代表一个样本,颜色越红说明特征本身数值越大,颜色越蓝说明特征本身数值越小。 我们可以直观地看出潜力`potential`是一个很重要的特征,而且基本上是与身价成正相关的。年龄`age`也会明显影响身价,蓝色点主要集中在SHAP小于0的区域,可见年纪小会降低身价估值,另一方面如果年纪很大,也会降低估值,甚至降低得更明显,因为`age`这一行最左端的点基本上都是红色的。 ```python shap.summary_plot(shap_values, data[cols]) ``` <img src='summary_plot.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> 我们也可以把一个特征对目标变量影响程度的绝对值的均值作为这个特征的重要性。 因为SHAP和feature_importance的计算方法不同,所以我们这里也得到了与第1节不同的重要性排序。 ```python shap.summary_plot(shap_values, data[cols], plot_type="bar") ``` <img src='feature_imp_2.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> #### 3.3 部分依赖图Partial Dependence Plot `SHAP`也提供了部分依赖图的功能,与传统的部分依赖图不同的是,这里纵坐标不是目标变量y的数值而是SHAP值。 比如下图中,年纪大概呈现出金字塔分布,也就是24到31岁这个年纪对球员的身价是拉抬作用,小于24以及大于31岁的球员身价则会被年纪所累。 ```python shap.dependence_plot('age', shap_values, data[cols], interaction_index=None, show=False) ``` <img src='pdp.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> #### 3.4 对多个变量的交互进行分析 我们也可以多个变量的交互作用进行分析。一种方式是采用`summary_plot`描绘出散点图,如下: ```python shap_interaction_values = shap.TreeExplainer(model).shap_interaction_values(data[cols]) shap.summary_plot(shap_interaction_values, data[cols], max_display=4) ``` <img src='summary_plot_interaction.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> 我们也可以用`dependence_plot`描绘两个变量交互下变量对目标值的影响。 ```python shap.dependence_plot('potential', shap_values, data[cols], interaction_index='international_reputation', show=False) ``` <img src='pdp_interaction.png' title='shap xgboost tutorial 教程-sofasofa.io' style="max-width: 75%"> ------ 如果本文对您有帮助,欢迎到<a href='https://zhuanlan.zhihu.com/p/64799119' target="_blank">知乎为我们点赞</a>! <ul class="pager"> <li class="next"><a href="../../tutorials.php"><b><i class="fa fa-graduation-cap" aria-hidden="true"></i>&nbsp; 学完咯!</b></a></li> </ul>